Load papers from PMT

paperloader.py

import subprocess
import sys

def inp(prmpt):
    result = input(prmpt)
    if result == "redo":
        raise IndexError # lol
    elif result == "quit":
        sys.exit(0)

    return result

urlstr = "http://pmt.physicsandmathstutor.com/download/Maths/A-level/{module}/Papers-OCR/{month}%20{year}%20!DOC!%20-%20{module}%20OCR.pdf"

def paper_open():
    print("----")
    module = inp("Module >>> ").upper()
    year   = inp("Year   >>> ")
    print("Jan: 1, Jun: 2")
    month  = {"1" : "January",
              "2" : "June"    }[inp("Month  >>> ")]
    paper  = urlstr.format(module = module, year = year, month = month)
    subprocess.call(["firefox", paper.replace("!DOC!", "QP"), paper.replace("!DOC!", "MS")])


if __name__ == "__main__":
    while True:
        try:
            paper_open()
        except IndexError:
            continue