From e688c2b674fc7ad6a964a48df379e5abd01843a7 Mon Sep 17 00:00:00 2001 From: Tom van der Lee Date: Mon, 4 Dec 2023 10:40:34 +0100 Subject: Day4 --- aoc/__main__.py | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) (limited to 'aoc/__main__.py') diff --git a/aoc/__main__.py b/aoc/__main__.py index f5f08d3..596abee 100644 --- a/aoc/__main__.py +++ b/aoc/__main__.py @@ -1,3 +1,4 @@ +# -*- coding: utf-8 -*- import enum import typing import importlib @@ -25,15 +26,17 @@ def kwargs(kwarg: str) -> List: @app.command() -def run(day: str, - part: str = typer.Option("One", '--part', help="Assignment part. Defaults to 'One'.", show_choices=True), - example: bool = typer.Option(False, '--example', help="Use an example input file"), - kwargs: List[str] = typer.Argument(None) - ): - +def run( + day: str, + part: str = typer.Option( + "1", "--part", help="Assignment part. Defaults to 'One'.", show_choices=True + ), + example: bool = typer.Option(False, "--example", help="Use an example input file"), + kwargs: List[str] = typer.Argument(None), +): assignment_day = importlib.import_module(day) - Assignment = getattr(assignment_day, f"Assignment{part}") + Assignment = getattr(assignment_day, f"Assignment{AssignmentPart(part).name}") assignment = Assignment( path=os.path.dirname(assignment_day.__file__), **dict(kwargs) ) @@ -46,7 +49,10 @@ def run(day: str, ) end = perf_counter() delta = end - start - typer.secho(f'\n{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}', fg="green") + typer.secho( + f'\n{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}', + fg="green", + ) @app.command() -- cgit v1.2.3