feat 更新响应模板

fix
This commit is contained in:
zhouchengrong
2024-06-17 17:14:10 +08:00
parent e0d9512b26
commit 88014bdb4c
2 changed files with 9 additions and 4 deletions

View File

@@ -1,9 +1,11 @@
import json import json
import logging import logging
from fastapi import APIRouter, HTTPException from fastapi import APIRouter, HTTPException
from app.core.config import DEBUG
from app.schemas.attribute_retrieve import * from app.schemas.attribute_retrieve import *
from app.schemas.response_template import ResponseModel from app.schemas.response_template import ResponseModel
from app.service.attribute.config import const from app.service.attribute.config import const, local_debug_const
from app.service.attribute.service_att_recognition import AttributeRecognition from app.service.attribute.service_att_recognition import AttributeRecognition
from app.service.attribute.service_category_recognition import CategoryRecognition from app.service.attribute.service_category_recognition import CategoryRecognition
@@ -16,13 +18,16 @@ logger = logging.getLogger()
def attribute_recognition(request_item: list[AttributeRecognitionModel]): def attribute_recognition(request_item: list[AttributeRecognitionModel]):
try: try:
logger.info(f"attribute_recognition request item is : @@@@@@:{request_item}") logger.info(f"attribute_recognition request item is : @@@@@@:{request_item}")
service = AttributeRecognition(const=const, request_data=request_item) if DEBUG:
service = AttributeRecognition(const=local_debug_const, request_data=request_item)
else:
service = AttributeRecognition(const=const, request_data=request_item)
data = service.get_result() data = service.get_result()
logger.info(f"attribute_recognition response @@@@@@:{json.dumps(data, indent=4)}") logger.info(f"attribute_recognition response @@@@@@:{json.dumps(data, indent=4)}")
except Exception as e: except Exception as e:
logger.warning(f"attribute_recognition Run Exception @@@@@@:{e}") logger.warning(f"attribute_recognition Run Exception @@@@@@:{e}")
raise HTTPException(status_code=404, detail=str(e)) raise HTTPException(status_code=404, detail=str(e))
return ResponseModel(data=data) return ResponseModel(data={"list": data})
# 类别识别 # 类别识别

View File

@@ -19,7 +19,7 @@ class Settings(BaseSettings):
LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py') LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
DEBUG = False DEBUG = True
if DEBUG: if DEBUG:
LOGS_PATH = "logs/" LOGS_PATH = "logs/"
CATEGORY_PATH = "service/attribute/config/descriptor/category/category_dis.csv" CATEGORY_PATH = "service/attribute/config/descriptor/category/category_dis.csv"