setup.py

import datetime
import os
import subprocess
import sys
import time

import lib.aoc

def error_exit(msg):
    print(msg)
    os.system('pause')
    sys.exit(-1)

def open_editor(path):
    # Opens the IDLE editor for the path requested. This is done with a
    # separate command window so that when this program closes its command
    # window can cleanly disappear
    subprocess.Popen(['start', 'cmd', '/c', 'editor.bat', path],
                     shell=True)

def get_template(year, day):
    parse_template = '''def parse_all_ints(s):
    return list(map(lambda r:r[0], parse.findall('{:d}', s)))

def parse_input(s):
    for group in s.split('\\n\\n'):
        pass

    for line in s.splitlines():
        pass

    pass'''

    import_template = '''import collections
import functools
import itertools
import json
import math
import parse
import re
import sympy # sympy.parse_expr, sympy.solve, sympy.Eq
import z3 # z3.Solver, z3.Int, solver.add(constraint), solver.check(), solver.model()[variable].as_long()

import lib.aoc
import lib.cyk
import lib.graph
from lib.graphics import *
import lib.grid
import lib.hex_coord
import lib.lazy_dict
import lib.math
import lib.ocr
import lib.parsing'''

    if year == 2019:
        import_template += '''

intcode = __import__('2019.intcode').intcode'''

    part_1_template = f'''def part1(s):
    nums = parse_all_ints(s)
    lines = s.splitlines()
    groups = s.split('\\n\\n')
    data = parse_input(s)
##    grid = lib.grid.FixedGrid.parse(s, value_fn=int)
##    grid = lib.grid.FixedGrid.parse(s,
##                                    linesplit_fn=lambda line: line.split(),
##                                    value_fn=int)

    print(f'The answer to part one is {{answer}}')
    lib.aoc.give_answer({year}, {day}, 1, answer)'''

    if day == 25:
        part_2_template = f'''def part2(s):
    print('There is no part two for Christmas!')'''
    else:
        part_2_template = f'''def part2(s):
    pass
##    print(f'The answer to part two is {{answer}}')
##    lib.aoc.give_answer({year}, {day}, 2, answer)'''

    return f'''{import_template}

{parse_template}

{part_1_template}

{part_2_template}

INPUT = lib.aoc.get_input({year}, {day})
part1(INPUT)
part2(INPUT)
'''

try:
    year = input('Year ("warmup" for warmup): ')
    if year == 'warmup':
        day = int(input('Day: '))

        if day not in range(1, 26):
            error_exit(f'Advent of code runs from December 1st through 25th. Day {day} is invalid')

        previous_solutions = {}

        year = 2015
        while True:
            time_to_release = lib.aoc.time_to_release(year, day)
            if time_to_release > datetime.timedelta(days=-1):
                # This is a future (or current) problem!
                break

            if not lib.aoc.knows_solutions_for(year, day):
                year += 1
                continue

            if input(f'Warm up with {year} Day {day}? ').lower() in ('y', 'yes', '1'):
                path = f'{year}/{day:02}/solution.py'
                if os.path.exists(path):
                    with open(path, 'rb') as f:
                        previous_solutions[year] = f.read()

                os.makedirs(os.path.split(path)[0], exist_ok=True)
                with open(path, 'w+') as f:
                    f.write(get_template(year, day))

                lib.aoc.ensure_valid_session_cookie()

                open_editor(path)

                # Give a moment for the editor to open properly
                time.sleep(5)

                lib.aoc.open_pages_for(year, day)

            year += 1

        print('Finished with warmups. Waiting to restore old solutions...')
        os.system('pause')

        for year, old_solution in previous_solutions.items():
            path = f'{year}/{day:02}/solution.py'
            with open(path, 'wb') as f:
                f.write(old_solution)
    else:
        year = int(year)

        if year < 2015:
            error_exit(f'Advent of Code started in 2015! Year {year} is invalid')

        day = int(input('Day: '))

        if day not in range(1, 26):
            error_exit(f'Advent of code runs from December 1st through 25th. Day {day} is invalid')

        time_to_release = lib.aoc.time_to_release(year, day)
        if time_to_release >= datetime.timedelta(days=1):
            error_exit(f'{year} Day {day} is more than a day in the future!')

        path = f'{year}/{day:02}/solution.py'

        if not os.path.exists(path):
            print(f'Writing template for {year} Day {day}')
            os.makedirs(os.path.split(path)[0], exist_ok=True)
            with open(path, 'w+') as f:
                f.write(get_template(year, day))

        open_editor(path)

        lib.aoc.ensure_valid_session_cookie()

        if time_to_release <= datetime.timedelta(days=-1):
            print(f'{year} Day {day} is more than a day in the past!')
            if input('Begin time trial? ').lower() in ('y', 'yes', '1'):
                lib.aoc.begin_time_trial(year, day)
        else:
            lib.aoc.download_input_when_live(year, day)
except SystemExit:
    raise
except:
    import traceback
    traceback.print_exc()
    error_exit('Unhandled exception')