初始提交

shierfeng-fastapi-backend/common/vo.py

@@ -0,0 +1,105 @@
+from datetime import datetime
+from typing import Any, Generic, TypeVar
+
+from pydantic import BaseModel, ConfigDict, Field, create_model
+from pydantic.alias_generators import to_camel
+from typing_extensions import Self
+
+from common.constant import HttpStatusConstant
+
+T = TypeVar('T')
+
+
+class CrudResponseModel(BaseModel):
+    """
+    操作响应模型
+    """
+
+    is_success: bool = Field(description='操作是否成功')
+    message: str = Field(description='响应信息')
+    result: Any | None = Field(default=None, description='响应结果')
+
+
+class ResponseBaseModel(BaseModel):
+    """
+    响应模型
+    """
+
+    code: int = Field(default=HttpStatusConstant.SUCCESS, description='响应码')
+    msg: str = Field(default='操作成功', description='响应信息')
+    success: bool = Field(default=True, description='响应是否成功')
+    time: datetime = Field(default_factory=datetime.now, description='响应时间')
+
+
+class DynamicResponseModel(ResponseBaseModel, Generic[T]):
+    """
+    动态响应模型
+    """
+
+    model_config = ConfigDict(alias_generator=to_camel)
+
+    def __class_getitem__(cls, item: Any) -> Any | Self:
+        """
+        当使用 DynamicResponseModel[Item] 语法时，动态创建一个包含所有字段的新模型
+        """
+        # 检查是否已经为该类型创建了模型
+        if not hasattr(cls, '_cached_models'):
+            cls._cached_models = {}
+
+        if item in cls._cached_models:
+            return cls._cached_models[item]
+
+        # 检查item是否为Pydantic模型
+        if not hasattr(item, 'model_fields'):
+            raise TypeError(f'{item} 不是一个Pydantic模型，请使用Pydantic模型作为泛型参数')
+
+        # 获取ResponseBaseModel的字段
+        base_fields = {}
+        for field_name, field in cls.model_fields.items():
+            base_fields[field_name] = (field.annotation, field)
+
+        # 获取泛型类型的字段
+        item_fields = {}
+        for field_name, field in item.model_fields.items():
+            item_fields[field_name] = (field.annotation, field)
+
+        # 合并所有字段
+        all_fields = {**base_fields, **item_fields}
+
+        # 动态创建新模型
+        new_model = create_model(
+            f'DynamicResponseModel[{item.__name__}]', __base__=cls, __config__=cls.model_config, **all_fields
+        )
+
+        # 缓存模型
+        cls._cached_models[item] = new_model
+
+        return new_model
+
+
+class PageModel(BaseModel, Generic[T]):
+    """
+    分页模型
+    """
+
+    model_config = ConfigDict(alias_generator=to_camel)
+
+    rows: list[T] = Field(description='记录列表')
+    page_num: int = Field(description='当前页码')
+    page_size: int = Field(description='每页记录数')
+    total: int = Field(description='总记录数')
+    has_next: bool = Field(description='是否有下一页')
+
+
+class PageResponseModel(PageModel, ResponseBaseModel, Generic[T]):
+    """
+    分页响应模型
+    """
+
+
+class DataResponseModel(ResponseBaseModel, Generic[T]):
+    """
+    数据响应模型
+    """
+
+    data: T = Field(description='响应数据')