加密后

service/model_service.py

@@ -11,6 +11,7 @@ from mysql.connector import Error as MySQLError
 
 # 复用项目依赖
 from ds.db import db
+from encryption.encrypt_decorator import encrypt_response
 from schema.model_schema import (
     ModelCreateRequest,
     ModelUpdateRequest,
@@ -190,6 +191,7 @@ def get_current_conf_threshold():
 
 # 1. 上传模型（保持不变）
 @router.post("", response_model=APIResponse, summary="上传YOLO模型（.pt格式）")
+@encrypt_response()
 async def upload_model(
         name: str = Form(..., description="模型名称"),
         description: str = Form(None, description="模型描述"),
@@ -281,6 +283,7 @@ async def upload_model(
 
 # 2. 获取模型列表（保持不变）
 @router.get("", response_model=APIResponse, summary="获取模型列表（分页）")
+@encrypt_response()
 async def get_model_list(
         page: int = Query(1, ge=1),
         page_size: int = Query(10, ge=1, le=100),
@@ -337,6 +340,7 @@ async def get_model_list(
 
 # 3. 获取默认模型（保持不变）
 @router.get("/default", response_model=APIResponse, summary="获取当前默认模型")
+@encrypt_response()
 async def get_default_model():
     conn = None
     cursor = None
@@ -379,6 +383,7 @@ async def get_default_model():
 
 # 4. 获取单个模型详情（保持不变）
 @router.get("/{model_id}", response_model=APIResponse, summary="获取单个模型详情")
+@encrypt_response()
 async def get_model(model_id: int):
     conn = None
     cursor = None
@@ -414,6 +419,7 @@ async def get_model(model_id: int):
 
 # 5. 更新模型信息（保持不变）
 @router.put("/{model_id}", response_model=APIResponse, summary="更新模型信息")
+@encrypt_response()
 async def update_model(model_id: int, model_update: ModelUpdateRequest):
     conn = None
     cursor = None
@@ -497,6 +503,7 @@ async def update_model(model_id: int, model_update: ModelUpdateRequest):
 
 # 5.1 更换默认模型（添加置信度参数）
 @router.put("/{model_id}/set-default", response_model=APIResponse, summary="更换默认模型（自动重启服务）")
+@encrypt_response()
 async def set_default_model(
     model_id: int,
     conf_threshold: float = Query(0.8, ge=0.01, le=0.99, description="模型检测置信度阈值（0.01-0.99）")
@@ -591,6 +598,7 @@ async def set_default_model(
 
 # 6. 删除模型（保持不变）
 @router.delete("/{model_id}", response_model=APIResponse, summary="删除模型")
+@encrypt_response()
 async def delete_model(model_id: int):
     conn = None
     cursor = None
@@ -650,6 +658,7 @@ async def delete_model(model_id: int):
 
 # 7. 下载模型文件（保持不变）
 @router.get("/{model_id}/download", summary="下载模型文件")
+@encrypt_response()
 async def download_model(model_id: int):
     conn = None
     cursor = None