如何根据两个菜单选项计算

Question

指令

应该有一个图像，但我不确定您是否可以看到它？

我必须向用户提供菜单，以确定哪种式卡车，然后再提供另一个菜单，以便他们每天或每天选择价格。然后，我必须得到卡车被乘坐多少英里的几天/周。所有这些信息都必须根据两个菜单选择及其输入来计算。How do do say如果A和1是否等于...。我知道如何使用一个变量的语句使用IF语句，但我不知道如何合并所有这些输入以有效计算。请指教。以下是我到目前为止的编码。

def main():
    print('Please select from the following rental truck options')
    firstMenu()
    print("Please Tell us how many Days/Weeks you have rented our truck and how many miles were driven")
    milesTime()
def firstMenu():
    menu = input("Please bhoose from the following options(A, B, C, D)and press enter:n"
                 "A = Class A Truckn"
                 "B = Class B Truckn"
                 "C = Class C Truck")
     if menu == str('A'):
         secondMenu()
     elif menu == str('B'):
         secondMenu()
     elif menu == str('C'):
         secondMenu()
    firstMenu()
def secondMenu():
    print('Please enter code for Daily or Weekly Rate')
    print("1 = Daily")
    print("2 = Weekly")
    option2 = input()
    secondMenu()
def milesTime():
    miles = input("Please Enter number of Miles Driven")
    time = input("Please Enter the number of days/weeks the truck was rented")
    milesTime()
def calculate():
    while firstMenu == 'A':

Answer 1

def main():
    while True:
        menu()
        print('*' * 80)
def menu():
    class_list = ['A', 'B', 'C']
    user_class = input("Please bhoose from the following options(A, B, C, D)and press enter:n"
                 "A = Class A Truckn"
                 "B = Class B Truckn"
                 "C = Class C Truckn")
    if user_class not in class_list:
        raise ValueError('invalid classification!')
    else:
      user_class = class_list.index(user_class)
    print('Please enter code for Daily or Weekly Rate')
    print("1 = Daily")
    print("2 = Weekly")
    rent_type = input()
    if rent_type not in ['1', '2']:
        raise ValueError('invalid rental type')
    miles_time(user_class, rent_type)
def miles_time(user_class, rent_type):
    data = [
      [19.95, 0.59, 125.68],
      [29.95, 0.79, 188.68],
      [39.95, 0.85, 251.68]
    ]
    miles = int(input("Please Enter number of Miles Drivenn"))
    time = int(input("Please Enter the number of days/weeks the truck was rentedn"))
    user_data = data[user_class]
    user_rental = 0
    if rent_type == '1':
        user_rental = user_data[0] * time + miles * user_data[1]
    else:
        if miles <= time * 200:
            user_rental = user_data[2] * time
        else:
            user_rental = user_data[2] * time + (miles - 200) * user_data[1]
    print(user_rental)

if __name__ == '__main__':
    main()

未完全测试。但是两个给定的测试用例。