thib8956/advent-of-code
1
0
Fork 0
mirror of https://github.com/thib8956/advent-of-code.git synced 2026-02-27 10:48:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: implement aoc init subcommand

Browse Source
This commit is contained in:
Thibaud Gasser
2026-02-19 12:08:01 +01:00
parent 93377a0295
commit 37f2b62db2
1 changed files with 48 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
adventofcode/aoc.py
View File

@@ -1,35 +1,68 @@
import argparse import argparse
from adventofcode.helper import run, get_input_file from pathlib import Path
from adventofcode.helper import run, get_input_file, get_auth
TEMPLATE = """#!/usr/bin/env python3
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 main(): def main():
parser = argparse.ArgumentParser(description="Advent of Code CLI") parser = argparse.ArgumentParser(description="Advent of Code CLI")
subparsers = parser.add_subparsers(dest='command') subparsers = parser.add_subparsers(dest="command")
# Sous-commande init init_parser = subparsers.add_parser("init", help="Init an aoc day")
init_parser = subparsers.add_parser('init', help='Init an aoc day') init_parser.add_argument("year", type=int)
init_parser.add_argument('year', type=int) init_parser.add_argument("day", type=int)
init_parser.add_argument('day', type=int)
# Sous-commande run run_parser = subparsers.add_parser("run", help="Run an aoc day")
run_parser = subparsers.add_parser('run', help='Run an aoc day') run_parser.add_argument("year", type=int)
run_parser.add_argument('year', type=int) run_parser.add_argument("day", type=int, nargs="?", default=None)
run_parser.add_argument('day', type=int, nargs='?', default=None)
args = parser.parse_args() args = parser.parse_args()
if args.command == 'init': if args.command == "init":
handle_init(args.year, args.day) handle_init(args.year, args.day)
elif args.command == 'run': elif args.command == "run":
handle_run(args.year, args.day) handle_run(args.year, args.day)
else: else:
parser.print_help() parser.print_help()
def handle_init(year, day): def handle_init(year, day):
# TODO initialize directory if needed, download input file and create if not 1 <= day <= 25:
# dayX.py from a template print(f"Invalid day: {day}. Must be between 1 and 25.")
raise NotImplementedError("init") 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)
def handle_run(year, day): def handle_run(year, day):
@@ -38,4 +71,3 @@ def handle_run(year, day):
if __name__ == "__main__": if __name__ == "__main__":
main() main()