template.py

from aocd import submit
from aocd.models import Puzzle
from datetime import datetime
import re
import collections
import itertools
from copy import deepcopy
import math
import heapq
import string
from functools import cmp_to_key as c2k

EXAMPLE = True
PART = 1


def process_input(file):
    return file.read().strip().splitlines()


# ==================================================
if EXAMPLE:
    day, year = datetime.today().strftime('%d'), datetime.today().strftime('%Y')

    example_data, p1_answer, p2_answer, *_ = Puzzle(year=int(year), day=int(day)).examples[0]
    with open('test.txt') as file:
        LINES = process_input(file)

else:
    with open('input.txt') as file:
        LINES = process_input(file)
# ==================================================

total = 0
for line in LINES:
    print(line)

# ==================================================
if EXAMPLE:
    if (p1_answer if PART == 1 else p2_answer) is not None:
        assert total == int(
            p1_answer if PART == 1 else p2_answer), f'should be {p1_answer if PART == 1 else p2_answer}, got {total}'
    else:
        print(f'Part {PART} answer:', total)
else:
    print(total)
    submit(total)