用python编写购物程序(2)

要求：

1. 启动程序后，让用户输入工资，然后打印商品列表
2. 允许用户根据商品编号购买商品
3. 用户选择商品后，检测余额是否充足，够就直接扣款，不够就提醒
4. 可随时推出，退出时打印以购买商品，购买商品数量及余额

代码：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:James Tao

list_of_goods=[
    ['iphone',5800],
    ['Mac Pro',12000],
    ['Bike',800],
    ['Watch',2000],
    ['Coffee',31],
    ['Book',120]
]
list_of_bought=[]
dict_of_bought={}
salary=input('请输入您的工资:')
if salary.isdigit():#判断是否是整数
    salary=int(salary)
    while True:

        #输出商品及其编号
        for index,item in enumerate(list_of_goods):#enumerate取出下标
            print(index,item)
            #print(list_of_goods.index(item),item)
        choice_of_user=input('选择购买商品编号:')

        #判断输入是否合法
        if choice_of_user.isdigit():
            choice_of_user=int(choice_of_user)

            #判断编号是否有对应商品
            if 0<=choice_of_user<len(list_of_goods):

                #判断余额是否足够买此商品
                if list_of_goods[choice_of_user][1]<=salary:

                    #加入购物清单
                    list_of_bought.append(list_of_goods[choice_of_user][0])

                    #计算余额`
                    salary-=list_of_goods[choice_of_user][1]

                    print('''添加{boughtgood}到您的购物车,此刻您的余额为{balance}.
                    '''.format(boughtgood=list_of_goods[choice_of_user][0],balance=salary))
                else:
                    print('您的余额不足,此实余额为%s,不够购买此商品',salary)
            else:
                print('商品不存在')

        elif choice_of_user=='q':

            #统计购买的商品及数量
            category_of_bought=set(list_of_bought)
            for item in category_of_bought:
                dict_of_bought[item]=list_of_bought.count(item)
            print('您购买的商品及数量分别为',dict_of_bought)

            print('您的余额为:',salary)
            exit()
        else:
            print('输入不合法')
else:
    print('输入不合法')