xxx

src/inventory/api/product_routes.py

@@ -14,10 +14,11 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy import select, or_, func, delete
 from typing import Optional, List, Dict
 from decimal import Decimal
+from pathlib import Path
 
 from shared.database.database import get_db_session
 from shared.services.auth_service import get_current_active_user, get_current_admin_user
-from shared.models.database import User, Product, ProductMaterial
+from shared.models.database import User, Product, ProductMaterial, STPFile, ProcessingTask
 from ..schemas import (
     ProductCreate,
     ProductResponse,
@@ -118,6 +119,70 @@ async def create_product(
     return _build_product_response(product, 0)
 
 
+@router.post("/from-task/{task_id}", response_model=ProductResponse, status_code=201)
+async def create_product_from_task(
+    task_id: str,
+    db_session: AsyncSession = Depends(get_db_session),
+    current_user: User = Depends(get_current_active_user),
+):
+    """从模具分析任务创建进销存成品，回写 stp_files.product_id（P2-1）"""
+    task_result = await db_session.execute(select(ProcessingTask).where(ProcessingTask.task_id == task_id))
+    task = task_result.scalar_one_or_none()
+    if not task:
+        raise HTTPException(status_code=404, detail="分析任务不存在")
+    stp_result = await db_session.execute(select(STPFile).where(STPFile.id == task.stp_file_id))
+    stp_file = stp_result.scalar_one_or_none()
+    if not stp_file:
+        raise HTTPException(status_code=404, detail="STP 分析记录不存在")
+
+    # 已关联成品则直接返回（幂等）
+    if stp_file.product_id:
+        existed = await db_session.execute(select(Product).where(Product.id == stp_file.product_id))
+        product = existed.scalar_one_or_none()
+        if product:
+            return _build_product_response(product, 0)
+
+    # 生成唯一 SKU：MI{stp_file_id}，冲突则追加序号
+    base_sku = f"MI{stp_file_id}"
+    sku = base_sku
+    n = 1
+    while True:
+        conflict = await db_session.execute(select(Product).where(Product.sku == sku))
+        if not conflict.scalar_one_or_none():
+            break
+        n += 1
+        sku = f"{base_sku}-{n}"
+
+    name = Path(stp_file.original_filename or f"mold_{stp_file_id}").stem or f"模具分析-{stp_file_id}"
+    desc_parts = []
+    if stp_file.volume:
+        desc_parts.append(f"体积 {stp_file.volume:.1f} mm³")
+    if stp_file.product_weight:
+        desc_parts.append(f"重量 {stp_file.product_weight:.2f} g")
+    if stp_file.surface_area:
+        desc_parts.append(f"表面积 {stp_file.surface_area:.1f} mm²")
+    description = "由模具分析创建" + ("：" + "；".join(desc_parts) if desc_parts else "")
+
+    product = Product(
+        sku=sku,
+        name=name,
+        description=description,
+        category="模具成品",
+        unit="件",
+        item_type="finished",
+        cost_price=0,
+        sale_price=0,
+        min_stock=0,
+        max_stock=0,
+    )
+    db_session.add(product)
+    await db_session.flush()
+    stp_file.product_id = product.id
+    await db_session.commit()
+    await db_session.refresh(product)
+    return _build_product_response(product, 0)
+
+
 @router.put("/{product_id}", response_model=ProductResponse)
 async def update_product(
     product_id: int,