From f4475ef1d408bd248056593d01c1d764a43715f8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Janek=20Hlavat=C3=BD?= Date: Sat, 21 Mar 2026 20:42:49 +0100 Subject: [PATCH] Modify cli --- src/beaky/cli.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/beaky/cli.py b/src/beaky/cli.py index 6097b49..04be9dc 100644 --- a/src/beaky/cli.py +++ b/src/beaky/cli.py @@ -20,7 +20,7 @@ def load_config(path: str) -> Config | None: def main() -> None: parser = argparse.ArgumentParser(prog="beaky") parser.add_argument("--config", help="Path to config file.", default="config/application.yml") - parser.add_argument("mode", choices=["screenshotter"], help="Mode of operation.") + parser.add_argument("mode", choices=["screenshotter", "parser", "class"], help="Mode of operation.") args = parser.parse_args() config = load_config(args.config) @@ -40,5 +40,12 @@ def main() -> None: screenshotter = Screenshotter(config) screenshotter.capture_tickets(data.links) + if args.mode == "parser": + for link in data.links: + print(link) + + if args.mode == "class": + pass + if __name__ == "__main__": main()