Implement top-level CLI for executing solvers and dynamically fetching inputs
This commit is contained in:
4
advent_of_code/__init__.py
Normal file
4
advent_of_code/__init__.py
Normal file
@@ -0,0 +1,4 @@
|
||||
"My solutions to Advent of Code 2023"
|
||||
|
||||
__author__ = "Nettika <nettika@leaf.ninja>"
|
||||
__version__ = "1.0.0"
|
67
advent_of_code/__main__.py
Normal file
67
advent_of_code/__main__.py
Normal file
@@ -0,0 +1,67 @@
|
||||
import time
|
||||
from argparse import ArgumentParser
|
||||
|
||||
import requests
|
||||
|
||||
from advent_of_code import __version__
|
||||
from advent_of_code.solver import Solver
|
||||
|
||||
solvers: dict[int, Solver] = {}
|
||||
|
||||
|
||||
def main():
|
||||
# Parse CLI arguments
|
||||
parser = ArgumentParser(prog="Advent of Code 2023 Solver")
|
||||
parser.add_argument(
|
||||
"--session",
|
||||
"-s",
|
||||
required=True,
|
||||
help="Session token",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--day",
|
||||
"-d",
|
||||
required=True,
|
||||
type=int,
|
||||
help="Day of the month to solve",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--version",
|
||||
action="version",
|
||||
version=f"%(prog)s {__version__}",
|
||||
)
|
||||
args = parser.parse_args()
|
||||
|
||||
# Validate day number
|
||||
if args.day > 30 or args.day < 0:
|
||||
print("Invalid day number")
|
||||
exit(1)
|
||||
|
||||
# Fetch input for the day
|
||||
print(f"Fetching Day {args.day} Input...")
|
||||
with requests.get(
|
||||
f"https://adventofcode.com/2023/day/{args.day}/input",
|
||||
cookies={"session": args.session},
|
||||
) as response:
|
||||
if not response.ok:
|
||||
print("Unable to fetch input. Check your session token.")
|
||||
exit(1)
|
||||
input = response.text.strip()
|
||||
|
||||
# Resolve solver
|
||||
solver = solvers.get(args.day)
|
||||
if not solver:
|
||||
print(f"No solver for day {args.day}")
|
||||
exit(1)
|
||||
|
||||
# Execute solvers
|
||||
print(f"Solving day {args.day}'s puzzle")
|
||||
start = time.time()
|
||||
print("Part 1 -", solver.solve_part_1(input))
|
||||
print("Part 2 -", solver.solve_part_2(input))
|
||||
end = time.time()
|
||||
print(f"Solved in {end-start:,} seconds.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
11
advent_of_code/solver.py
Normal file
11
advent_of_code/solver.py
Normal file
@@ -0,0 +1,11 @@
|
||||
from abc import ABC, abstractmethod
|
||||
|
||||
|
||||
class Solver(ABC):
|
||||
@abstractmethod
|
||||
def solve_part_1(self, input: str) -> None:
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def solve_part_2(self, input: str) -> None:
|
||||
pass
|
@@ -1,7 +1,14 @@
|
||||
[build-system]
|
||||
requires = ["setuptools"]
|
||||
build-backend = "setuptools.build_meta"
|
||||
|
||||
[project]
|
||||
name = "advent-of-code"
|
||||
name = "advent_of_code"
|
||||
version = "0.1.0"
|
||||
dependencies = ["pytest"]
|
||||
dependencies = ["requests"]
|
||||
|
||||
[project.optional-dependencies]
|
||||
test = ["pytest", "types-requests"]
|
||||
|
||||
[project.scripts]
|
||||
advent-of-code-2023 = "advent_of_code.__main__:main"
|
||||
|
Reference in New Issue
Block a user