Python 写一段仓库管理的代码

Python 可以编写一个简单的仓库管理系统，这个系统可以用字典或者列表来模拟库存。以下是一个基本的示例，展示如何创建商品及其库存，并进行增删查的操作：

class Warehouse:
    def __init__(self):
        self.stock = {}  # 商品ID作为键，库存数量作为值

    def add_item(self, item_id, quantity):
        """
        添加商品到仓库
        :param item_id: 商品ID
        :param quantity: 库存量
        """
        if item_id not in self.stock:
            self.stock[item_id] = quantity
        else:
            print(f"商品 {item_id} 已存在，无法添加更多")

    def remove_item(self, item_id, quantity=1):
        """
        减少指定商品的库存
        :param item_id: 商品ID
        :param quantity: 减少的数量，默认为1
        """
        if item_id in self.stock and self.stock[item_id] >= quantity:
            self.stock[item_id] -= quantity
        else:
            print("商品不存在或库存不足")

    def check_stock(self, item_id):
        """
        检查指定商品的库存情况
        :param item_id: 商品ID
        """
        return f"商品 {item_id} 的库存为 {self.stock.get(item_id, '无')}"


# 使用示例
warehouse = Warehouse()
warehouse.add_item('001', 100)
print(warehouse.check_stock('001'))  # 输出：商品 001 的库存为 100
warehouse.remove_item('001', 50)
print(warehouse.check_stock('001'))  # 输出：商品 001 的库存为 50

这只是一个基础版本，实际的仓库管理系统可能会更复杂，包含商品类别、供应商信息、历史记录等功能，并可能利用数据库存储数据。