From d0f3fc61d12c1a13206a0b7c3b5065bc922d91ec Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Tue, 30 Jun 2026 16:45:24 -0400 Subject: [PATCH] Add event argument --- main.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 5ebf6a3..6d6a203 100644 --- a/main.py +++ b/main.py @@ -1,9 +1,11 @@ import argparse -def main(day: int, test: bool = False) -> str: +FIRST_EVENT = 2015 +LAST_EVENT = 2015 + +def main(day: int, event: int, test: bool = False) -> str: print("Called main...") - print("Day is: {}".format(day)) - print("Testing: {}".format(test)) + return f"Day {day} of event {event} processing [Test: {test}]" if __name__ == "__main__": @@ -15,9 +17,15 @@ if __name__ == "__main__": "-d", "--day", type=int, - choices=range(1, 25), + choices=range(1, 26), help="Integer day number to solve for (1-25)." ) + parser.add_argument( + "-e", + "--event", + type=int, + choices=range(FIRST_EVENT, LAST_EVENT + 1) + ) args = parser.parse_args() - main(day=args.day, test=args.test) + print(main(day=args.day, event=args.event, test=args.test))