import pandas
import random

# 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_answers=('yes', 'no'),
                 num_letters=1):
    """Checks that users enter the full word
    or the first letter of a word from alist of valid responses"""

    while True:

        response = input(question).lower()

        for item in valid_answers:

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

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

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

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

    print('''

For each ticket holder enter ...
- Their name
- Their age
- The payment method (cash / credit)

The program will record the ticket sale and calculate the 
ticket cost (and the profit).

Once you have either sold all of the 
tickets or entered the exit code ('xxx'), the program will display the ticket
sales information adn writ ethe fata to a text file.

It will also choose one lucky ticket holder who wins the 
draw (their ticket is free)

''')

def not_blank(question):
    """Check 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.\n")

def int_check(question):
    """Checks users enter an integer"""

    error = "Oops - please enter an integer"

    while True:

       try:
           # Return the response if it's an integer
           response = int(input(question))

           return response

       except ValueError:
           print(error)

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

# Main Routine goes here

# Initialise ticket numbers
MAX_TICKETS = 5
tickets_sold = 0

# initialise variables / non-default options for string checker
payment_ans = ('cash', 'credit')

# Ticket Price List
CHILD_PRICE = 7.50
ADULT_PRICE = 10.50
SENIOR_PRICE = 6.50

# Credit card surcharge (currently 5%)
CREDIT_SURCHARGE = 0.05

# Lists to hold ticket details
all_names = []
all_ticket_costs = []
all_surcharges = []

mini_movie_dict = {
    'Name': all_names,
    'Ticket Price': all_ticket_costs,
    'Surcharge': all_surcharges,
}



# Program main heading
print(make_statement("Mini-Movie Fundraiser Program", "🍿"))

# Ask user if they want to see the instructions and\
# display them if necessary
print()
want_instructions = string_check("Do you want to see the instructions? ")

if want_instructions == "yes":
    instructions()

print()

while tickets_sold < MAX_TICKETS:
    # ask user for their name (and check it's not blank)
    print()
    name = not_blank("Name: ")

    # if name is exit code, break out of loop
    if name == "xxx":
        break

    if name == "XXX":
        break

    # Ask for their age and check it's between 12 and 120
    age = int_check("Age: ")

    # Output error message / success message
    if age < 12:
        print("Sorry you are too young for this movie")
        continue

    # Child ticket is $7.50
    elif age < 16:
        ticket_price = CHILD_PRICE

    # Adult ticket is ($10.50)
    elif age < 65:
        ticket_price = ADULT_PRICE

    # Senior Citizen ticket ($6.50)
    elif age < 121:
        ticket_price = SENIOR_PRICE

    else:
        print(f"{name} is too old")
        continue

    # ask user for payment method (cash / credit / ca / cr)
    payment_method = string_check("Payment method: ", payment_ans, 2)
    print(f"{name} has bought a ticket ({payment_method})") \

    if payment_method == "cash":
        surcharge = 0
    else:
        surcharge = 0.5


    # add name, ticket cost and surcharge
    all_names.append(name)
    all_ticket_costs.append(ticket_price)
    all_surcharges.append(surcharge)

    tickets_sold += 1

# End of Ticket Loop!

# Create dataframe / table from dictionary
mini_movie_frame = pandas.DataFrame(mini_movie_dict)

# Calculate the total payable for each ticket
mini_movie_frame['Total'] = mini_movie_frame['Ticket Price'] + mini_movie_frame['Surcharge']
mini_movie_frame['Profit'] = mini_movie_frame['Ticket Price'] - 5

# Work out total paid and total profit...
total_paid = mini_movie_frame['Total'].sum()
total_profit = mini_movie_frame['Profit'].sum()

# choose random winner...
winner = random.choice(all_names)

# find index of winner (ie: position in list)
winner_index = all_names.index(winner)

# retrieve winner ticket price and profit (so we can adjust
# profit numbers so that the winning ticket is excluded)

ticket_won = mini_movie_frame.at[winner_index, 'Total']
profit_won = mini_movie_frame.at[winner_index, 'Profit']

# Currency Formatting (uses currency fund)
add_dollars = ['Ticket Price', 'Surcharge', 'Total', 'Profit']
for var_item in add_dollars:
    mini_movie_frame[var_item] = mini_movie_frame[var_item].apply(currency)

# Output movie from without index
mini_movie_sting = mini_movie_frame.to_string(index=False)

total_paid_string = f"Total Paid: ${total_paid:.2f}"
total_profit_string = f"Total Profit: ${total_profit:.2f}"

adjusted_explanation = (f" We have given away a ticket worth ${ticket_won:.2f} which \n"
                        f"means our sales have decreased by ${ticket_won:.2f} and our\n"
                        f"profit decreased by ${profit_won:.2f}.")
# winner announcement
lucky_winner_string = (f"The lucky winner is {winner}. "
                       f"Their ticket worth ${ticket_won:.2f} is free!")
final_total_paid_string = f"Total paid is now {total_paid - ticket_won:.2f}"
final_profit_string = f"Total paid is now {total_profit - profit_won:.2f}"

if tickets_sold == MAX_TICKETS:
    num_sold_string = make_statement(f"You have sold all the tickets "
                                     f"(ie: {MAX_TICKETS} tickets)", "-")
else:
    num_sold_string = make_statement(f"You have sold {tickets_sold} / {MAX_TICKETS} / "
                                     f"{MAX_TICKETS} tickets.", "-")

# Additional string / Headings
heading_string = make_statement("Mini Movie Fundraiser", "=")
ticket_details_heading = make_statement("Ticket Details", "-")
raffle_heading = make_statement("--- Raffle Winner ---", "-")
adjusted_sales_heading = make_statement("Adjusted Sales & Profit", "-")


adjusted_explanation = (f"We have given away a ticket worth ${ticket_won:.2f} which means \nour"
                        f"sales have decreased by ${ticket_won:.2f} and our profit \n"
                        f"decreased by ${profit_won:.2f}")


# List of strings to be outputted / written to file
to_write = [heading_string, "\n",
            ticket_details_heading,
            mini_movie_sting, "\n",
            total_paid_string,
            total_profit_string, "\n",
            raffle_heading,
            lucky_winner_string, "\n",
            adjusted_sales_heading, "\n",
            adjusted_explanation, "\n",
            final_total_paid_string, "\n",
            final_profit_string, "\n",
            num_sold_string]

# Print area
print()
for item in to_write:
    print(item)

# create file to hold data (add .txt extension)
file_name = "MMF_ticket_details"
write_to = "{}.txt".format(file_name)

text_file = open(write_to, "w+")

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