x

src/api/inventory/sales_order_routes.py

@@ -13,6 +13,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy import select, func, delete, update
 from typing import Optional, List
 from math import ceil
+from pydantic import BaseModel, Field
 
 from database.database import get_db_session
 from services.auth_service import get_current_active_user
@@ -513,7 +514,7 @@ async def delete_sales_order(
     db_session: AsyncSession = Depends(get_db_session),
     current_user: User = Depends(get_current_active_user)
 ):
-    order, _ = await _get_sales_order_with_customer(db_session, order_id)
+    order, customer = await _get_sales_order_with_customer(db_session, order_id)
     if order.status == "delivered":
         raise HTTPException(status_code=400, detail="已交付的销售订单禁止删除")
     await _rollback_issued_materials(db_session, order, current_user)
@@ -522,6 +523,88 @@ async def delete_sales_order(
     return {"message": "销售订单已删除"}
 
 
+class MaterialConsumptionItem(BaseModel):
+    material_id: int
+    quantity: float = Field(gt=0)
+    remark: Optional[str] = None
+
+
+class MaterialConsumptionRequest(BaseModel):
+    items: List[MaterialConsumptionItem] = Field(min_length=1)
+
+
+@router.post("/{order_id}/consume-materials")
+async def consume_materials(
+    order_id: int,
+    request: MaterialConsumptionRequest,
+    db_session: AsyncSession = Depends(get_db_session),
+    current_user: User = Depends(get_current_active_user)
+):
+    """记录销售订单的物料消耗"""
+    order, customer = await _get_sales_order_with_customer(db_session, order_id)
+    
+    # 计算总物料成本
+    total_cost = 0
+    
+    # 处理每个物料消耗项
+    for item in request.items:
+        # 获取物料信息
+        material = await db_session.get(Product, item.material_id)
+        if not material:
+            raise HTTPException(status_code=404, detail=f"物料 ID {item.material_id} 不存在")
+        if material.item_type != "material":
+            raise HTTPException(status_code=400, detail=f"只能消耗物料类型的产品: {material.name}")
+        
+        # 计算成本
+        cost = material.cost_price * item.quantity
+        total_cost += cost
+        
+        # 更新物料库存
+        inventory_result = await db_session.execute(
+            select(Inventory)
+            .where(Inventory.product_id == material.id)
+            .where(Inventory.warehouse_id == 1)
+        )
+        inventory = inventory_result.scalar()
+        if inventory:
+            before_qty = inventory.quantity
+            inventory.quantity -= item.quantity
+            after_qty = inventory.quantity
+            if after_qty < 0:
+                raise HTTPException(status_code=400, detail=f"物料 {material.name} 库存不足")
+        else:
+            raise HTTPException(status_code=400, detail=f"物料 {material.name} 没有库存记录")
+        
+        # 记录物料消耗
+        movement = StockMovement(
+            product_id=material.id,
+            warehouse_id=1,  # 默认仓库
+            quantity=-item.quantity,
+            before_quantity=before_qty,
+            after_quantity=after_qty,
+            movement_type="consumption",
+            reference_type="sales_order",
+            reference_id=order.id,
+            unit_price=material.cost_price,
+            total_amount=cost,
+            operator_id=current_user.id,
+            remark=item.remark
+        )
+        db_session.add(movement)
+    
+    # 更新订单的实际物料成本
+    order.actual_material_cost = total_cost
+    
+    await db_session.commit()
+    await db_session.refresh(order)
+    
+    return {
+        "message": "物料消耗记录保存成功",
+        "total_cost": total_cost,
+        "order": _build_sales_order_response(order, customer.name)
+    }
+
+
 @router.get("/{order_id}/production-plan", response_model=SalesOrderProductionPlanResponse)
 async def get_sales_order_production_plan(
     order_id: int,