feat(面向对象): 完成第三阶段差异化课程

03_面向对象/3_5_面向对象综合项目/domain_model_example.py

@@ -0,0 +1,82 @@
+# 第 3-5 课参考示例：小型仓库领域模型
+#
+# 本示例使用与综合项目不同的“仓库”场景，演示如何让对象各自负责自己的数据和行为。
+# 综合项目仍需独立完成，不能直接复制本示例得到答案。
+
+from dataclasses import dataclass, field
+
+
+class InventoryError(Exception):
+    """表示仓库业务规则错误。"""
+
+
+@dataclass
+class Product:
+    """保存商品数据，并负责与商品自身有关的显示。"""
+
+    code: str
+    name: str
+    quantity: int = 0
+
+    def __post_init__(self) -> None:
+        """确保创建商品时库存数量合法。"""
+        if self.quantity < 0:
+            raise InventoryError("库存数量不能小于 0。")
+
+    @property
+    def in_stock(self) -> bool:
+        """根据当前数量计算是否有库存。"""
+        return self.quantity > 0
+
+    def __str__(self) -> str:
+        """返回面向使用者的商品信息。"""
+        return f"{self.name}｜库存：{self.quantity}"
+
+
+@dataclass
+class Warehouse:
+    """通过组合管理多个 Product 对象。"""
+
+    name: str
+    products: dict[str, Product] = field(default_factory=dict)
+
+    def add_product(self, product: Product) -> None:
+        """添加商品，商品编码重复时阻止操作。"""
+        if product.code in self.products:
+            raise InventoryError("商品编码已存在。")
+        self.products[product.code] = product
+
+    def remove_stock(self, code: str, quantity: int) -> None:
+        """扣减指定商品库存。"""
+        if code not in self.products:
+            raise InventoryError("商品不存在。")
+
+        product = self.products[code]
+        if quantity <= 0:
+            raise InventoryError("出库数量必须大于 0。")
+        if product.quantity < quantity:
+            raise InventoryError("库存不足。")
+
+        product.quantity -= quantity
+
+
+def main() -> None:
+    """运行仓库领域模型示例。"""
+    warehouse = Warehouse("教学仓库")
+    keyboard = Product("P001", "机械键盘", 3)
+
+    warehouse.add_product(keyboard)
+    print(keyboard)
+    print(f"是否有库存：{keyboard.in_stock}")
+
+    warehouse.remove_stock("P001", 2)
+    print(f"出库后：{keyboard}")
+
+    try:
+        warehouse.remove_stock("P001", 2)
+    except InventoryError as error:
+        print(f"业务操作失败：{error}")
+
+
+if __name__ == "__main__":
+    main()