mirror of
https://github.com/thib8956/advent-of-code.git
synced 2026-05-25 19:53:26 +00:00
chore: implement "aoc run all" subcommand
This commit is contained in:
+44
-5
@@ -1,21 +1,49 @@
|
||||
import argparse
|
||||
from pathlib import Path
|
||||
from adventofcode.helper import run, get_input_file, get_auth
|
||||
|
||||
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__':
|
||||
import fileinput
|
||||
lines = [x.rstrip() for x in fileinput.input()]
|
||||
main(lines)
|
||||
"""
|
||||
|
||||
|
||||
def year_or_all(value):
|
||||
"""Custom type function to validate 'year' as either an integer or 'all'."""
|
||||
if value.lower() == "all":
|
||||
return value
|
||||
try:
|
||||
year = int(value)
|
||||
if 2015 <= year <= 2025:
|
||||
return year
|
||||
else:
|
||||
raise argparse.ArgumentTypeError(
|
||||
f"Year must be between 2015 and 2025. Got: {year}"
|
||||
)
|
||||
except ValueError:
|
||||
raise argparse.ArgumentTypeError(
|
||||
f"Invalid value: {value}. Must be an integer or 'all'."
|
||||
)
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Advent of Code CLI")
|
||||
subparsers = parser.add_subparsers(dest="command")
|
||||
@@ -25,13 +53,15 @@ def main():
|
||||
init_parser.add_argument("day", type=int)
|
||||
|
||||
run_parser = subparsers.add_parser("run", help="Run an aoc day")
|
||||
run_parser.add_argument("year", type=int)
|
||||
run_parser.add_argument("year", type=year_or_all)
|
||||
run_parser.add_argument("day", type=int, nargs="?", default=None)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.command == "init":
|
||||
handle_init(args.year, args.day)
|
||||
elif args.command == "run" and args.year == "all":
|
||||
handle_run_all()
|
||||
elif args.command == "run":
|
||||
handle_run(args.year, args.day)
|
||||
else:
|
||||
@@ -39,8 +69,13 @@ def main():
|
||||
|
||||
|
||||
def handle_init(year, day):
|
||||
if not 1 <= day <= 25:
|
||||
print(f"Invalid day: {day}. Must be between 1 and 25.")
|
||||
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
|
||||
@@ -69,5 +104,9 @@ def handle_run(year, day):
|
||||
run(year, day)
|
||||
|
||||
|
||||
def handle_run_all():
|
||||
run_all()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user