Day 9 [WIP]

1 files changed, 41 insertions, 17 deletions

diff --git a/aoc/__main__.py b/aoc/__main__.py
index 337bdfe..e1b2a85 100644
--- a/aoc/__main__.py
+++ b/aoc/__main__.py
@@ -2,8 +2,10 @@
 import argparse
 import importlib
 import os
+from pathlib import Path
+from shutil import copytree
 from time import perf_counter
-from typing import List
+from typing import List, Callable
 
 
 def day(assignment_part: str) -> str:
@@ -17,8 +19,41 @@ def kwargs(kwarg: str) -> List:
     return kwarg.split("=")
 
 
+def run(args: argparse.Namespace):
+    assignment_day = importlib.import_module(args.day)
+
+    Assignment = getattr(assignment_day, f"Assignment{args.part}")
+    assignment = Assignment(
+        path=os.path.dirname(assignment_day.__file__), **dict(args.kwargs)
+    )
+
+    start = perf_counter()
+    print(
+        assignment.run(
+            input=assignment.read_input(example=args.example),
+        )
+    )
+    end = perf_counter()
+    delta = end - start
+    print()
+    print(f'{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}')
+
+
+def new(args: argparse.Namespace):
+    self = importlib.import_module(".", package="aoc")
+    path = Path(self.__file__)
+    template = path.parent.joinpath("template")
+    target = path.parent.parent.joinpath(args.day)
+
+    try:
+        copytree(template, target)
+    except FileExistsError:
+        print(f"{args.day} already exists")
+
+
 parser = argparse.ArgumentParser(description="Advent of Code")
 
+parser.add_argument("command", type=str, choices=["run", "new"])
 parser.add_argument("day", type=str, help="Assignment day")
 parser.add_argument(
     "-p",
@@ -34,20 +69,9 @@ parser.add_argument("kwargs", type=kwargs, nargs="*")
 
 if __name__ == "__main__":
     args = parser.parse_args()
-    assignment_day = importlib.import_module(args.day)
-
-    Assignment = getattr(assignment_day, f"Assignment{args.part}")
-    assignment = Assignment(
-        path=os.path.dirname(assignment_day.__file__), **dict(args.kwargs)
-    )
 
-    start = perf_counter()
-    print(
-        assignment.run(
-            input=assignment.read_input(example=args.example),
-        )
-    )
-    end = perf_counter()
-    delta = end - start
-    print()
-    print(f'{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}')
+    match args.command:
+        case "run":
+            run(args)
+        case "new":
+            new(args)