import pandas
import numpy


# Functions go here
def make_statement(statement, decoration):
    """Emphasises headings by adding decoration
    at the start and end"""

    return f"{decoration * 3} {statement} {decoration * 3}"


def string_check(question, valid_ans_list=('yes', 'no'),
                 num_letters=1):
    """Checks that users enter the full word
    or the first letter of a word from a list of valid responses"""

    while True:

        response = input(question).lower()

        for item in valid_ans_list:

            # check if the response is the entire world
            if response == item:
                return item

            # check if it's the first letter
            elif response == item[:num_letters]:
                return item

        print(f"Please choose an option from {valid_ans_list}")


def instructions():
    print(make_statement("Instructions", "ℹ️"))

    print('''
    
When you begin enter ...
- budget
- the number of the row which has the cheese you wish to purchase

The program will record the purchases and calculate the cost.

Users can buy multiple cheeses or say no to repurchasing
and the program will end

It will finally show an itemised list of all the purchases and will save it
to a writing file

    ''')


def not_blank(question):
    """Checks that a user response is not blank"""

    while True:
        response = input(question)

        if response != "":
            return response

        print("Sorry, this can't be blank. Please try again.")


def int_check(question, low, high):
    """Checks users enter an integer between two values"""

    error = f"Oops - please enter an integer between {low} and {high}."

    while True:

        try:
            # Change the response to an integer and check that it's more than zero
            response = int(input(question))

            if low <= response <= high:
                return response
            else:
                print(error)

        except ValueError:
            print(error)


def currency(x):
    """Formats numbers as currency ($#.##)"""
    return "${:.2f}".format(x)


# Main Routine goes here

# Initialise cheese numbers
MAX_SPEND = 1000
budget = 0

# lists to hold cheese details
all_cheese = ["Colby", "Mozzarella", "Cheddar", "Gouda", "Brie",
              "Manchego", "Roquefort", "Parmigiano Reggiano",
              "Époisses de Bourgogne", "Pule Cheese"]
all_cheese_costs = [20, 30, 35, 40, 60, 80, 100, 150, 200, 300]

cheese_dict = {
    'Cheese': all_cheese,
    'Cheese Price': all_cheese_costs,
}

# create dataframe / table from dictionary
cheese_frame = pandas.DataFrame(cheese_dict)

# Rearranging index
cheese_frame.index = numpy.arange(1, len(cheese_frame) + 1)

# Currency Formatting for the original cheese frame
add_dollars = ['Cheese Price']
for var_item in add_dollars:
    cheese_frame[var_item] = cheese_frame[var_item].apply(currency)

# Program main heading
print(make_statement("The Online Cheese Market", "🧀"))

# Ask user for their name (and check it's not blank)
print()
name = not_blank("Name: ")

# Ask user if they want to see the instructions
# display them if necessary
print()
want_instructions = string_check(f"Hi {name}, do you want to see the instructions? ")

if want_instructions == "yes":
    instructions()

print()

# Ask for the users budget
budget = int_check("What is your budget (Maximum budget of $1000)? ", 20, 1000)

print(f"Your budget is ${budget}")
print()

# set purchased cheeses to none
purchased_cheese = []
purchased_prices = []

# create a purchased dictionary
purchased_dict = {
    'Cheese': purchased_cheese,
    'Cheese Price': purchased_prices,
}

# import list of cheeses
cheese_string = cheese_frame.to_string(index=False)

# Loop to get name, age and payment details
while budget <= MAX_SPEND:

    print()

    # Display menu items
    print("Here is a list of Cheeses you can choose from:")
    print(cheese_frame)
    print()

    # Prompt user to select a row from the index
    try:
        # ask user for their choice
        choice = int_check("Choose your cheese with the number of the row: ", 1, 10)

        selected_row = all_cheese[choice - 1]
        cheese_price = all_cheese_costs[choice - 1]

        if budget >= cheese_price:
            budget -= cheese_price

            purchased_cheese.append(selected_row)
            purchased_prices.append(cheese_price)

            # display user choice
            print(f"You selected {selected_row}, costing ${cheese_price} ")
            print(f"Your new budget is ${budget} ")
            print()

        else:
            print(f"Sorry, the {selected_row} costing {cheese_price} "
                  f"is above your current budget of ${budget}. Please try again.")
            continue

        if budget <= 19:
            print(f"Sorry {name}, your budget is now less than the cheapest "
                  f"Cheese ($20). Thanks for shopping!")
            print()
            break

        repurchase = string_check("Would you like to purchase another Cheese? ")

        if repurchase == "no":
            break

    except ValueError:
        print("Please enter a valid number")

# End of Loop

# itemised purchase list
total_spent = sum(purchased_prices)

# create dataframe / table from dictionary
purchased_frame = pandas.DataFrame(purchased_dict)

# Rearranging index
purchased_frame.index = numpy.arange(1, len(purchased_frame) + 1)

# Currency Formatting for the purchased frame
add_dollars = ['Cheese Price']
for var_item in add_dollars:
    purchased_frame[var_item] = purchased_frame[var_item].apply(currency)

if purchased_cheese:
    print(make_statement(f"Here are {name}'s purchases:", "🛒"))
    print()
    print(purchased_frame)
    print(f"The total amount you spent was ${total_spent}")