feat: pick up logic of api - stock_record_change.post to stock - services module

stock/services.py

@@ -1,13 +1,101 @@
 from . import models
+from django.db import transaction
 from django.db.models import Sum
 from django.utils import timezone
 from sse.services import push_simple_message_with_object_id
+from basic_info import models as basic_models
 import logging
+from typing import List, Dict, Any, Tuple
 
 
 logger = logging.getLogger(__name__)
 
 
+def create_stock_change_record_with_details(
+    *,
+    merchant: basic_models.Merchant,
+    created_by,
+    type: int,
+    warehouse_id: int,
+    source_type: int,
+    source_id: int | None = None,
+    products: List[Dict[str, Any]] | None = None,
+) -> Tuple[models.StockChangeRecord, List[models.StockChangeDetail], int]:
+    """
+    创建库存变动记录及其明细
+
+    参数:
+        merchant: 当前商户
+        created_by: 操作人（可为空）
+        type: 出入库类型
+        warehouse_id: 仓库ID
+        source_type: 来源类型
+        source_id: 来源单据ID，可选
+        products: 产品及数量列表
+    """
+
+    if not merchant:
+        raise ValueError('必须提供商户信息')
+    if not warehouse_id:
+        raise ValueError('必须提供仓库ID')
+    if not products:
+        raise ValueError('产品列表不能为空')
+
+    try:
+        warehouse = basic_models.WareHouse.objects.get(id=warehouse_id)
+    except basic_models.WareHouse.DoesNotExist:
+        raise ValueError(f'仓库ID {warehouse_id} 不存在')
+
+    if warehouse.merchant_id != merchant.id:
+        raise ValueError('仓库不属于当前商户')
+
+    created_details: List[models.StockChangeDetail] = []
+    created_count = 0
+
+    with transaction.atomic():
+        stock_change_record = models.StockChangeRecord.objects.create(
+            type=type,
+            warehouse=warehouse,
+            source_type=source_type,
+            source_id=source_id,
+            merchant=merchant,
+            created_by=created_by,
+        )
+        logger.info('创建新库存变动记录 ID: %s', stock_change_record.id)
+
+        for product_data in products:
+            product_id = product_data.get('product')
+            quantities = product_data.get('quantity', [])
+
+            if not product_id or not quantities:
+                raise ValueError('产品数据不完整，缺少 product 或 quantity')
+
+            try:
+                product = basic_models.Product.objects.get(id=product_id)
+            except basic_models.Product.DoesNotExist:
+                raise ValueError(f'产品ID {product_id} 不存在')
+
+            if product.merchant_id != merchant.id:
+                raise ValueError(f'产品ID {product_id} 不属于当前商户')
+
+            for quantity in quantities:
+                detail = models.StockChangeDetail.objects.create(
+                    stock_change_record=stock_change_record,
+                    product=product,
+                    quantity=quantity,
+                    merchant=merchant,
+                    unit=product.unit,
+                )
+                created_details.append(detail)
+                created_count += 1
+
+        if warehouse.merchant.auto_complete_stock_change:
+            make_stock_change_completed(stock_change_record)
+            logger.info('自动确认库存变动记录 ID: %s', stock_change_record.id)
+
+    return stock_change_record, created_details, created_count
+
+
 def find_inventory(product_id: int, warehouse_id: int) -> models.Inventory | None:
     """根据产品ID和仓库ID查找库存记录"""
 

stock/tests.py

@@ -371,6 +371,72 @@ class MakeStockChangeCompletedTestCase(StockServicesTestCase):
         # 验证记录状态
         self.stock_change_in.refresh_from_db()
         self.assertTrue(self.stock_change_in.is_finished)
+
+
+class CreateStockChangeRecordWithDetailsTestCase(StockServicesTestCase):
+    """测试 create_stock_change_record_with_details 服务"""
+
+    def setUp(self):
+        super().setUp()
+        self.products_payload = [
+            {'product': self.product_fabric_a.id, 'quantity': [Decimal('10.00'), Decimal('20.50')]},
+            {'product': self.product_fabric_b.id, 'quantity': [Decimal('5.00')]}
+        ]
+
+    def test_create_record_success(self):
+        """创建库存变动记录并返回所有明细"""
+        record, details, created_count = services.create_stock_change_record_with_details(
+            merchant=self.merchant,
+            created_by=None,
+            type=models.StockChangeTypeEnum.ADD,
+            warehouse_id=self.warehouse_main.id,
+            source_type=models.StockChangeSourceEnum.PURCHASE,
+            source_id=self.purchase_order.id,
+            products=self.products_payload,
+        )
+
+        self.assertEqual(record.details.count(), 3)
+        self.assertEqual(len(details), 3)
+        self.assertEqual(created_count, 3)
+        self.assertEqual(record.merchant, self.merchant)
+        self.assertEqual(record.warehouse, self.warehouse_main)
+        self.assertCountEqual(
+            [Decimal('10.00'), Decimal('20.50'), Decimal('5.00')],
+            [detail.quantity for detail in details]
+        )
+
+    def test_auto_complete_triggers_completion(self):
+        """商户开启自动确认时会调用完成逻辑"""
+        self.merchant.auto_complete_stock_change = True
+        self.merchant.save()
+
+        with patch('stock.services.make_stock_change_completed') as mock_complete:
+            services.create_stock_change_record_with_details(
+                merchant=self.merchant,
+                created_by=None,
+                type=models.StockChangeTypeEnum.ADD,
+                warehouse_id=self.warehouse_main.id,
+                source_type=models.StockChangeSourceEnum.PURCHASE,
+                source_id=self.purchase_order.id,
+                products=self.products_payload,
+            )
+
+        mock_complete.assert_called_once()
+
+    def test_invalid_product_raises_error(self):
+        """无效的产品ID会触发 ValueError"""
+        products = [{'product': 99999, 'quantity': [Decimal('1.00')]}]
+
+        with self.assertRaises(ValueError):
+            services.create_stock_change_record_with_details(
+                merchant=self.merchant,
+                created_by=None,
+                type=models.StockChangeTypeEnum.ADD,
+                warehouse_id=self.warehouse_main.id,
+                source_type=models.StockChangeSourceEnum.PURCHASE,
+                source_id=self.purchase_order.id,
+                products=products,
+            )
     
     def test_complete_outbound_record(self):
         """测试完成出库记录"""