购物车程序

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Author:Huan

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

product_list = [
    ('Iphone',5800),
    ('Mac Pro',9800),
    ('Bike', 800),
    ('Watch', 106000),
    ('Coffee', 31),
    ('Python', 120),
]
shopping_list = []
salary = input("input your salary:")
if salary.isdigit():
    salary = int(salary)
    while True:
        for i,k in enumerate(product_list):
            print(i,k)
        user_choice = input("select by?>>>:")
        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" % salary)
            else:
                print("product code [%s] is not exit!" % 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 opthion")
else:
    print("您输入的不是数字")