refactor: move init command to helper.py

Makefile

@@ -11,10 +11,12 @@ all: venv install
 
 # Create virtual environment if it doesn't exist
 venv:
-	@echo "Creating virtual environment..."
 	@if [ ! -d $(VENV_DIR) ]; then \
+		echo "Creating virtual environment..."; \
 		python3 -m venv $(VENV_DIR); \
 	fi
+	@echo "Source virtualenv with:"
+	@echo "source $(VENV_DIR)/bin/activate"
 
 install: venv
 	@$(PYTHON) -m pip install --upgrade pip
@@ -23,4 +25,3 @@ install: venv
 clean:
 	@echo "Removing virtual environment..."
 	@rm -rf $(VENV_DIR)
-

README.md

@@ -27,12 +27,23 @@ $ pip install -e .
 ### Run a single day
 
 ```shell
-$ aoc <year> <day>
+$ aoc run <year> <day>
 ```
 
 ### Run a whole year
 
 ```shell
-$ aoc <year>
+$ aoc run <year>
 ```
 
+## Run all years
+
+```shell
+$ aoc run all
+```
+
+## Init a new day
+
+```shell
+$ aoc init <year> <day>
+```

adventofcode/2019/day13/day13.py

@@ -1,14 +1,11 @@
 #!/usr/bin/env python3
-# TODO replace PYTHONPATH hack with a proper solution, like making intcode an
-# installed module https://stackoverflow.com/a/50194143
 import sys
 import time
-from itertools import zip_longest
-from pathlib import Path
-from collections import defaultdict, Counter
 from dataclasses import dataclass
+from itertools import zip_longest
+
+from adventofcode.intcode import Interpreter, interpret_intcode
 
-from adventofcode.intcode import interpret_intcode, Interpreter
 
 @dataclass
 class State:
@@ -42,8 +39,7 @@ def parse_map(output):
             grid[x, y] = tile
     paddle = next((k for k, v in grid.items() if v == 3), None)
     ball = next((k for k, v in grid.items() if v == 4), None)
-    bounds = (max(x for x, y in grid.keys()),
-              max(y for x, y in grid.keys()))
+    bounds = (max(x for x, y in grid.keys()), max(y for x, y in grid.keys()))
     return grid, bounds, score, paddle, ball
 
 
@@ -108,9 +104,10 @@ def main(inp):
     part1(program)
     part2(program)
 
+
 if __name__ == "__main__":
     import sys
+
     infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
     with open(infile) as raw_input:
         main(raw_input)
-

adventofcode/aoc.py

@@ -1,29 +1,6 @@
 import argparse
-from pathlib import Path
 
-from adventofcode.helper import (
-    MAX_YEAR,
-    MIN_YEAR,
-    get_auth,
-    get_input_file,
-    get_max_day,
-    run,
-    run_all,
-)
-
-TEMPLATE = """#!/usr/bin/env python3
-import fileinput
-
-
-def main(inp):
-    for l in inp:
-        print(l)
-
-
-if __name__ == '__main__':
-    lines = [x.rstrip() for x in fileinput.input()]
-    main(lines)
-"""
+from adventofcode.helper import init, run, run_all
 
 
 def year_or_all(value):
@@ -69,35 +46,7 @@ def main():
 
 
 def handle_init(year, day):
-    if not MIN_YEAR <= year <= MAX_YEAR:
-        print(f"Invalid year: {year}. Must be between {MIN_YEAR} and {MAX_YEAR}.")
-        return
-
-    max_day = get_max_day(year)
-    if not 1 <= day <= max_day:
-        print(f"Invalid day: {day}. Must be between 1 and {max_day}.")
-        return
-
-    root = Path(__file__).parent
-    day_dir = root / str(year) / f"day{day}"
-    day_dir.mkdir(parents=True, exist_ok=True)
-
-    script_path = day_dir / f"day{day}.py"
-    if not script_path.exists():
-        script_path.write_text(TEMPLATE)
-        print(f"Created {script_path}")
-    else:
-        print(f"{script_path} already exists")
-
-    input_path = day_dir / "input.txt"
-    if not input_path.exists():
-        try:
-            get_auth()
-            res = get_input_file(year, day)
-            input_path.write_bytes(res.read())
-            print(f"Downloaded {input_path}")
-        except Exception as e:
-            print(f"Could not download input: {e}", file=__import__("sys").stderr)
+    init(year, day)
 
 
 def handle_run(year, day):

adventofcode/helper.py

@@ -9,6 +9,22 @@ from pathlib import Path
 MIN_YEAR = 2015
 MAX_YEAR = 2025
 ROOTPATH = Path(os.path.dirname(os.path.realpath(__file__)))
+TEMPLATE = """#!/usr/bin/env python3
+import fileinput
+
+
+def main(inp):
+    for l in inp:
+        print(l)
+
+
+if __name__ == '__main__':
+    lines = [x.rstrip() for x in fileinput.input()]
+    main(lines)
+"""
+
+
+_auth = None
 
 
 def get_max_day(year):
@@ -16,7 +32,16 @@ def get_max_day(year):
     return 25 if year < 2025 else 12
 
 
-_auth = None
+def validate_day(day, year):
+    if not 1 <= day <= get_max_day(year):
+        return f"Invalid day: {day} for year {year}. Must be between 1 and {get_max_day(year)}"
+    return None
+
+
+def validate_year(year):
+    if not MIN_YEAR <= year <= MAX_YEAR:
+        return f"Invalid year: {year}. Must be between {MIN_YEAR} and {MAX_YEAR}"
+    return None
 
 
 def _load_env_file(env_path: Path) -> dict:
@@ -69,25 +94,56 @@ def resolve_paths(year, day):
     return script_path, input_path
 
 
+def init(year, day):
+    if (err := validate_year(year)) is not None:
+        print(err)
+        return
+
+    if (err := validate_day(day, year)) is not None:
+        print(err)
+        return
+
+    path = ROOTPATH / Path(f"{year}/day{day}")
+    path.mkdir(parents=True, exist_ok=True)
+
+    script_path, input_path = resolve_paths(year, day)
+    if script_path.exists():
+        print(f"Script {script_path} already exists.")
+    else:
+        script_path.write_text(TEMPLATE)
+        print(f"Created {script_path}")
+
+    if input_path.exists():
+        print(f"Input {input_path} already exists.")
+    else:
+        try:
+            get_auth()
+            res = get_input_file(year, day)
+            input_path.write_bytes(res.read())
+            print(f"Downloaded {input_path}")
+        except Exception as e:
+            print(f"Could not download input: {e}", file=sys.stderr)
+
+
 def run_all():
-    for year in range(2015, MAX_YEAR + 1):
+    for year in range(MIN_YEAR, MAX_YEAR + 1):
         print(f"Running year {year}")
         run(year, None)
 
 
 def run(year, day):
-    if not MIN_YEAR <= year <= MAX_YEAR:
-        print(f"Invalid year {year}", file=sys.stderr)
+    if (err := validate_year(year)) is not None:
+        print(err, file=sys.stderr)
         exit(1)
 
     if day is not None:
-        if not 1 <= day <= get_max_day(year):
-            print(f"Invalid day {day}", file=sys.stderr)
+        if (err := validate_day(day, year)) is not None:
+            print(err, file=sys.stderr)
             exit(1)
 
         script_path, input_path = resolve_paths(year, day)
         if not script_path.exists():
-            print(f"Invalid day {day}", file=sys.stderr)
+            print(f"Script file {script_path} does not exist", file=sys.stderr)
             exit(1)
         if not input_path.exists():
             print(f"Downloading input file {input_path}")
@@ -98,7 +154,7 @@ def run(year, day):
 
         run_day(script_path, input_path)
     else:
-        for day in range(1, 26):
+        for day in range(1, get_max_day(year) + 1):
             script_path, input_path = resolve_paths(year, day)
             if script_path.exists():
                 if not input_path.exists():