import random
import os
import options

def cls():
    """Clears screen"""
    os.system('clear')


def generate_random_capital():
    """Generates capital to guess"""
    with open('hangman.txt') as cities:
        countries_and_capitals = cities.readlines()
        countries_and_capitals = [x.strip() for x in countries_and_capitals]
        random_temp = random.choice(countries_and_capitals)
        capital = random_temp[random_temp.index('|') + 2:]
        return capital


def question():
    return input('What would you like to guess: 1.Letter 2.Word 3.I need a hint!\n')


def win(word, dashed_word):
    """Win conditions"""
    if word == dashed_word:
        print ('win')


while True:

    city = generate_random_capital().upper()
    good_letters = []
    bad_letters = []
    lifes = 5

    for letter in city:
        good_letters.append(letter.upper())
    print(city)
    print(good_letters)

    dash_letters = []

    for element in good_letters:
        if element == ' ':
            dash_letters.append(' ')
        else:
            dash_letters.append('_')

    while True: #main function

        print(*dash_letters)
        print(*bad_letters)
        print ('Remaining lifes: ' + str(lifes))

        if question() == '1':
            answer = input('Guess a letter: ').upper()
            dash_letters = options.check_letter(answer, good_letters, dash_letters)
            bad_letters = options.check_wrong_letter(answer, good_letters, bad_letters)
            lifes = options.life_change(answer, good_letters, lifes)
            win(good_letters, dash_letters)
        elif question() == '2':
            answer = input('give answer ').upper()
            if answer == city:
                print ('win')
                break
            else:
                 lifes -= 2