用python帮我写一个仓库管理系统

好的，以下是一个简单的仓库管理系统的Python代码示例：

class InventoryItem:
    def __init__(self, name, quantity):
        self.name = name
        self.quantity = quantity

class Inventory:
    def __init__(self):
        self.items = []

    def add_item(self, name, quantity):
        for item in self.items:
            if item.name == name:
                item.quantity += quantity
                break
        else:
            new_item = InventoryItem(name, quantity)
            self.items.append(new_item)

    def remove_item(self, name, quantity):
        for item in self.items:
            if item.name == name:
                if item.quantity >= quantity:
                    item.quantity -= quantity
                    break
                else:
                    print("Not enough quantity to remove.")
                    break
        else:
            print("Item not found.")

    def display_inventory(self):
        print("Inventory:")
        for item in self.items:
            print(f"{item.name}: {item.quantity}")

# Example usage
inventory = Inventory()

inventory.add_item("Apple", 5)
inventory.add_item("Banana", 10)
inventory.add_item("Apple", 3)

inventory.display_inventory()

inventory.remove_item("Apple", 2)
inventory.remove_item("Grape", 1)

inventory.display_inventory()

该系统由两个类组成：`InventoryItem`表示一个库存条目，其中包含名称和数量属性，`Inventory`表示整个库存，其中包含一个包含所有库存条目的列表。`Inventory`类有三个主要方法：`add_item`用于添加新的库存项，如果该项已存在，则增加其数量；`remove_item`用于从库存中删除指定数量的库存项；`display_inventory`用于显示整个库存清单。

在这个简单的例子中，我们首先创建一个`Inventory`对象，然后添加三个库存项。接下来，我们显示整个库存清单，然后尝试删除两个苹果和一个葡萄。最后，我们再次显示整个库存清单以查看删除操作的结果。