mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2026-03-13 09:31:31 +08:00
1a6aba6105
* Optimize code generation data processing * Add SQL script generation * Update code gen table scripts * Update types * Fix model jinja and type conversion
42 lines
1.4 KiB
Python
42 lines
1.4 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 GenColumnSchemaBase(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='是否可为空')
|
|
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 CreateGenColumnParam(GenColumnSchemaBase):
|
|
"""创建代码生成模型列参数"""
|
|
|
|
|
|
class UpdateGenColumnParam(GenColumnSchemaBase):
|
|
"""更新代码生成模型列参数"""
|
|
|
|
|
|
class GetGenColumnDetail(GenColumnSchemaBase):
|
|
"""获取代码生成模型列详情"""
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int = Field(description='主键 ID')
|
|
pd_type: str = Field(description='列类型对应的 pydantic 类型')
|