初始提交
This commit is contained in:
焦龙言
@@ -0,0 +1,106 @@
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import CHAR, BigInteger, Column, DateTime, Integer, String
|
||||
from sqlalchemy.orm import foreign, relationship
|
||||
|
||||
from config.database import Base
|
||||
from config.env import DataBaseConfig
|
||||
from utils.common_util import SqlalchemyUtil
|
||||
|
||||
|
||||
class GenTable(Base):
|
||||
"""
|
||||
代码生成业务表
|
||||
"""
|
||||
|
||||
__tablename__ = 'gen_table'
|
||||
__table_args__ = {'comment': '代码生成业务表'}
|
||||
|
||||
table_id = Column(BigInteger, primary_key=True, nullable=False, autoincrement=True, comment='编号')
|
||||
table_name = Column(String(200), nullable=True, server_default="''", comment='表名称')
|
||||
table_comment = Column(String(500), nullable=True, server_default="''", comment='表描述')
|
||||
sub_table_name = Column(
|
||||
String(64),
|
||||
nullable=True,
|
||||
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
|
||||
comment='关联子表的表名',
|
||||
)
|
||||
sub_table_fk_name = Column(
|
||||
String(64),
|
||||
nullable=True,
|
||||
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
|
||||
comment='子表关联的外键名',
|
||||
)
|
||||
class_name = Column(String(100), nullable=True, server_default="''", comment='实体类名称')
|
||||
tpl_category = Column(
|
||||
String(200), nullable=True, server_default='crud', comment='使用的模板（crud单表操作 tree树表操作）'
|
||||
)
|
||||
tpl_web_type = Column(
|
||||
String(30), nullable=True, server_default="''", comment='前端模板类型（element-ui模版 element-plus模版）'
|
||||
)
|
||||
package_name = Column(String(100), nullable=True, comment='生成包路径')
|
||||
module_name = Column(String(30), nullable=True, comment='生成模块名')
|
||||
business_name = Column(String(30), nullable=True, comment='生成业务名')
|
||||
function_name = Column(String(50), nullable=True, comment='生成功能名')
|
||||
function_author = Column(String(50), nullable=True, comment='生成功能作者')
|
||||
gen_type = Column(CHAR(1), nullable=True, server_default='0', comment='生成代码方式（0zip压缩包 1自定义路径）')
|
||||
gen_path = Column(String(200), nullable=True, server_default='/', comment='生成路径（不填默认项目路径）')
|
||||
options = Column(String(1000), nullable=True, comment='其它生成选项')
|
||||
create_by = Column(String(64), nullable=True, server_default="''", comment='创建者')
|
||||
create_time = Column(DateTime, nullable=True, default=datetime.now(), comment='创建时间')
|
||||
update_by = Column(String(64), nullable=True, server_default="''", comment='更新者')
|
||||
update_time = Column(DateTime, nullable=True, default=datetime.now(), comment='更新时间')
|
||||
remark = Column(
|
||||
String(500),
|
||||
nullable=True,
|
||||
server_default=SqlalchemyUtil.get_server_default_null(DataBaseConfig.db_type),
|
||||
comment='备注',
|
||||
)
|
||||
|
||||
columns = relationship(
|
||||
'GenTableColumn',
|
||||
primaryjoin=lambda: GenTable.table_id == foreign(GenTableColumn.table_id),
|
||||
order_by='GenTableColumn.sort',
|
||||
back_populates='tables',
|
||||
)
|
||||
|
||||
|
||||
class GenTableColumn(Base):
|
||||
"""
|
||||
代码生成业务表字段
|
||||
"""
|
||||
|
||||
__tablename__ = 'gen_table_column'
|
||||
__table_args__ = {'comment': '代码生成业务表字段'}
|
||||
|
||||
column_id = Column(BigInteger, primary_key=True, autoincrement=True, nullable=False, comment='编号')
|
||||
table_id = Column(BigInteger, nullable=True, comment='归属表编号')
|
||||
column_name = Column(String(200), nullable=True, comment='列名称')
|
||||
column_comment = Column(String(500), nullable=True, comment='列描述')
|
||||
column_type = Column(String(100), nullable=True, comment='列类型')
|
||||
python_type = Column(String(500), nullable=True, comment='PYTHON类型')
|
||||
python_field = Column(String(200), nullable=True, comment='PYTHON字段名')
|
||||
is_pk = Column(CHAR(1), nullable=True, comment='是否主键（1是）')
|
||||
is_increment = Column(CHAR(1), nullable=True, comment='是否自增（1是）')
|
||||
is_required = Column(CHAR(1), nullable=True, comment='是否必填（1是）')
|
||||
is_unique = Column(CHAR(1), nullable=True, comment='是否唯一（1是）')
|
||||
is_insert = Column(CHAR(1), nullable=True, comment='是否为插入字段（1是）')
|
||||
is_edit = Column(CHAR(1), nullable=True, comment='是否编辑字段（1是）')
|
||||
is_list = Column(CHAR(1), nullable=True, comment='是否列表字段（1是）')
|
||||
is_query = Column(CHAR(1), nullable=True, comment='是否查询字段（1是）')
|
||||
query_type = Column(
|
||||
String(200), nullable=True, server_default='EQ', comment='查询方式（等于、不等于、大于、小于、范围）'
|
||||
)
|
||||
html_type = Column(
|
||||
String(200), nullable=True, comment='显示类型（文本框、文本域、下拉框、复选框、单选框、日期控件）'
|
||||
)
|
||||
dict_type = Column(String(200), nullable=True, server_default="''", comment='字典类型')
|
||||
sort = Column(Integer, nullable=True, comment='排序')
|
||||
create_by = Column(String(64), server_default="''", comment='创建者')
|
||||
create_time = Column(DateTime, nullable=True, default=datetime.now(), comment='创建时间')
|
||||
update_by = Column(String(64), server_default="''", comment='更新者')
|
||||
update_time = Column(DateTime, nullable=True, default=datetime.now(), comment='更新时间')
|
||||
|
||||
tables = relationship(
|
||||
'GenTable', primaryjoin=lambda: foreign(GenTableColumn.table_id) == GenTable.table_id, back_populates='columns'
|
||||
)
|
||||
@@ -0,0 +1,294 @@
|
||||
from datetime import datetime
|
||||
from typing import Literal, Optional
|
||||
|
||||
from pydantic import BaseModel, ConfigDict, Field, model_validator
|
||||
from pydantic.alias_generators import to_camel
|
||||
from pydantic_validation_decorator import NotBlank
|
||||
|
||||
from common.constant import GenConstant
|
||||
from utils.string_util import StringUtil
|
||||
|
||||
|
||||
class GenTableBaseModel(BaseModel):
|
||||
"""
|
||||
代码生成业务表对应pydantic模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
||||
|
||||
table_id: int | None = Field(default=None, description='编号')
|
||||
table_name: str | None = Field(default=None, description='表名称')
|
||||
table_comment: str | None = Field(default=None, description='表描述')
|
||||
sub_table_name: str | None = Field(default=None, description='关联子表的表名')
|
||||
sub_table_fk_name: str | None = Field(default=None, description='子表关联的外键名')
|
||||
class_name: str | None = Field(default=None, description='实体类名称')
|
||||
tpl_category: str | None = Field(default=None, description='使用的模板（crud单表操作 tree树表操作）')
|
||||
tpl_web_type: str | None = Field(default=None, description='前端模板类型（element-ui模版 element-plus模版）')
|
||||
package_name: str | None = Field(default=None, description='生成包路径')
|
||||
module_name: str | None = Field(default=None, description='生成模块名')
|
||||
business_name: str | None = Field(default=None, description='生成业务名')
|
||||
function_name: str | None = Field(default=None, description='生成功能名')
|
||||
function_author: str | None = Field(default=None, description='生成功能作者')
|
||||
gen_type: Literal['0', '1'] | None = Field(default=None, description='生成代码方式（0zip压缩包 1自定义路径）')
|
||||
gen_path: str | None = Field(default=None, description='生成路径（不填默认项目路径）')
|
||||
options: str | None = Field(default=None, description='其它生成选项')
|
||||
create_by: str | None = Field(default=None, description='创建者')
|
||||
create_time: datetime | None = Field(default=None, description='创建时间')
|
||||
update_by: str | None = Field(default=None, description='更新者')
|
||||
update_time: datetime | None = Field(default=None, description='更新时间')
|
||||
remark: str | None = Field(default=None, description='备注')
|
||||
|
||||
@NotBlank(field_name='table_name', message='表名称不能为空')
|
||||
def get_table_name(self) -> str | None:
|
||||
return self.table_name
|
||||
|
||||
@NotBlank(field_name='table_comment', message='表描述不能为空')
|
||||
def get_table_comment(self) -> str | None:
|
||||
return self.table_comment
|
||||
|
||||
@NotBlank(field_name='class_name', message='实体类名称不能为空')
|
||||
def get_class_name(self) -> str | None:
|
||||
return self.class_name
|
||||
|
||||
@NotBlank(field_name='package_name', message='生成包路径不能为空')
|
||||
def get_package_name(self) -> str | None:
|
||||
return self.package_name
|
||||
|
||||
@NotBlank(field_name='module_name', message='生成模块名不能为空')
|
||||
def get_module_name(self) -> str | None:
|
||||
return self.module_name
|
||||
|
||||
@NotBlank(field_name='business_name', message='生成业务名不能为空')
|
||||
def get_business_name(self) -> str | None:
|
||||
return self.business_name
|
||||
|
||||
@NotBlank(field_name='function_name', message='生成功能名不能为空')
|
||||
def get_function_name(self) -> str | None:
|
||||
return self.function_name
|
||||
|
||||
@NotBlank(field_name='function_author', message='生成功能作者不能为空')
|
||||
def get_function_author(self) -> str | None:
|
||||
return self.function_author
|
||||
|
||||
def validate_fields(self) -> None:
|
||||
self.get_table_name()
|
||||
self.get_table_comment()
|
||||
self.get_class_name()
|
||||
self.get_package_name()
|
||||
self.get_module_name()
|
||||
self.get_business_name()
|
||||
self.get_function_name()
|
||||
self.get_function_author()
|
||||
|
||||
|
||||
class GenTableRowModel(GenTableBaseModel):
|
||||
"""
|
||||
代码生成业务表行数据模型
|
||||
"""
|
||||
|
||||
columns: list['GenTableColumnBaseModel'] | None = Field(default=None, description='表列信息')
|
||||
|
||||
|
||||
class GenTableDbRowModel(BaseModel):
|
||||
"""
|
||||
代码生成业务表数据库行数据模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel)
|
||||
|
||||
table_name: str | None = Field(default=None, description='表名称')
|
||||
table_comment: str | None = Field(default=None, description='表描述')
|
||||
create_time: datetime | None = Field(default=None, description='创建时间')
|
||||
update_time: datetime | None = Field(default=None, description='更新时间')
|
||||
|
||||
|
||||
class GenTableModel(GenTableBaseModel):
|
||||
"""
|
||||
代码生成业务表模型
|
||||
"""
|
||||
|
||||
pk_column: Optional['GenTableColumnModel'] = Field(default=None, description='主键信息')
|
||||
sub_table: Optional['GenTableModel'] = Field(default=None, description='子表信息')
|
||||
columns: list['GenTableColumnModel'] | None = Field(default=None, description='表列信息')
|
||||
tree_code: str | None = Field(default=None, description='树编码字段')
|
||||
tree_parent_code: str | None = Field(default=None, description='树父编码字段')
|
||||
tree_name: str | None = Field(default=None, description='树名称字段')
|
||||
parent_menu_id: int | None = Field(default=None, description='上级菜单ID字段')
|
||||
parent_menu_name: str | None = Field(default=None, description='上级菜单名称字段')
|
||||
sub: bool | None = Field(default=None, description='是否为子表')
|
||||
tree: bool | None = Field(default=None, description='是否为树表')
|
||||
crud: bool | None = Field(default=None, description='是否为单表')
|
||||
|
||||
@model_validator(mode='after')
|
||||
def check_some_is(self) -> 'GenTableModel':
|
||||
self.sub = bool(self.tpl_category and self.tpl_category == GenConstant.TPL_SUB)
|
||||
self.tree = bool(self.tpl_category and self.tpl_category == GenConstant.TPL_TREE)
|
||||
self.crud = bool(self.tpl_category and self.tpl_category == GenConstant.TPL_CRUD)
|
||||
return self
|
||||
|
||||
|
||||
class EditGenTableModel(GenTableModel):
|
||||
"""
|
||||
修改代码生成业务表模型
|
||||
"""
|
||||
|
||||
params: Optional['GenTableParamsModel'] = Field(default=None, description='业务表参数')
|
||||
|
||||
|
||||
class GenTableParamsModel(BaseModel):
|
||||
"""
|
||||
代码生成业务表参数模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel)
|
||||
|
||||
tree_code: str | None = Field(default=None, description='树编码字段')
|
||||
tree_parent_code: str | None = Field(default=None, description='树父编码字段')
|
||||
tree_name: str | None = Field(default=None, description='树名称字段')
|
||||
parent_menu_id: int | None = Field(default=None, description='上级菜单ID字段')
|
||||
|
||||
|
||||
class GenTableQueryModel(GenTableBaseModel):
|
||||
"""
|
||||
代码生成业务表不分页查询模型
|
||||
"""
|
||||
|
||||
begin_time: str | None = Field(default=None, description='开始时间')
|
||||
end_time: str | None = Field(default=None, description='结束时间')
|
||||
|
||||
|
||||
class GenTablePageQueryModel(GenTableQueryModel):
|
||||
"""
|
||||
代码生成业务表分页查询模型
|
||||
"""
|
||||
|
||||
page_num: int = Field(default=1, description='当前页码')
|
||||
page_size: int = Field(default=10, description='每页记录数')
|
||||
|
||||
|
||||
class GenTableDetailModel(BaseModel):
|
||||
"""
|
||||
代码生成业务表详情模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel)
|
||||
|
||||
info: GenTableModel | None = Field(default=None, description='业务表信息')
|
||||
rows: list['GenTableColumnModel'] | None = Field(default=None, description='表列信息')
|
||||
tables: list['GenTableModel'] | None = Field(default=None, description='所有业务表信息')
|
||||
|
||||
|
||||
class DeleteGenTableModel(BaseModel):
|
||||
"""
|
||||
删除代码生成业务表模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel)
|
||||
|
||||
table_ids: str = Field(description='需要删除的代码生成业务表ID')
|
||||
|
||||
|
||||
class GenTableColumnBaseModel(BaseModel):
|
||||
"""
|
||||
代码生成业务表字段对应pydantic模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
||||
|
||||
column_id: int | None = Field(default=None, description='编号')
|
||||
table_id: int | None = Field(default=None, description='归属表编号')
|
||||
column_name: str | None = Field(default=None, description='列名称')
|
||||
column_comment: str | None = Field(default=None, description='列描述')
|
||||
column_type: str | None = Field(default=None, description='列类型')
|
||||
python_type: str | None = Field(default=None, description='PYTHON类型')
|
||||
python_field: str | None = Field(default=None, description='PYTHON字段名')
|
||||
is_pk: str | None = Field(default=None, description='是否主键（1是）')
|
||||
is_increment: str | None = Field(default=None, description='是否自增（1是）')
|
||||
is_required: str | None = Field(default=None, description='是否必填（1是）')
|
||||
is_unique: str | None = Field(default=None, description='是否唯一（1是）')
|
||||
is_insert: str | None = Field(default=None, description='是否为插入字段（1是）')
|
||||
is_edit: str | None = Field(default=None, description='是否编辑字段（1是）')
|
||||
is_list: str | None = Field(default=None, description='是否列表字段（1是）')
|
||||
is_query: str | None = Field(default=None, description='是否查询字段（1是）')
|
||||
query_type: str | None = Field(default=None, description='查询方式（等于、不等于、大于、小于、范围）')
|
||||
html_type: str | None = Field(
|
||||
default=None, description='显示类型（文本框、文本域、下拉框、复选框、单选框、日期控件）'
|
||||
)
|
||||
dict_type: str | None = Field(default=None, description='字典类型')
|
||||
sort: int | None = Field(default=None, description='排序')
|
||||
create_by: str | None = Field(default=None, description='创建者')
|
||||
create_time: datetime | None = Field(default=None, description='创建时间')
|
||||
update_by: str | None = Field(default=None, description='更新者')
|
||||
update_time: datetime | None = Field(default=None, description='更新时间')
|
||||
|
||||
@NotBlank(field_name='python_field', message='Python属性不能为空')
|
||||
def get_python_field(self) -> str | None:
|
||||
return self.python_field
|
||||
|
||||
def validate_fields(self) -> None:
|
||||
self.get_python_field()
|
||||
|
||||
|
||||
class GenTableColumnModel(GenTableColumnBaseModel):
|
||||
"""
|
||||
代码生成业务表字段模型
|
||||
"""
|
||||
|
||||
cap_python_field: str | None = Field(default=None, description='字段大写形式')
|
||||
pk: bool | None = Field(default=None, description='是否主键')
|
||||
increment: bool | None = Field(default=None, description='是否自增')
|
||||
required: bool | None = Field(default=None, description='是否必填')
|
||||
unique: bool | None = Field(default=None, description='是否唯一')
|
||||
insert: bool | None = Field(default=None, description='是否为插入字段')
|
||||
edit: bool | None = Field(default=None, description='是否编辑字段')
|
||||
list: bool | None = Field(default=None, description='是否列表字段')
|
||||
query: bool | None = Field(default=None, description='是否查询字段')
|
||||
super_column: bool | None = Field(default=None, description='是否为基类字段')
|
||||
usable_column: bool | None = Field(default=None, description='是否为基类字段白名单')
|
||||
|
||||
@model_validator(mode='after')
|
||||
def check_some_is(self) -> 'GenTableModel':
|
||||
self.cap_python_field = self.python_field[0].upper() + self.python_field[1:] if self.python_field else None
|
||||
self.pk = self.is_pk and self.is_pk == '1'
|
||||
self.increment = bool(self.is_increment and self.is_increment == '1')
|
||||
self.required = bool(self.is_required and self.is_required == '1')
|
||||
self.unique = bool(self.is_unique and self.is_unique == '1')
|
||||
self.insert = bool(self.is_insert and self.is_insert == '1')
|
||||
self.edit = bool(self.is_edit and self.is_edit == '1')
|
||||
self.list = bool(self.is_list and self.is_list == '1')
|
||||
self.query = bool(self.is_query and self.is_query == '1')
|
||||
self.super_column = bool(
|
||||
StringUtil.equals_any_ignore_case(self.python_field, GenConstant.TREE_ENTITY + GenConstant.BASE_ENTITY)
|
||||
)
|
||||
self.usable_column = bool(
|
||||
StringUtil.equals_any_ignore_case(self.python_field, ['parentId', 'orderNum', 'remark'])
|
||||
)
|
||||
return self
|
||||
|
||||
|
||||
class GenTableColumnQueryModel(GenTableColumnBaseModel):
|
||||
"""
|
||||
代码生成业务表字段不分页查询模型
|
||||
"""
|
||||
|
||||
begin_time: str | None = Field(default=None, description='开始时间')
|
||||
end_time: str | None = Field(default=None, description='结束时间')
|
||||
|
||||
|
||||
class GenTableColumnPageQueryModel(GenTableColumnQueryModel):
|
||||
"""
|
||||
代码生成业务表字段分页查询模型
|
||||
"""
|
||||
|
||||
page_num: int = Field(default=1, description='当前页码')
|
||||
page_size: int = Field(default=10, description='每页记录数')
|
||||
|
||||
|
||||
class DeleteGenTableColumnModel(BaseModel):
|
||||
"""
|
||||
删除代码生成业务表字段模型
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(alias_generator=to_camel)
|
||||
|
||||
column_ids: str = Field(description='需要删除的代码生成业务表字段ID')
|
||||
Reference in New Issue
Block a user