'''
    DS2000
    Spring 2020
    Source code from class - driver for the Iowa Caucuses

    Who won? Who knows????
'''

from iowa import get_winner


REPUBS = ["trump.97.1", "walsh.1.13", "weld.1.3"]

DEMS = ["biden.15.6", "buttigieg.26.9", "klobuchar.12.62", "sanders.26.1",
        "steyer.0", "warren.18.3", "yang.1.11"]

def main():

    while True:
        choice = int(input("Enter 1 for repubs or 2 "
                           "for dems or 3 to quit\n"))
        if choice == 1:
            lst = REPUBS
        elif choice == 2:
            lst = DEMS
        else:
            break
                  
        winner = get_winner(lst)
        print("Who won the iowa cacauses?")
        print(winner, "won!!")
        
main()