使用python3简单完成购物过程

功能：
要求用户输入购物卡余额，打印购物清单
用户可以连续的购买商品，直至钱不够为止
退出时，打印用户已经购买的商品信息和剩余金额

本次技术要求：使用list, dict, function，class。不能使用数据库，下一个版本会使用Django完成类似功能

代码如下：

import sys
import copy


class Product:
    def __init__(self, pid, name, price, count):
        self.pid = pid
        self.name = name
        self.price = price
        self.count = count

    def __str__(self):
        return "id>>" + str(self.pid) + "  \tname>>" + self.name+"  \tprice>>" + str(self.price) + " \tcount>>" + str(self.count)


class ProductList:
    # 输入列表，抓成以id为key的字典，便于以后查找
    def __init__(self, products=[]):
        self.prod_list = dict()
        for product in products:
            self.prod_list[product.pid] = product

    def minus(self, productId, num):
        if productId in self.prod_list:
            if self.prod_list[productId].count > num:
                self.prod_list[productId].count -= num
                return True
            elif self.prod_list[productId].count == num:
                self.prod_list.pop(productId)
                return True
            else:
                print("库存不够，请重新选择数量")
                return False
        else:
            print("本仓库没有此件商品，请重新选择")

    def add(self, product, num=1):
        if product.pid in self.prod_list:
            self.prod_list[product.pid].count += num
        else:
            new_product = copy.deepcopy(product)
            new_product.count = num
            self.prod_list[new_product.pid] = new_product

    def print_all_product(self):
        for key in self.prod_list:
            print(self.prod_list[key])


class Shopping:
    @classmethod
    def get_salary(cls):
        salary = input("请输入您购买卡的余额：")
        if not salary.isdecimal():
            print("您输入的卡的余额不是数值型，请重新输入 ")
            return Shopping.get_salary()
        else:
            return int(salary)

    def __init__(self, sell_product=[]):
        self.salary = Shopping.get_salary()
        self.sell_product = ProductList(sell_product)
        self.buy_product = ProductList()

    def buy_some_product(self):
        self.sell_product.print_all_product()
        select_product = input("请输入您的产品编码： ")
        product_num = input("请输入商品的数量： ")
        if product_num.isdigit():
            product_num = int(product_num)
        else:
            print("数量必须是数字！")
            self.buy_some_product()
        # 如果够支付，则工资减少，同时buy_product增加一个商品
        if self.is_afford(select_product, product_num):
            self.buy_product.add(self.sell_product.prod_list[select_product], product_num)
            self.salary -= self.sell_product.prod_list[select_product].price*product_num
            self.sell_product.minus(select_product, product_num)
            print("您当前购买商品如下：")
            self.buy_product.print_all_product()
            print("您的余额是 %s 元" % self.salary)
        else:
            print("您的余额不足，请重新选择")

    # 判断当前的工资余额是否够支付产品
    def is_afford(self, procudtId, product_num):
        if type(procudtId) == int:
            procudtId = str(procudtId)
        if self.salary >= self.sell_product.prod_list[procudtId].price * product_num:
            return True
        else:
            return False

    def select_process(self):
        while True:
            selector = input("""
                请输入您的选择？
                    购买(b)
                    退出(q)
                """)
            if selector == 'b':
                self.buy_some_product()
            else:
                print("您购买了如下商品")
                self.buy_product.print_all_product()
                print("您的余额是 %s 元" % self.salary)
                print("欢迎您下次光临")
                sys.exit(0)



if __name__ == '__main__':
    # 测试初始化产品列表的信息的功能
    init_product=[Product('1', 'bike', 100, 10), Product('2', 'ipad', 3000, 3), Product('3', 'paper', 500, 30)]
    shop = Shopping(init_product)
    shop.select_process()

 

类product：产品的基本信息

ProductList：以产品pid为key来存储许多产品，同时提供增加、减少产品数量的功能

Shopping：主要完成购物功能。初始化销售产品列表和购物卡余额。循环购物的过程。

 

注意：

ProductList.add(self, product, num=1)一定不能使用输入的product的信息，要保证各个列表中的product的信息是独立（产品中的数量是独立的）。

 

说明：这个小功能仅仅是练习。距离产品还有很大的差距。