AoC/aoc2025/aocutils.py

import os
import requests
import time
import bs4
import markdownify as md

YEAR = "2025"
session_id = "53616c7465645f5f74bf2aff62945c49fba5f96848423a482d4c5b51ac366cb778fe7aaa623cedb5d38ff64f30317c6146590748dab77ccd9e902633765a29ce"


python_text = """import os

with open(os.path.join(os.path.dirname(__file__), "example.txt")) as example:
    example_data = example.read().splitlines()

with open(os.path.join(os.path.dirname(__file__), "input.txt")) as example:
    input_data = example.read().splitlines()
"""


def create_files(task_dir: str, day: int):
    go_path = os.path.join(task_dir, f"day{day}.go")
    python_path = os.path.join(task_dir, f"day{day}.py")
    if os.path.exists(go_path):
        os.utime(go_path, None)
    else:
        open(go_path, "a").close()
    if os.path.exists(python_path):
        os.utime(python_path, None)
    else:
        with open(python_path, "a") as python_file:
            python_file.write(python_text)
            python_file.close()


def generate_readme(task_dir: str, day: int):
    os.makedirs(task_dir, exist_ok=True)
    readme_path = os.path.join(task_dir, "README.md")
    cookies_dict = {"session": session_id}

    soup = bs4.BeautifulSoup(
        requests.get(f"https://adventofcode.com/{YEAR}/day/{day}", cookies=cookies_dict).content,
        features="html.parser",
    )
    with open(readme_path, "w") as readme:
        readme.write(md.markdownify(str(soup.find_all("article")[0])))
    if len(soup.find_all("article")) > 1:
        with open(readme_path, "a") as readme:
            readme.write("\n\n")
            readme.write(md.markdownify(str(soup.find_all("article")[1])))


def get_input(task_dir: str, day: int) -> tuple[list[str], list[str]] | None:
    input_path = os.path.join(task_dir, "input.txt")
    example_path = os.path.join(task_dir, "example.txt")
    readme_path = os.path.join(task_dir, "README.md")

    cookies_dict = {"session": session_id}

    os.makedirs(task_dir, exist_ok=True)

    if os.path.exists(input_path):
        with open(input_path, "r") as f:
            input = f.read().splitlines()
    else:
        input = requests.get(f"https://adventofcode.com/{YEAR}/day/{day}/input", cookies=cookies_dict).text
        with open(input_path, "w") as f:
            f.write(input.strip())
        input = input.splitlines()

    if os.path.exists(example_path):
        with open(example_path, "r") as e:
            example = e.read().splitlines()
    elif os.path.exists(readme_path):
        with open(example_path, "w") as e:
            with open(readme_path, "r") as r:
                example = r.read().split("\n\n```\n")[1]
            e.write(example)
            example = example.splitlines()
    else:
        print("call `generate_readme()` first!")
        return

    return input, example


def bench(part):
    def wrapper(*args, **kwargs):
        start = time.perf_counter()
        value = part(*args, **kwargs)
        print(f"\tevaluation time: {time.perf_counter() - start} s")
        return value

    return wrapper


if __name__ == "__main__":
    day = 6
    root = os.path.dirname(__file__)
    task_dir = os.path.join(root, f"day{day}")
    generate_readme(task_dir, day)
    get_input(task_dir, day)
    create_files(task_dir, day)