python - work5 - 类与对象 - 拓展题

'''

5：购物车类，包含的功能如下，请自行设计这个类以及类里面的方法:
1)用户输入工资后，打印商品列表（商品列表自行设计展示模式）
2）允许用户根据商品编号去选择商品
3)用户选择商品后，检查余额是否足够，够的话直接扣款，不够的话就对用户做出提醒
4）用户可以随时退出，退出时打印用户的购买商品以及余额
'''

class ShoppingTrolley:

    commodity_list = {'1':{'computer':'8000'},'2':{'phone':'6000'},'3':{'camera':'4000'},'4':{'keyboard':'2000'},'5':{'t-shirt':'500'},'6':{'chocolate':'100'},'7':{'banana':'20'}}
    salary = input('请输入您的工资：')

    # 打印商品列表
    def print_commodity(self):
        com_list = self.commodity_list
        for key1 in com_list.keys():
            for key2 in com_list[key1]:
                print('''
商品编号：{0}
商品名称：{1}
商品价格：{2}
                 '''.format(key1,key2,com_list[key1][key2]) )

        return com_list

    # 选择商品
    def choose_commodity(self):
        while True:
            com_id = input('请输入商品编号库；,若退出请输入q：')
            if com_id in self.commodity_list.keys() or com_id == 'q':
                break
            else:
                print('您输入的商品编号不存在！')
        return com_id

    # 计算金额
    def charge(self):
        com_list = self.print_commodity()
        money_total = int(self.salary)
        com_buy_list = []
        while True:
            com_id = self.choose_commodity() # 获取要购买商品编号
            if com_id == 'q': # 若获取的商品编号为q，退出循环
                break
            # 根据获取的商品编号，取出对应的商品名称和商品价格
            for key,value in com_list[com_id].items():
                com_name = key
                com_price = int(value)

            if money_total - com_price >= 0: # 计算当前余额是否足够购买当前商品
                money_total -= com_price
                print('你购买的商品是{0}，花费{1}元，扣除费用后余额{2}元'.format(com_name,com_price,money_total))
                com_buy_list.append(com_name) # 每次购买成功后，记录购买的商品
            else: # 余额不足时，退出循环
                print('你购买的商品是{0}，需花费{1}元，余额不足！'.format(com_name,com_price))
                break
        print('您购买的商品为：{0},余额{1}元'.format('、'.join(com_buy_list),money_total))

s1 = ShoppingTrolley()
s1.charge(