Integrate advent-of-code-data to the template

Author: nekeal

README.md

@@ -31,6 +31,13 @@ aoc new-day <day>
 It will create a `src/aoc/day_<day>` directory with a `__init__.py` and a `test_solution.py` python files.
 It will also create text files for both test and real data in the `data/` directory.
 
+#### NEW! Downloading input data
+
+Now this template uses [advent-of-code-data](https://github.com/wimglenn/advent-of-code-data) package to download
+your personal input data. To use it you need to set `AOC_SESSION` environment variable to your session cookie value.
+You can find it in your [browser's dev tools](https://github.com/wimglenn/advent-of-code-wim/issues/1)
+after logging in to the Advent of Code website.
+
 ### Implementing a solution for a day
 
 `BaseChallenge` class is provided to help you with the boilerplate code. By default, each day's solution inherits
@@ -100,6 +107,18 @@ package that will run tests for you after each change in the code.
 ptw src/aoc
 ```
 
+### Submitting solution
+
+Thanks to [advent-of-code-data](https://github.com/wimglenn/advent-of-code-wim/issues/1) package, you can submit your
+solution directly from the command line.
+
+```sh
+aoc submit <day> <part>
+```
+
+Under the hood, it will run the solution for the given day and part,
+and submit the answer to the Advent of Code website.
+
 ## Development
 
 * Clone this repository

poetry.lock

@@ -30,6 +30,7 @@ aoc = 'aoc.main:app'
 [tool.poetry.dependencies]
 python = ">=3.11, <4.0"
 typer = "^0.9.0"
+advent-of-code-data = "^2.0.1"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "*"
@@ -101,3 +102,7 @@ warn_unused_ignores = true
 warn_redundant_casts = true
 warn_return_any = true
 check_untyped_defs = true
+
+[[tool.mypy.overrides]]
+module = "aocd.*"
+ignore_missing_imports = true

pyproject.toml

@@ -3,8 +3,12 @@
 from pathlib import Path
 from typing import Optional
 
+import aocd
 import pytest
 import typer
+from aocd import AocdError
+from aocd.exceptions import PuzzleLockedError
+from aocd.models import Puzzle
 
 app = typer.Typer(no_args_is_help=True)
 
@@ -46,6 +50,35 @@ def run_challenge(day: int, test_data: bool):
         module.Challenge(use_test_data=False).run()
 
 
+def get_puzzle_object(day: int) -> Puzzle | None:
+    try:
+        puzzle = Puzzle(year=2023, day=day)
+        return puzzle
+    except AocdError:
+        pass
+    return None
+
+
+def write_example_data(puzzle, test_input_data):
+    try:
+        test_input_data.write_text(puzzle.examples[0].input_data)
+    except PuzzleLockedError:
+        echo(
+            "Failed writing test input data. Puzzle data is not available yet.",
+            fg=typer.colors.YELLOW,
+        )
+
+
+def write_input_data(puzzle, real_input_data):
+    try:
+        real_input_data.write_text(puzzle.input_data)
+    except PuzzleLockedError:
+        echo(
+            "Failed writing user input data. Puzzle data is not available yet.",
+            fg=typer.colors.YELLOW,
+        )
+
+
 @app.command()
 def run(
     day: int = typer.Argument(..., help="Day of the challenge to run."),
@@ -93,10 +126,35 @@ def verify(
             raise typer.Exit(1)
         del locals()["module"]
         pytest_args.append(module.__file__)
-    print(pytest_args)
     pytest.main(pytest_args)
 
 
+@app.command()
+def submit(
+    day: int,
+    part: str = typer.Argument(
+        help="Which part of the solution to submit. "
+        "You can use numbers 1/2 or letters a/b."
+    ),
+):
+    if part in ("1", "2"):
+        part = chr(ord("a") + int(part) - 1)
+    if part not in ("a", "b"):
+        typer.echo(
+            typer.style(
+                f"Invalid part {part}. Must be one of 1, 2, a, b.",
+                fg=typer.colors.RED,
+            )
+        )
+        raise typer.Exit(1)
+    module = import_challenge_module(day)
+    if part == "a":
+        solution = module.Challenge(use_test_data=False).part_1()
+    else:
+        solution = module.Challenge(use_test_data=False).part_2()
+    aocd.submit(solution, day=day, year=2023, part=part)
+
+
 @app.command()
 def new_day(
     day: int = typer.Argument(help="Day for which to create a directory."),
@@ -108,14 +166,15 @@ def new_day(
     ),
 ):
     """Create a directory for a new day challenge."""
+    puzzle = get_puzzle_object(day)
     if day < 1 or day > 25:
         typer.echo(typer.style("Day must be between 1 and 25.", fg=typer.colors.RED))
         raise typer.Exit(1)
     destination = Path(f"src/aoc/day_{day:02}")
     if destination.exists() and not force:
         typer.echo(
             typer.style(
-                f"Day {day} already exists. Add -f flag if you want to ovewrite",
+                f"Day {day} already exists. Add -f flag if you want to overwrite",
                 fg=typer.colors.RED,
             )
         )
@@ -126,26 +185,33 @@ def new_day(
     typer.echo(
         typer.style(f"Created solution directory for day {day}.", fg=typer.colors.GREEN)
     )
-    if (real_input_data := Path(f"data/{day:02}_input.txt")).exists():
+    if (
+        real_input_data := Path(f"data/{day:02}_input.txt")
+    ).exists() and real_input_data.stat().st_size:
         typer.echo(
             typer.style(
                 f"Input data already exists for day {day}.", fg=typer.colors.YELLOW
             )
         )
     else:
         real_input_data.touch()
+        if puzzle is not None:
+            write_input_data(puzzle, real_input_data)
         typer.echo(
             typer.style(f"Created input data for day {day}.", fg=typer.colors.GREEN)
         )
-    real_input_data.touch()
-    if (test_input_data := Path(f"data/{day:02}_test_input.txt")).exists():
+    if (
+        test_input_data := Path(f"data/{day:02}_test_input.txt")
+    ).exists() and test_input_data.stat().st_size:
         typer.echo(
             typer.style(
                 f"Test input data already exists for day {day}.", fg=typer.colors.YELLOW
             )
         )
     else:
         test_input_data.touch()
+        if puzzle is not None and puzzle.examples:
+            write_example_data(puzzle, test_input_data)
         typer.echo(
             typer.style(
                 f"Created test input data for day {day}.", fg=typer.colors.GREEN