101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
from typing import Generic, Optional, TypeVar, Any
|
|
from pydantic import BaseModel, Field
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
class BaseResponse(BaseModel, Generic[T]):
|
|
"""通用响应模型基类"""
|
|
|
|
code: int = Field(200, description="状态码")
|
|
message: str = Field("成功", description="提示信息")
|
|
data: Optional[T] = Field(None, description="响应数据")
|
|
|
|
@classmethod
|
|
def success(cls, data: T, message: str = "获取成功") -> "BaseResponse[T]":
|
|
"""创建成功响应"""
|
|
return cls(code=200, message=message, data=data)
|
|
|
|
@classmethod
|
|
def error(
|
|
cls, message: str = "请求失败", code: int = 500, data: Optional[T] = None
|
|
) -> "BaseResponse[T]":
|
|
"""创建错误响应"""
|
|
return cls(code=code, message=message, data=data)
|
|
|
|
@classmethod
|
|
def from_data(
|
|
cls,
|
|
data: Any,
|
|
success_message: str = "获取成功",
|
|
error_message: str = "网络请求失败，已进行多次重试",
|
|
) -> "BaseResponse[T]":
|
|
"""
|
|
根据数据自动判断是否为错误模型并生成相应响应
|
|
|
|
Args:
|
|
data: 要检查的数据
|
|
success_message: 成功时的消息
|
|
error_message: 失败时的消息
|
|
|
|
Returns:
|
|
BaseResponse: 相应的响应模型
|
|
"""
|
|
if cls._is_error_data(data):
|
|
return cls.error(message=error_message, code=500, data=data)
|
|
else:
|
|
return cls.success(data=data, message=success_message)
|
|
|
|
@staticmethod
|
|
def _is_error_data(data: Any) -> bool:
|
|
"""
|
|
检测数据是否为错误模型
|
|
|
|
Args:
|
|
data: 要检查的数据
|
|
|
|
Returns:
|
|
bool: 如果是错误数据返回True
|
|
"""
|
|
if data is None:
|
|
return True
|
|
|
|
# 检查是否有错误指示符
|
|
if hasattr(data, "total_score") and data.total_score == -1.0:
|
|
return True
|
|
if hasattr(data, "completed_courses") and data.completed_courses == -1:
|
|
return True
|
|
if hasattr(data, "gpa") and data.gpa == -1.0:
|
|
return True
|
|
if hasattr(data, "plan_name") and data.plan_name == "请求失败，请稍后重试":
|
|
return True
|
|
if hasattr(data, "code") and data.code == -1:
|
|
return True
|
|
if hasattr(data, "total_count") and data.total_count == -1:
|
|
return True
|
|
if hasattr(data, "result") and data.result == "failed":
|
|
return True
|
|
if (
|
|
hasattr(data, "can_select")
|
|
and hasattr(data, "start_time")
|
|
and data.start_time == "请求失败"
|
|
):
|
|
return True
|
|
|
|
# 检查列表类型的错误数据
|
|
if isinstance(data, list) and len(data) > 0:
|
|
first_item = data[0]
|
|
if hasattr(first_item, "id") and first_item.id == "error":
|
|
return True
|
|
if hasattr(first_item, "type_name") and first_item.type_name == "请求失败":
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
class ErrorResponse(BaseResponse[None]):
|
|
"""专用错误响应模型"""
|
|
|
|
def __init__(self, message: str = "请求失败，请稍后重试", code: int = 500):
|
|
super().__init__(code=code, message=message, data=None)
|