#! /bin/python

import random

pins = []

print("Sparrows⚿ Revolver Loadout Generator")
print("Generates lists of pins for filling the Sparrows⚿ Revolver")
print("Hard mode assumes you have the Check and Ammo pin sets from Sparrows⚿")
while (pins == []):
    choice = input("Choose Difficulty (Stock), (E)asy, (M)edium, (H)ard: ")

    ez = ["Std","Spl"]
    md = ["Serr", "Mshrm"]
    hd = md + ["Check", "Ammo"]
    md = md + ez

    if choice.upper() == "E":
        pins = ez
    elif choice.upper() == "M":
        pins = md
    elif choice.upper() == "H":
        pins = hd
    else:
        print ("|\t1\t|\t2\t|\t3\t|\t4\t|\t5\t|\t6\t|")
        print ("|\tStd\t|\tStd\t|\tStd\t|\tStd\t|\tStd\t|\tX\t|")
        print ("|\tStd\t|\tSpl\t|\tStd\t|\tSpl\t|\tStd\t|\tX\t|")
        print ("|\tStd\t|\tSerr\t|\tStd\t|\tSerr\t|\tStd\t|\tX\t|")
        print ("|\tStd\t|\tMshrm\t|\tStd\t|\tMshrm\t|\tStd\t|\tX\t|")
        print("Good Luck!")
        quit()

print ("|\t1\t|\t2\t|\t3\t|\t4\t|\t5\t|\t6\t|")
for row in range (0, 4):
    if choice.upper() == "E":
        print("|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" "|\tX\t|")
    else:
        print("|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" + "|\t" + random.choice(pins) + "\t" "|\t" + random.choice(pins) + "\t|")
print ("Good Luck!")