mirror of
https://github.com/simple-login/app.git
synced 2024-11-14 08:01:13 +01:00
113 lines
3.4 KiB
Python
113 lines
3.4 KiB
Python
import argparse
|
|
from enum import Enum
|
|
from sys import argv, exit
|
|
|
|
from app.config import EVENT_LISTENER_DB_URI
|
|
from app.log import LOG
|
|
from events import event_debugger
|
|
from events.runner import Runner
|
|
from events.event_source import DeadLetterEventSource, PostgresEventSource
|
|
from events.event_sink import ConsoleEventSink, HttpEventSink
|
|
|
|
_DEFAULT_MAX_RETRIES = 100
|
|
|
|
|
|
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, max_retries: int):
|
|
if mode == Mode.DEAD_LETTER:
|
|
LOG.i("Using DeadLetterEventSource")
|
|
source = DeadLetterEventSource(max_retries)
|
|
elif mode == Mode.LISTENER:
|
|
LOG.i("Using PostgresEventSource")
|
|
source = PostgresEventSource(EVENT_LISTENER_DB_URI)
|
|
else:
|
|
raise ValueError(f"Invalid mode: {mode}")
|
|
|
|
if dry_run:
|
|
LOG.i("Starting with ConsoleEventSink")
|
|
sink = ConsoleEventSink()
|
|
else:
|
|
LOG.i("Starting with HttpEventSink")
|
|
sink = HttpEventSink()
|
|
|
|
runner = Runner(source=source, sink=sink)
|
|
runner.run()
|
|
|
|
|
|
def debug_event(event_id: str):
|
|
LOG.i(f"Debugging event {event_id}")
|
|
try:
|
|
event_id_int = int(event_id)
|
|
except ValueError:
|
|
raise ValueError(f"Invalid event id: {event_id}")
|
|
event_debugger.debug_event(event_id_int)
|
|
|
|
|
|
def run_event(event_id: str, delete_on_success: bool):
|
|
LOG.i(f"Running event {event_id}")
|
|
try:
|
|
event_id_int = int(event_id)
|
|
except ValueError:
|
|
raise ValueError(f"Invalid event id: {event_id}")
|
|
event_debugger.run_event(event_id_int, delete_on_success)
|
|
|
|
|
|
def args():
|
|
parser = argparse.ArgumentParser(description="Run event listener")
|
|
subparsers = parser.add_subparsers(dest="command")
|
|
|
|
listener_parser = subparsers.add_parser(Mode.LISTENER.value)
|
|
listener_parser.add_argument(
|
|
"--max-retries", type=int, default=_DEFAULT_MAX_RETRIES
|
|
)
|
|
listener_parser.add_argument("--dry-run", action="store_true")
|
|
|
|
dead_letter_parser = subparsers.add_parser(Mode.DEAD_LETTER.value)
|
|
dead_letter_parser.add_argument(
|
|
"--max-retries", type=int, default=_DEFAULT_MAX_RETRIES
|
|
)
|
|
dead_letter_parser.add_argument("--dry-run", action="store_true")
|
|
|
|
debug_parser = subparsers.add_parser("debug")
|
|
debug_parser.add_argument("event_id", help="ID of the event to debug")
|
|
|
|
run_parser = subparsers.add_parser("run")
|
|
run_parser.add_argument("event_id", help="ID of the event to run")
|
|
run_parser.add_argument("--delete-on-success", action="store_true")
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(argv) < 2:
|
|
print("Invalid usage. Pass a valid subcommand as argument")
|
|
exit(1)
|
|
|
|
args = args()
|
|
|
|
if args.command in [Mode.LISTENER.value, Mode.DEAD_LETTER.value]:
|
|
main(
|
|
mode=Mode.from_str(args.command),
|
|
dry_run=args.dry_run,
|
|
max_retries=args.max_retries,
|
|
)
|
|
elif args.command == "debug":
|
|
debug_event(args.event_id)
|
|
elif args.command == "run":
|
|
run_event(args.event_id, args.delete_on_success)
|
|
else:
|
|
print("Invalid command")
|
|
exit(1)
|