feat 更新响应模板

fix
This commit is contained in:
zhouchengrong
2024-06-13 14:31:14 +08:00
parent 6034a3539b
commit b012b91613
11 changed files with 120 additions and 100 deletions

View File

@@ -1,9 +1,8 @@
import json
import logging
import time
from fastapi import APIRouter
from fastapi import APIRouter, HTTPException
from app.schemas.pre_processing import DesignPreProcessingModel
from app.schemas.response_template import ResponseModel
from app.service.design_pre_processing.service import DesignPreprocessing
router = APIRouter()
@@ -14,16 +13,10 @@ logger = logging.getLogger()
def design_pre_processing(request_data: DesignPreProcessingModel):
try:
logger.info(f"design_pre_processing request item is : @@@@@@:{request_data}")
code = 200
message = "OK!"
start_time = time.time()
server = DesignPreprocessing()
data = server.pipeline(image_list=request_data.sketches)
logger.info(f"design_pre_processing Run time is @@@@@@:{time.time() - start_time}")
logger.info(f"design response @@@@@@:{json.dumps(data, indent=4)}")
except Exception as e:
code = 400
message = str(e)
data = str(e)
logger.warning(f"design Run Exception @@@@@@:{e}")
logger.info({"code": code, "message": message, "data": data})
return {"code": code, "msg": message, "data": data}
raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data)