feat design batch

fix

app/api/api_design.py

@@ -1,12 +1,14 @@
 import json
 import logging
+import os
 
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter, HTTPException, UploadFile, File, Form
 
-from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel
+from app.schemas.design import DesignModel, DesignProgressModel, ModelProgressModel, DBGConfigModel
 from app.schemas.response_template import ResponseModel
 from app.service.design.model_process_service import model_transpose
 from app.service.design.service import generate
+from app.service.design.service_design_batch_generate import start_design_batch_generate
 from app.service.design.utils.redis_utils import Redis
 
 router = APIRouter()
@@ -238,3 +240,38 @@ def model_process(request_data: ModelProgressModel):
         logger.warning(f"model_process Run Exception @@@@@@:{e}")
         raise HTTPException(status_code=404, detail=str(e))
     return ResponseModel(data=data)
+
+
+# ##############################################################
+
+
+@router.post("/design_batch_generate")
+async def design(file: UploadFile = File(...),
+                 tasks_id: str = Form(...),
+                 user_id: str = Form(...),
+                 priority: int = Form(...),
+                 total: int = Form(...)
+                 ):
+    # file_content = await file.read()
+    dbg_config = DBGConfigModel(
+        tasks_id=tasks_id,
+        user_id=user_id,
+        priority=priority,
+        total=total
+    )
+    contents = await file.read()
+    file_name = file.filename
+    await save_request_file(contents, file_name)
+
+    return await start_design_batch_generate(dbg_config, contents)
+
+
+async def save_request_file(contents, file_name):
+    # 创建保存文件的目录（如果不存在）
+    save_dir = os.path.join(os.getcwd(), "design_batch", "request_data")
+    if not os.path.exists(save_dir):
+        os.makedirs(save_dir)
+    # 处理文件
+    file_path = os.path.join(save_dir, file_name)
+    with open(file_path, "wb") as f:
+        f.write(contents)