Modify cli

This commit is contained in:
2026-03-21 20:42:49 +01:00
parent 5add445949
commit f4475ef1d4

View File

@@ -20,7 +20,7 @@ def load_config(path: str) -> Config | None:
def main() -> None: def main() -> None:
parser = argparse.ArgumentParser(prog="beaky") parser = argparse.ArgumentParser(prog="beaky")
parser.add_argument("--config", help="Path to config file.", default="config/application.yml") 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() args = parser.parse_args()
config = load_config(args.config) config = load_config(args.config)
@@ -40,5 +40,12 @@ def main() -> None:
screenshotter = Screenshotter(config) screenshotter = Screenshotter(config)
screenshotter.capture_tickets(data.links) screenshotter.capture_tickets(data.links)
if args.mode == "parser":
for link in data.links:
print(link)
if args.mode == "class":
pass
if __name__ == "__main__": if __name__ == "__main__":
main() main()