响应格式修改，原始数据查询修改

app/schemas/account.py

@@ -1,5 +1,5 @@
 from pydantic import BaseModel, Field, ConfigDict, field_serializer
-from typing import Optional
+from typing import Optional, List, Any
 from datetime import datetime
 
 class AccountBase(BaseModel):
@@ -62,4 +62,18 @@ class AccountUpdateRequest(BaseModel):
     account_data: AccountUpdate
 
 class AccountDeleteRequest(BaseModel):
-    account_id: int
+    account_id: int
+
+# 统一响应格式
+class AccountApiResponse(BaseModel):
+    """账号API统一响应格式"""
+    code: int = 0
+    message: str
+    data: Optional[Any] = None
+
+class AccountListResponse(BaseModel):
+    """账号列表响应格式"""
+    code: int = 0
+    message: str
+    total: int
+    data: List[AccountResponse] = []

app/schemas/comprehensive_data.py

@@ -70,7 +70,7 @@ class SectionDataImportRequest(BaseModel):
 
 # 原始数据查询请求
 class OriginalDataQueryRequest(BaseModel):
-    account_id: int  # 账号ID,必填
+    account_id: Optional[int] = None  # 账号ID，可选。不填则查询所有分表
     linecode: Optional[str] = None
     id: Optional[int] = None
     bfpcode: Optional[str] = None
@@ -190,19 +190,40 @@ class BatchOriginalDataImportRequest(BaseModel):
 
 # 新增响应模型
 class DataImportResponse(BaseModel):
-    success: bool
+    code: int = 0  # 响应状态码，0表示成功
     message: str
-    total_count: int
-    success_count: int
-    failed_count: int
-    failed_items: List[Dict[str, Any]] = []
+    data: Optional[Dict[str, Any]] = None  # 将详细信息放入data中
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "code": 0,
+                "message": "批量导入完成",
+                "data": {
+                    "total_count": 100,
+                    "success_count": 100,
+                    "failed_count": 0,
+                    "failed_items": []
+                }
+            }
+        }
 
 # 查询响应模型
 class DataResponse(BaseModel):
-    success: bool
+    code: int = 0  # 响应状态码，0表示成功
     message: str
-    count: int
     data: List[Dict[str, Any]] = []
+    total: Optional[int] = None  # 总数
+
+    class Config:
+        schema_extra = {
+            "example": {
+                "code": 0,
+                "message": "查询成功",
+                "total": 10,
+                "data": []
+            }
+        }
 
 # 兼容旧接口的模型
 class ComprehensiveDataImportRequest(BaseModel):

app/schemas/database.py

@@ -5,10 +5,9 @@ class SQLExecuteRequest(BaseModel):
     sql: str
 
 class SQLExecuteResponse(BaseModel):
-    success: bool
+    code: int = 0  # 响应状态码
     message: str
     data: Optional[Any] = None
-    rows_affected: Optional[int] = None
 
 class TableDataRequest(BaseModel):
     table_name: str
@@ -16,10 +15,10 @@ class TableDataRequest(BaseModel):
     offset: Optional[int] = 0
 
 class TableDataResponse(BaseModel):
-    success: bool
+    code: int = 0  # 响应状态码
     message: str
+    total: Optional[int] = None
     data: Optional[List[Dict[str, Any]]] = None
-    total_count: Optional[int] = None
 
 class CreateTableRequest(BaseModel):
     table_name: str

app/schemas/task.py

@@ -36,6 +36,20 @@ class AddDateJobRequest(BaseModel):
     run_date: datetime
 
 class TaskResponse(BaseModel):
-    success: bool
+    """任务API统一响应格式"""
+    code: int = 0
     message: str
-    data: Optional[Any] = None
+    data: Optional[Any] = None
+
+class TaskListResponse(BaseModel):
+    """任务列表响应格式"""
+    code: int = 0
+    message: str
+    total: int
+    data: List[JobResponse] = []
+
+class FunctionListResponse(BaseModel):
+    """可用函数列表响应格式"""
+    code: int = 0
+    message: str
+    data: List[str] = []