app/event_listener.py

import argparse
from enum import Enum
from sys import argv, exit

from app.config import DB_URI
from events.runner import Runner
from events.event_source import DeadLetterEventSource, PostgresEventSource
from events.event_sink import ConsoleEventSink


class Mode(Enum):
    DEAD_LETTER = "dead_letter"
    LISTENER = "listener"

    @staticmethod
    def from_str(value: str):
        if value == Mode.DEAD_LETTER.value:
            return Mode.DEAD_LETTER
        elif value == Mode.LISTENER.value:
            return Mode.LISTENER
        else:
            raise ValueError(f"Invalid mode: {value}")


def main(mode: Mode, dry_run: bool):
    if mode == Mode.DEAD_LETTER:
        source = DeadLetterEventSource()
    elif mode == Mode.LISTENER:
        source = PostgresEventSource(DB_URI)
    else:
        raise ValueError(f"Invalid mode: {mode}")

    if dry_run:
        sink = ConsoleEventSink()
    else:
        sink = ConsoleEventSink()

    runner = Runner(source=source, sink=sink)
    runner.run()


def args():
    parser = argparse.ArgumentParser(description="Run event listener")
    parser.add_argument(
        "mode",
        help="Mode to run",
        choices=[Mode.DEAD_LETTER.value, Mode.LISTENER.value],
    )
    parser.add_argument("--dry-run", help="Dry run mode", action="store_true")
    return parser.parse_args()


if __name__ == "__main__":
    if len(argv) < 2:
        print("Invalid usage. Pass 'listener' or 'dead_letter' as argument")
        exit(1)

    args = args()
    main(Mode.from_str(args.mode), args.dry_run)