AoC input downloader

aoc.py

import os.path
import requests

SESSION_COOKIE = "" # Add your session cookie here
URL = "http://adventofcode.com/2017/day/{}/input"

def download(day, filename):
    r = requests.get(
        URL.format(day),
        cookies=dict(session=SESSION_COOKIE))

    if r.status_code != 200:
        raise Exception(r.text)

    with open(filename, "w") as fd:
        fd.write(r.text)

def get_input(day):
    filename = "{:02}-input.txt".format(day)

    if not os.path.isfile(filename):
        download(day, filename)

    with open(filename) as fd:
        for line in fd:
            yield line