advent-of-code/run.py

import urllib.request
import getpass
import sys
import time
import subprocess
import os
from pathlib import Path


_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 main(year, day):
    if day is not None:
        path = 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 = 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)


if __name__ == "__main__":
    if len(sys.argv) <= 1:
        print(f"Usage: {__file__} <year> [<day>]", file=sys.stderr)
        exit(1)
    year = sys.argv[1] 
    day = sys.argv[2] if len(sys.argv) > 2 else None
    main(year, day)