mirror of
https://github.com/thib8956/advent-of-code.git
synced 2025-08-23 16:01:59 +00:00
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
|
import urllib.request
|
|
import getpass
|
|
import sys
|
|
import time
|
|
import subprocess
|
|
import os
|
|
from pathlib import Path
|
|
|
|
ROOTPATH = Path(os.path.dirname(os.path.realpath(__file__)))
|
|
|
|
_auth = None
|
|
|
|
|
|
def get_auth():
|
|
global _auth
|
|
if _auth is None:
|
|
if "AUTH" in os.environ:
|
|
_auth = os.environ["AUTH"]
|
|
else:
|
|
_auth = getpass.getpass(prompt="Cookie: ")
|
|
|
|
|
|
def get_input_file(year, day):
|
|
url = f"https://adventofcode.com/{year}/day/{day}/input"
|
|
r = urllib.request.Request(url)
|
|
r.add_header("Cookie", f"session={_auth}")
|
|
res = urllib.request.urlopen(r)
|
|
return res
|
|
|
|
|
|
def run(year, day):
|
|
if day is not None:
|
|
path = ROOTPATH / Path(f"{year}/day{day}")
|
|
script_path = path / Path(f"day{day}.py")
|
|
input_path = path / Path("input.txt")
|
|
if not script_path.exists():
|
|
print(f"Invalid day {day}", file=sys.stderr)
|
|
exit(1)
|
|
if not input_path.exists():
|
|
print(f"Downloading input file {input_path}")
|
|
get_auth()
|
|
with open(input_path, "wb") as f:
|
|
res = get_input_file(year, day)
|
|
f.write(res.read())
|
|
|
|
run_day(script_path, input_path)
|
|
else:
|
|
for day in range(1, 26):
|
|
path = ROOTPATH / Path(f"{year}/day{day}")
|
|
script_path = path / Path(f"day{day}.py")
|
|
input_path = path / Path("input.txt")
|
|
if script_path.exists():
|
|
if not input_path.exists():
|
|
print(f"- downloading input file {input_path}")
|
|
get_auth()
|
|
with open(input_path, "wb") as f:
|
|
res = get_input_file(year, day)
|
|
f.write(res.read())
|
|
run_day(script_path, input_path)
|
|
|
|
|
|
def run_day(script_path, input_path):
|
|
try:
|
|
print(f"> running {script_path}")
|
|
start = time.time()
|
|
res = subprocess.run([sys.executable, script_path.absolute(), input_path.absolute()], check=True, stdout=subprocess.PIPE, timeout=30)
|
|
elapsed = time.time() - start
|
|
print(res.stdout.decode())
|
|
print(f"> ran {script_path} in {elapsed:.3f}s")
|
|
except subprocess.TimeoutExpired:
|
|
print(f"> timeout {script_path} after 30s", file=sys.stderr)
|
|
|