python简易购物车练习

只是一个 随手小练习，对功能的简单实现。

"""
需求;
    1.启动程序后， 让用户输入工资，然后打印商品列表
    2.允许用户根据商品编号购买商品
    3.用户选择商品后，检测余额是否够，够就直接扣款，不够就提醒
    4.可随时退出，退出时，打印已购买的商品和余额
"""

product_list = [
    ("Iphone", 5000),
    ("bike", 500),
    ("iwatch", 4000),
    ("book", 30),
    ("cloth", 100)
]
salary = input("please input your salary:")
shopping_list = []
if salary.isdigit():
    salary = int(salary)
    while True :
        for index, item in enumerate (product_list):
            print(index, item)
        user_choice = input("你要买什么？")
        if user_choice.isdigit() :
            user_choice = int(user_choice)
            if user_choice < len(product_list) and user_choice >= 0:
                p_item = product_list[user_choice]
                if p_item[1] < salary : # 买得起
                    shopping_list.append(p_item)
                    salary -= p_item[1]
                    print("Added %s into shopping cart, your current balance is\033[31;1m %s \033[0m" %(p_item, salary))
                else:
                    print("\033[41;1m你的余额只剩[%s]啦，买毛毛虫啊\033[0m")
            else:
                print("product 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 option")