import pandas


def string_check(question, valid_list=('yes', 'no'), num_letters=1):
    """Checks that users enter the full word or the first 'n' letters from a list"""
    while True:
        response = input(question).strip().lower()

        for item in valid_list:
            item_low = item.lower()
            if response == item_low or response == item_low[:num_letters]:
                return item

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


def instructions():
    print("Pizza Paladin Instructions ℹ️")
    print("""

The program will record the users data that they entered,
then proceed to the main menu of what Pizza Paladin offers.

The program proceeds by asking would you like to pick-up or
deliver the pizza to your address. You enter 'Pick-up' or 'Deliver'

Once you have ordered your pizza (up to 5 maximum allowed), 
or entered the exit code (‘xxx’), the program will display the receipt
information regarding the customers order and information, 
writing the data to a text file.

For each customer should enter ... 
------ (IF PICK-UP WAS CHOSEN) ------
- Their name
- Their age
- Phone number
- The payment method (cash / credit)

------ (IF DELIVERY WAS CHOSEN) ------
- Their name
- Their age
- Phone number
- Address
- The payment method (cash / credit)

The program will record the users data that they entered,
then proceed to the order by printing a receipt 

    """)


def make_statement(statement, decoration, lines=1):
    middle = f"{decoration * 3} {statement} {decoration * 3}"
    top_bottom = decoration * len(middle)
    if lines == 1:
        print(middle)
    elif lines == 2:
        print(middle)
        print(top_bottom)
    else:
        print(top_bottom)
        print(middle)
        print(top_bottom)


def not_blank(question):
    """Checks that a user response is not blank"""
    while True:
        response = input(question).strip()
        if response != "":
            return response
        print("Sorry, this can't be blank. Please try again.\n")


def int_check(question):
    """Checks users enter an integer that is more than zero (or the 'xxx' exit code)"""
    error = "Oops - please enter an integer."
    while True:
        response = input(question).lower()
        if response == "xxx":
            return response
        try:
            response = int(response)
            if response > 0:
                return response
            else:
                print(error)
        except ValueError:
            print(error)


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


# Variables
MAX_PIZZAS = 5
pizzas_sold = 0
payment_ans = ('cash', 'credit')
yes_no_list = ('yes', 'no')
CREDIT_SURCHARGE = 0.025
DELIVERY_FEE = 10

pizza_list = ["Pepperoni", "Hawaiian", "Margherita", "BBQ Chicken",
              "Meat Lovers", "Cheese Pizza", "Veggie Pizza",
              "Kebab Pizza", "Spaghetti Pizza", "Smash Burger"]

pizza_sizes = ["Small", "Medium", "Large"]
pizza_sizes_cost = [2.70, 3.50, 4.20]

special_paladin_pizzas = ["Sushi Pizza", "S'more Pizza"]
extra_toppings = ["Extra cheese", "Mushrooms", "Onions", "Pineapple", "Green Peppers"]

pizza_cost = [8.30] * len(pizza_list)
special_prices = [6.70, 6.40]
topping_prices = [1.80, 1.20, 1.20, 1.30, 1.10]

# Main routine starts here

make_statement(statement="Welcome To Pizza Paladin!", decoration="🍕")
print()
want_instructions = string_check("Do you want to see the instructions? ", yes_no_list)
if want_instructions == "yes":
    instructions()
print()

# Customer name
name = not_blank("Name (or 'xxx' to quit): ")
if name.lower() == "xxx":
    print("Order cancelled. Hope you come back and order a pizza from Pizza Paladin!")
    exit()
print()
# Phone Number
phone_number = not_blank("What is your phone number? ")

print()
# Ask Pick up or Delivery (Delivery fee was added
# we add the question before the payments)
delivery_choice = None
address = ""

while True:
    pickup_or_delivery = string_check("Would you like delivery or pick up? ", ("delivery", "pick up"), 3)
    if pickup_or_delivery == "delivery":
        confirm_delivery = string_check("There is a $10 delivery fee. Would you like to proceed? ", yes_no_list)
        if confirm_delivery == "yes":
            delivery_choice = "delivery"
            address = not_blank("Please enter your address: ")
            break
    else:
        delivery_choice = "pick up"
        break

print()

# Main menu
make_statement(statement="Pizza Paladin Main Menu", decoration="🍕")
print("\n--- Gourmet Pizzas ---")
for i, pizza in enumerate(pizza_list, start=1):
    print(f"{i}. {pizza} - {currency(pizza_cost[i - 1])}")

print("\n--- Pizza Sizes ---")
for i, size in enumerate(pizza_sizes, start=len(pizza_list) + 1):
    print(f"{i}. {size} - {currency(pizza_sizes_cost[i - len(pizza_list) - 1])}")

print("\n--- Special Pizzas ---")
for i, special in enumerate(special_paladin_pizzas, start=len(pizza_list) + len(pizza_sizes) + 1):
    print(f"{i}. {special} - {currency(special_prices[i - (len(pizza_list) + len(pizza_sizes) + 1)])}")

print("\n--- Extra Toppings ---")
for i, topping in enumerate(extra_toppings, start=len(pizza_list) + len(pizza_sizes) + len(special_paladin_pizzas) + 1):
    print(
        f"{i}. {topping} - {currency(topping_prices[i - (len(pizza_list) + len(pizza_sizes) + len(special_paladin_pizzas) + 1)])}")

print()

# Pizza order tracking
order_summary = []
while pizzas_sold < MAX_PIZZAS:
    choose_pizza_num = int_check(f"Choose a gourmet pizza number (1-{len(pizza_list)}) or type 'xxx' to quit: ")
    if choose_pizza_num == "xxx":
        break

    print()

    if 1 <= choose_pizza_num <= len(pizza_list):
        pizza_name = pizza_list[choose_pizza_num - 1]
        pizza_price = pizza_cost[choose_pizza_num - 1]
        print(f"You chose {pizza_name}, this costs {currency(pizza_price)}")
    else:
        print("Invalid pizza number.")
        continue

    print()

    # Choose pizza size
    size_num = int_check(f"Choose pizza size number ({len(pizza_list) + 1}-{len(pizza_list) + len(pizza_sizes)}): ")
    if isinstance(size_num, str) and size_num.lower() == "xxx":
        break
    if len(pizza_list) + 1 <= size_num <= len(pizza_list) + len(pizza_sizes):
        size_index = size_num - len(pizza_list) - 1
        size_name = pizza_sizes[size_index]
        size_price = pizza_sizes_cost[size_index]
        print(f"You chose {size_name}, this costs {currency(size_price)}")
    else:
        print("Invalid size number.")
        continue

    print()

    pizza_total = pizza_price + size_price
    order_summary.append((f"{pizza_name} ({size_name})", pizza_total))

    # Extra toppings
    extra_toppings_pizza = string_check("Would you like extra toppings? ", yes_no_list)
    if extra_toppings_pizza == "yes":
        topping_num = int_check(
            f"Choose topping number ({len(pizza_list) + len(pizza_sizes) + len(special_paladin_pizzas) + 1}-{len(pizza_list) + len(pizza_sizes) + len(special_paladin_pizzas) + len(extra_toppings)}): ")
        topping_index = topping_num - (len(pizza_list) + len(pizza_sizes) + len(special_paladin_pizzas))
        topping_name = extra_toppings[topping_index - 1]
        topping_price = topping_prices[topping_index - 1]
        print(f"You chose {topping_name}, this costs {currency(topping_price)}")
        order_summary.append((f"Extra: {topping_name}", topping_price))

    print()

    # Special pizzas
    want_special_pizzas = string_check("Would you like a special pizza? ", yes_no_list)
    if want_special_pizzas == "yes":
        special_num = int_check(
            f"Choose special pizza number ({len(pizza_list) + len(pizza_sizes) + 1}-{len(pizza_list) + len(pizza_sizes) + len(special_paladin_pizzas)}): ")
        special_index = special_num - (len(pizza_list) + len(pizza_sizes))
        special_name = special_paladin_pizzas[special_index - 1]
        special_price = special_prices[special_index - 1]
        print(f"You chose {special_name}, this costs {currency(special_price)}")
        order_summary.append((f"Special: {special_name}", special_price))

    pizzas_sold += 1
    if pizzas_sold >= MAX_PIZZAS:
        print(f"You've reached the maximum of {MAX_PIZZAS} pizzas.")
        break

    print()

    another_pizza = string_check("Would you like another pizza? ", yes_no_list)
    if another_pizza == "no":
        break

print()
# payment method
payment_method = string_check("Please choose a payment: ", payment_ans, 1)
print(f"You chose {payment_method}")
print()

# Prints out the Receipt for the customer
make_statement(f"{name}'s Order Receipt", decoration="📜")
subtotal = 0
for item, price in order_summary:
    print(f"{item:<30} {currency(price)}")
    subtotal += price

# delivery fee
if delivery_choice == "delivery":
    print(f"{'Delivery Fee':<30} {currency(DELIVERY_FEE)}")
    subtotal += DELIVERY_FEE

# Credit surcharge
surcharge = 0
if payment_method == "credit":
    surcharge = subtotal * CREDIT_SURCHARGE
    print(f"{'Credit Card Surcharge (2.5%)':<30} {currency(surcharge)}")

total = subtotal + surcharge
print(f"{'Subtotal':<30} {currency(subtotal)}")
print(f"{'Total':<30} {currency(total)}")
print("------------------------------------")

print()
# create a file to hold in the data
file_name = "Pizza_data_experiment"
write_to = "{}.txt".format(file_name)

text_file = open(write_to, "w+")

# strings to write to file...
heading = "=== Pizza Test ===\n"
user_information = f"{name}, {payment_method}, {phone_number},"
user_order = ""
pickup_or_delivery = f"{address}"

# list of strings
to_write = [heading, user_information, pickup_or_delivery]

# print the output
for item in to_write:
    print(item)

# write the item to file
for item in to_write:
    text_file.write(item)
    text_file.write("\n")

print()
make_statement("Thank you for ordering at Pizza Paladin", decoration="🍕")
print("We hope you come back again!!")