main.py

import copy, importlib, os

from src.utils import Color, colorize, cliargs

solution_dirs = []

if cliargs.solution:
    solution_dir = (
        f"0{str(cliargs.solution)}" if cliargs.solution < 10 else str(cliargs.solution)
    )

    if os.path.exists(f"src/{solution_dir}"):
        solution_dirs.append(f"src/{solution_dir}")
    else:
        raise Exception(f"Solution directory 'src/{solution_dir}' does not exist!")
else:
    for dirno in range(1, 25):
        solution_dir = str(dirno).zfill(2)

        if os.path.isdir(f"src/{solution_dir}") and solution_dir:
            solution_dirs.append(f"src/{solution_dir}")

print(colorize("Printing Advent of Code solutions for 2024!", Color.YELLOW))

for index, solution_dir in enumerate(solution_dirs):
    solution_package = solution_dir.replace("/", ".") + ".solution"
    solution_runner = importlib.import_module(solution_package)

    values = solution_runner.transform_prompt()

    indent = " " * 2

    if index > 0:
        print()

    day = str(int(solution_dir.split("/")[-1]))

    print(indent + colorize(f"- DAY {day}", Color.CYAN))

    print(
        indent * 3
        + "Part 1: "
        + colorize(solution_runner.part_1_solution(copy.deepcopy(values)), Color.YELLOW)
    )
    print(
        indent * 3
        + "Part 2: "
        + colorize(solution_runner.part_2_solution(copy.deepcopy(values)), Color.YELLOW)
    )