1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import enum
import typing
import importlib
import os
from pathlib import Path
from shutil import copytree
from time import perf_counter
from typing import List, Callable
import typer
app = typer.Typer()
class AssignmentPart(str, enum.Enum):
One = "1"
Two = "2"
def day(assignment_part: str) -> str:
return assignment_part
def kwargs(kwarg: str) -> List:
return kwarg.split("=")
@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)
):
assignment_day = importlib.import_module(day)
Assignment = getattr(assignment_day, f"Assignment{part}")
assignment = Assignment(
path=os.path.dirname(assignment_day.__file__), **dict(kwargs)
)
start = perf_counter()
typer.echo(
assignment.run(
input=assignment.read_input(example=example),
)
)
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")
@app.command()
def new(day: str):
self = importlib.import_module(".", package="aoc")
path = Path(self.__file__)
template = path.parent.joinpath("template")
target = path.parent.parent.joinpath(day)
try:
copytree(template, target)
except FileExistsError:
typer.secho(f"{day} already exists", fg="red")
if __name__ == "__main__":
app()
|
