46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
|
import time
|
|||
|
|
from typing import Generic, TypeVar, Union
|
|||
|
|
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
from loveace.router.schemas.model import ErrorModel, ValidationErrorModel
|
|||
|
|
|
|||
|
|
T = TypeVar("T")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class UniResponseModel(BaseModel, Generic[T]):
|
|||
|
|
"""
|
|||
|
|
统一响应模型，适用于所有API响应。
|
|||
|
|
Attributes:
|
|||
|
|
success (bool): 操作是否成功。
|
|||
|
|
message (str | None): 操作的详细信息。
|
|||
|
|
data (ResponseModel | None): 操作返回的数据。
|
|||
|
|
error (DetailModel | None): 操作错误信息，支持 ErrorModel 或 ValidationErrorDetail。
|
|||
|
|
timestamp (str): 响应生成的时间戳，格式为 "YYYY-MM-DD HH:MM:SS"。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
success: bool = Field(..., description="操作是否成功")
|
|||
|
|
message: str | None = Field(..., description="操作的详细信息")
|
|||
|
|
data: T | None = Field(..., description="操作返回的数据")
|
|||
|
|
error: Union[ErrorModel, ValidationErrorModel] | None = Field(
|
|||
|
|
None, description="操作错误信息"
|
|||
|
|
)
|
|||
|
|
timestamp: str = Field(
|
|||
|
|
default_factory=lambda: time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
|
|||
|
|
description="响应生成的时间戳",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
def from_response(
|
|||
|
|
cls,
|
|||
|
|
success: bool,
|
|||
|
|
message: str,
|
|||
|
|
data: T | None = None,
|
|||
|
|
) -> "UniResponseModel":
|
|||
|
|
return cls(
|
|||
|
|
success=success,
|
|||
|
|
message=message,
|
|||
|
|
data=data,
|
|||
|
|
error=None,
|
|||
|
|
)
|