feat generate 升级 attribute retrieve 迁移

app/api/api_attribute_retrieve.py

@@ -0,0 +1,45 @@
+import json
+import logging
+from fastapi import APIRouter
+
+from app.schemas.attribute_retrieve import *
+from app.service.attribute.config import const
+from app.service.attribute.service_att_recognition import AttributeRecognition
+from app.service.attribute.service_category_recognition import CategoryRecognition
+
+router = APIRouter()
+logger = logging.getLogger()
+
+
+# 属性识别
+@router.post("/attribute_recognition")
+def attribute_recognition(request_item: list[AttributeRecognitionModel]):
+    try:
+        service = AttributeRecognition(const=const, request_data=request_item)
+        data = service.get_result()
+        code = 200
+        message = "access"
+        logger.info(f"attribute_recognition response @@@@@@:{json.dumps(data, indent=4)}")
+    except Exception as e:
+        code = 400
+        message = e
+        data = e
+        logger.warning(f"attribute_recognition Run Exception @@@@@@:{e}")
+    return {"code": code, "message": message, "data": data}
+
+
+# 类别识别
+@router.post("/category_recognition")
+def category_recognition(request_item: list[CategoryRecognitionModel]):
+    try:
+        service = CategoryRecognition(request_data=request_item)
+        data = service.get_result()
+        code = 200
+        message = "access"
+        logger.info(f"category_recognition response @@@@@@:{json.dumps(data, indent=4)}")
+    except Exception as e:
+        code = 400
+        message = e
+        data = e
+        logger.warning(f"category_recognition Run Exception @@@@@@:{e}")
+    return {"code": code, "message": message, "data": data}