帮我写一个仓库管理系统Python代码

好的，以下是一个简单的仓库管理系统的Python代码，您可以根据需要进行修改和优化：

# 定义产品类
class Product:
    def __init__(self, id, name, quantity):
        self.id = id
        self.name = name
        self.quantity = quantity

# 定义仓库类
class Warehouse:
    def __init__(self):
        self.products = []

    # 添加产品
    def add_product(self, product):
        self.products.append(product)

    # 删除产品
    def remove_product(self, product_id):
        for product in self.products:
            if product.id == product_id:
                self.products.remove(product)

    # 更新产品数量
    def update_quantity(self, product_id, quantity):
        for product in self.products:
            if product.id == product_id:
                product.quantity = quantity

    # 查找产品
    def find_product(self, product_id):
        for product in self.products:
            if product.id == product_id:
                return product
        return None

    # 显示所有产品
    def show_products(self):
        for product in self.products:
            print(f"ID: {product.id}, Name: {product.name}, Quantity: {product.quantity}")

# 主程序
if __name__ == "__main__":
    warehouse = Warehouse()

    # 添加产品
    product1 = Product(1, "iPhone", 100)
    product2 = Product(2, "iPad", 50)
    warehouse.add_product(product1)
    warehouse.add_product(product2)

    # 显示所有产品
    warehouse.show_products()

    # 更新产品数量
    warehouse.update_quantity(1, 50)

    # 删除产品
    warehouse.remove_product(2)

    # 查找产品
    product = warehouse.find_product(1)
    if product:
        print(f"ID: {product.id}, Name: {product.name}, Quantity: {product.quantity}")
    else:
        print("Product not found.")

该代码实现了一个简单的仓库管理系统，包括添加产品、删除产品、更新产品数量、查找产品和显示所有产品等功能。您可以根据实际需求进行修改和扩展。