借库逻辑实现

inventory-backend/app/services/inbound/semi_service.py

@@ -11,7 +11,44 @@ import json
 class SemiInboundService:
 
     # ============================================================
-    # 1. 基础物料搜索 (已修复：支持空关键词返回最新数据)
+    # 0. 辅助：唯一性校验 (新增核心逻辑)
+    # ============================================================
+    @staticmethod
+    def _check_unique(base_id, serial_number, batch_number, exclude_id=None):
+        """
+        校验半成品的唯一性
+        :param base_id: 基础物料ID
+        :param serial_number: 序列号
+        :param batch_number: 批号
+        :param exclude_id: 排除的ID
+        """
+        from app.models.inbound.semi import StockSemi
+
+        # 1. 序列号 (SN) 校验 - 全局唯一
+        if serial_number:
+            query = StockSemi.query.filter(StockSemi.serial_number == serial_number)
+            if exclude_id:
+                query = query.filter(StockSemi.id != exclude_id)
+
+            exists = query.first()
+            if exists:
+                occupied_name = exists.material.name if (hasattr(exists, 'material') and exists.material) else "未知物料"
+                raise ValueError(f"序列号【{serial_number}】已存在！被半成品 [{occupied_name}] 占用，请核查。")
+
+        # 2. 批号 (BN) 校验 - 同物料下不能重复开单
+        if batch_number and base_id:
+            query = StockSemi.query.filter(
+                StockSemi.base_id == base_id,
+                StockSemi.batch_number == batch_number
+            )
+            if exclude_id:
+                query = query.filter(StockSemi.id != exclude_id)
+
+            if query.first():
+                raise ValueError(f"该物料已存在批号【{batch_number}】，请勿重复建单，建议在原批次上追加库存。")
+
+    # ============================================================
+    # 1. 基础物料搜索
     # ============================================================
     @staticmethod
     def search_base_material(keyword):
@@ -63,6 +100,13 @@ class SemiInboundService:
             if not material:
                 raise ValueError(f"ID为 {base_id} 的基础物料不存在")
 
+            # --- [核心修改] 执行唯一性校验 ---
+            SemiInboundService._check_unique(
+                base_id=base_id,
+                serial_number=data.get('serial_number'),
+                batch_number=data.get('batch_number')
+            )
+
             # [核心修改] 强制北京时间
             beijing_tz = timezone(timedelta(hours=8))
             current_time = datetime.now(beijing_tz).replace(tzinfo=None)
@@ -194,6 +238,18 @@ class SemiInboundService:
             if not stock:
                 raise ValueError("记录不存在")
 
+            # --- [核心修改] 编辑时也要校验唯一性 ---
+            new_base_id = data.get('base_id', stock.base_id)
+            new_sn = data.get('serial_number', stock.serial_number)
+            new_bn = data.get('batch_number', stock.batch_number)
+
+            SemiInboundService._check_unique(
+                base_id=new_base_id,
+                serial_number=new_sn,
+                batch_number=new_bn,
+                exclude_id=stock_id
+            )
+
             field_mapping = {
                 'sku': 'sku',
                 'barcode': 'barcode',