import pandas # Functions go here def int_check(question): """ Check users enter an integer""" error = "oops - please enter an integer." while True: try: # Return the response if its an integer response = int(input(question)) return response except ValueError: print(error) def not_blank(question): """Checks that the user response is not blank""" while True: response = input(question) if response != "": return response print("Sorry this cant be blank. Please try again") def string_check(question, valid_answers=('yes', 'no'), num_letters=1): """Checks that users enter the full word or the n letters of a word from a range 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 letters elif response == item[:num_letters]: return item print(f"Please choose an option from {valid_answers}") # currency formatting function def currency(x): """Formats numbers as currency ($#.##)""" return "${:.2f}".format(x) # Main routine goes here # 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 # list 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 } # Get ticket details while True: print() # ask the user for their name (and check it's not blank) name = not_blank("Name: ") if name == "xxx": break # ask the user for their age and check it's between 12 and 120 age = int_check("Age: ") # Out error message / success message if age < 12: print(f"{name} is too young") continue # child ticket price is 7.50 elif age < 16: ticket_price =CHILD_PRICE #adult ticket price is 10.50 elif age < 121: ticket_price = ADULT_PRICE # senior price is 6.50 elif age < 121: ticket_price = SENIOR_PRICE else: print(f"{name} is too old") #ask user for payment method ( cash / credit / ca / cr) pay_method = string_check("Payment method: ", payment_ans, 2) if pay_method == "cash": surcharge = 0 # if paying by credit calculate surcharge else: surcharge = ticket_price * CREDIT_SURCHARGE # add name ticket cost and surcharge to 'all_lists' all_names.append(name) all_ticket_costs.append(ticket_price) all_surcharges.append(surcharge) # create dataframe / table from dictionary mini_movie_frame = pandas.DataFrame(mini_movie_dict) # calculate the total payable & profit 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 and paid profits total_paid = mini_movie_frame['Total'].sum() total_profit = mini_movie_frame['Profit'].sum() # currency formatting ( uses currency function add_dollars = ['Total', 'Surcharge', 'Ticket Price', 'Profit'] for var_item in add_dollars: mini_movie_frame[var_item] = mini_movie_frame[var_item].apply(currency) # output movie frame without index print(mini_movie_frame.to_string(index=False)) print() print(f"Total Paid: ${total_paid:.2f}") print(f"Total Profit: ${total_profit:.2f}")