python入门 之购物车程序

# 1.启动程序后，让用户输入工资，然后打印商品列表

# 2.允许用户根据商品编号购买商品

# 3.用户选择商品后，检查余额是否足够，够就直接扣款，不够就提醒

# 4.可随时退出，退出时，打印已购买商品和余额

product_list = [
    ('iphone', 5800),
    ('mac pro', 9800),
    ('bike', 800),
    ('watch', 10600),
    ('coffee', 31),
    ('book', 120),
]
shopping_list = []
salary = input("Input your salary:")
if salary.isdigit():
    salary = int(salary)
    while True:
        for index, itme in enumerate(product_list):
            print(index, itme)
        user_choice = input("选择要购买的商品：>>>")
        if user_choice.isdigit():
            user_choice = int(user_choice)
            if len(product_list) > user_choice >= 0:
                p_itme = product_list[user_choice]
                if p_itme[1] <= salary:  # 钱够用
                    shopping_list.append(p_itme)
                    salary -= p_itme[1]
                    print("Added \033[31;1m%s\033[0m into shopping cart, your current balance is \033[31;1m%s\033[0m" % (p_itme, salary))
                else:
                    print("\033[41;1m你的余额只剩[%s]啦，还不买了]\033[0m" % salary)
            else:
                print("produt code [%s] is not exist!" % user_choice)

        elif user_choice == 'q':
            print("--------shopping list--------")
            for p in shopping_list:
                print(p)
            print("Your current balance:", salary)
            exit()
        else:
            print("invalid potion")