x

src/services/processing_service.py

@@ -17,6 +17,7 @@ from core.aluminum_foam_mold import AluminumFoamMoldGenerator
 from core.mold_quality_inspector import AluminumFoamMoldQualityInspector
 from core.mesh_generator import MeshGenerator
 from core.multi_scheme_planner import MultiSchemeMoldPlanner
+from core.cad_exporter import CADExporter
 from services.storage_integration_rustfs import StorageIntegrationService
 from services.redis_task_manager import redis_task_manager
 from services.material_service import MaterialService
@@ -43,6 +44,7 @@ class ProcessingService:
         self.html_generator = HTMLGenerator()
         self.storage_service = StorageIntegrationService()
         self.multi_scheme_planner = MultiSchemeMoldPlanner()
+        self.cad_exporter = CADExporter()
         self._export_shapes_cache: Dict[str, Dict[str, Dict[str, Any]]] = {}
 
     # ─── 对外入口 ───
@@ -145,10 +147,16 @@ class ProcessingService:
             )
             stage_timings["generate_cavity"] = round(time.perf_counter() - stage_started, 3)
             export_shapes = {}
+            export_artifacts = None
             if plan_result:
                 export_shapes = plan_result.pop("_export_shapes", {}) or {}
                 if export_shapes:
                     self._cache_export_shapes(task_id, export_shapes)
+                    export_artifacts = self._persist_step_exports(
+                        task_id=task_id,
+                        original_filename=Path(file_path).name,
+                        export_shapes=export_shapes,
+                    )
 
             # 4. 生成详细JSON数据 — 委托 CalculationService
             await self.storage_service.update_task_status(
@@ -315,6 +323,7 @@ class ProcessingService:
                     "material": requested_material,
                     "verification": verification_result,
                     "llm_report": llm_report,
+                    "export_artifacts": export_artifacts,
                     **process_params,
                 },
             )
@@ -334,6 +343,7 @@ class ProcessingService:
                 "html_file": best_scheme.get("html_file", f"/html/{Path(html_file_path).name}") if best_scheme else f"/html/{Path(html_file_path).name}",
                 "verification": verification_result,
                 "llm_report": llm_report,
+                "export_artifacts": export_artifacts,
                 "status": ProcessingStatus.COMPLETED,
                 "completed_at": str(datetime.now()),
             })
@@ -447,6 +457,55 @@ class ProcessingService:
     def _cache_export_shapes(self, task_id: str, export_shapes: Dict[str, Dict[str, Any]]):
         self._export_shapes_cache[task_id] = export_shapes
 
+    def _persist_step_exports(
+        self,
+        task_id: str,
+        original_filename: str,
+        export_shapes: Dict[str, Dict[str, Any]],
+    ) -> Optional[Dict[str, Any]]:
+        if not export_shapes:
+            return None
+
+        base_filename = Path(original_filename).stem or f"mold_{task_id}"
+        manifest = {
+            "version": 1,
+            "task_id": task_id,
+            "generated_at": datetime.now().isoformat(),
+            "schemes": {},
+        }
+        components = ["cavity", "core", "parting_surface"]
+
+        for scheme_id, cavity_data in export_shapes.items():
+            try:
+                result = self.cad_exporter.export_mold_results(
+                    cavity_data=cavity_data,
+                    base_filename=base_filename,
+                    formats=["step"],
+                    components=components,
+                    task_id=task_id,
+                    scheme_id=scheme_id,
+                )
+                manifest["schemes"][scheme_id] = {
+                    "base_filename": result.get("base_filename"),
+                    "generated_at": datetime.now().isoformat(),
+                    "files": result.get("files", []),
+                    "errors": result.get("errors", []),
+                    "total_files": result.get("total_files", 0),
+                    "total_errors": result.get("total_errors", 0),
+                }
+            except Exception as exc:
+                logger.warning("持久化 STEP 导出失败: task=%s scheme=%s error=%s", task_id, scheme_id, exc)
+                manifest["schemes"][scheme_id] = {
+                    "base_filename": base_filename,
+                    "generated_at": datetime.now().isoformat(),
+                    "files": [],
+                    "errors": [str(exc)],
+                    "total_files": 0,
+                    "total_errors": 1,
+                }
+
+        return manifest
+
     def get_export_shapes(self, task_id: str, scheme_id: Optional[str] = None) -> Optional[Dict[str, Any]]:
         scheme_map = self._export_shapes_cache.get(task_id, {})
         if not scheme_map: