mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-23 13:33:08 +00:00
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from pydantic import ConfigDict, Field, field_validator
|
|
|
|
from backend.common.schema import SchemaBase
|
|
from backend.plugin.code_generator.utils.type_conversion import sql_type_to_sqlalchemy
|
|
|
|
|
|
class CodeGenColumnSchemaBase(SchemaBase):
|
|
"""代码生成模型基础模型"""
|
|
|
|
name: str = Field(description='列名称')
|
|
comment: str | None = Field(None, description='列描述')
|
|
type: str = Field(description='SQLA 模型列类型')
|
|
default: str | None = Field(None, description='列默认值')
|
|
sort: int = Field(description='列排序')
|
|
length: int = Field(description='列长度')
|
|
is_pk: bool = Field(False, description='是否主键')
|
|
is_nullable: bool = Field(False, description='是否可为空')
|
|
code_gen_business_id: int = Field(description='代码生成业务ID')
|
|
|
|
@field_validator('type')
|
|
@classmethod
|
|
def normalize_type(cls, v: str) -> str:
|
|
"""规范化类型"""
|
|
return sql_type_to_sqlalchemy(v)
|
|
|
|
|
|
class CreateCodeGenColumnParam(CodeGenColumnSchemaBase):
|
|
"""创建代码生成模型列参数"""
|
|
|
|
|
|
class CreateCodeGenColumnInternalParam(CreateCodeGenColumnParam):
|
|
"""创建代码生成模型列内部参数"""
|
|
|
|
pd_type: str | None = Field(None, description='列类型对应的 pydantic 类型')
|
|
|
|
|
|
class UpdateCodeGenColumnParam(CodeGenColumnSchemaBase):
|
|
"""更新代码生成模型列参数"""
|
|
|
|
|
|
class GetCodeGenColumnDetail(CodeGenColumnSchemaBase):
|
|
"""获取代码生成模型列详情"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int = Field(description='主键 ID')
|
|
pd_type: str = Field(description='列类型对应的 pydantic 类型')
|