while True: try: display_output('MENU') x = input('Please input command!') if x == '1': display_output('Login') u_input = input('Input customer username! ') for i in data_all: if u_input == i[0]: while True: z = Customer(i[1][0], i[1][1],0,0) display_output('Customer Menu') d = input('Please Input command!') if d == '1': print('First Name : ', z.getFirstName()) print('Last Name : ', z.getLastName()) print('All Account') display_account(u_input,0) elif d == '2': display_output('Add Account') x = input('Account Name: ') print('Your Account Name is : {0}'.format(z.newAccount(x,i[-1]))) elif d == '3': display_output('Display All Account') display_account(u_input,1) elif d == '4': a_input = int(input('Which Account to Login?')) while True: display_output('Account Menu') x = input('Please input command!') z = Customer(i[1][0],i[1][1],i[-1][a_input-1][0],i[-1][a_input-1][1])
if action not in ["add", "del", "show"]: print("Invalid input.") continue elif action == "add": firstName = input("Please provide the first name of the customer.\n") lastName = input("Please provide the last name of the customer.\n") address = input("Please provide the address of the customer.\n") customer = Customer(firstName, lastName) customer.setAddress(address) conn = sqlite3.connect('customers.db') c = conn.cursor() params = (customer.getFirstName(), customer.getLastName(), customer.getAddress()) c.execute("INSERT INTO customers VALUES (NULL, ?, ?, ?)", params) conn.commit() conn.close() print("Done.\n") elif action == "del": count = getRowCount() if count == 0: print("There are no customers in the list.") else: print("These are your current customers:")