第一个pyhton购物车小程序

iphoneX = 8000
mac = 4000
book = 20
bicyle = 400
cat = []
while True:
    print("欢迎来到超市")
    function = '''
    1. 买商品
    2. 查看购物车
    3. 退出
    '''
    print(function)
    function = int(input("请选择功能:"))
    if function == 1:
        salary = int(input("请输入你的工资:"))
        while True:
            msg = '''
            1. iphoneX      %s
            2. mac book     %s
            3. book         %s
            4. bicyle       %s
            5. 退出
            '''%(iphoneX,mac,book,bicyle)
            print(msg)
            num = int(input("请选择你的商品:"))
            if num == 1:
                if salary >= iphoneX:
                    salary = salary - iphoneX
                    cat.append('iphoneX')
                else:
                    print("您的余额不足")
                print("您的余额为:",salary)
                input("")
                continue
            elif num == 2:
                if salary >= mac:
                    salary = salary - mac
                    cat.append('mac book')
                else:
                    print("您的余额不足")
                print("您的余额为:",salary)
                input("")
                continue
            elif num ==3:
                if salary >= book:
                    salary = salary - book
                    cat.append('book')
                else:
                     print("您的余额不足")
                print("您的余额为:",salary)
                input("")
                continue
            elif num == 4:
                if salary >= bicyle:
                    salary = salary - bicyle
                    cat.append('bicyle')
                else:
                    print("您的余额不足")
                print("您的余额为:",salary)
                input("")
                continue
            elif num == 5:
                break
            else:
                print("请输入正确的选项")
    elif function == 2:
        print(cat)
        input("按任意键退出")
        continue
    elif function == 3:
        break