FastapiAdmin/backend/app/api/v1/module_generator/gencode/model.py

# -*- coding: utf-8 -*-

from typing import Optional, List
from sqlalchemy import String, Integer, ForeignKey, Boolean
from sqlalchemy.orm import Mapped, mapped_column, relationship, validates
from sqlalchemy.sql import expression

from app.config.setting import settings
from app.core.base_model import CreatorMixin
from app.utils.common_util import SqlalchemyUtil


class GenTableModel(CreatorMixin):
    """
    代码生成表
    """
    __tablename__ = 'gen_table'
    __table_args__ = ({'comment': '代码生成表'})
    __loader_options__ = ["columns", "creator"]

    table_name: Mapped[str] = mapped_column(String(200), nullable=False, default='', comment='表名称')
    table_comment: Mapped[Optional[str]] = mapped_column(String(500), nullable=True, comment='表描述')
    
    class_name: Mapped[str] = mapped_column(String(100), nullable=False, default='', comment='实体类名称')
    package_name: Mapped[Optional[str]] = mapped_column(String(100), nullable=True, comment='生成包路径')
    module_name: Mapped[Optional[str]] = mapped_column(String(30), nullable=True, comment='生成模块名')
    business_name: Mapped[Optional[str]] = mapped_column(String(30), nullable=True, comment='生成业务名')
    function_name: Mapped[Optional[str]] = mapped_column(String(100), nullable=True, comment='生成功能名')
    
    sub_table_name: Mapped[Optional[str]] = mapped_column(String(64), nullable=True, server_default=SqlalchemyUtil.get_server_default_null(settings.DATABASE_TYPE), comment='关联子表的表名')
    sub_table_fk_name: Mapped[Optional[str]] = mapped_column(String(64), nullable=True, server_default=SqlalchemyUtil.get_server_default_null(settings.DATABASE_TYPE), comment='子表关联的外键名')
    
    parent_menu_id: Mapped[Optional[int]] = mapped_column(Integer, nullable=True, comment='父菜单ID')
    
    # 关系定义
    columns: Mapped[List['GenTableColumnModel']] = relationship('GenTableColumnModel', order_by='GenTableColumnModel.sort', back_populates='table',cascade='all, delete-orphan')
    
    @validates('table_name')
    def validate_table_name(self, key: str, table_name: str) -> str:
        """验证表名不为空"""
        if not table_name or not table_name.strip():
            raise ValueError('表名称不能为空')
        return table_name.strip()
    
    @validates('class_name')
    def validate_class_name(self, key: str, class_name: str) -> str:
        """验证类名不为空"""
        if not class_name or not class_name.strip():
            raise ValueError('实体类名称不能为空')
        return class_name.strip()


class GenTableColumnModel(CreatorMixin):
    """
    代码生成表字段
    """
    __tablename__ = 'gen_table_column'
    __table_args__ = ({'comment': '代码生成表字段'})
    __loader_options__ = ["table", "creator"]

    # 数据库设计表字段
    column_name: Mapped[str] = mapped_column(String(200), nullable=False, comment='列名称')
    column_comment: Mapped[Optional[str]] = mapped_column(String(500), nullable=True, comment='列描述')
    column_type: Mapped[str] = mapped_column(String(100), nullable=False, comment='列类型')
    column_length: Mapped[Optional[str]] = mapped_column(String(50), nullable=True, comment='列长度')
    column_default: Mapped[Optional[str]] = mapped_column(String(200), nullable=True, comment='列默认值')
    is_pk: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, server_default=expression.false(), comment='是否主键')
    is_increment: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, server_default=expression.false(), comment='是否自增')
    is_nullable: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default=expression.true(), comment='是否允许为空')
    is_unique: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, server_default=expression.false(), comment='是否唯一')
    
    # Python字段映射
    python_type: Mapped[Optional[str]] = mapped_column(String(100), nullable=True, comment='Python类型')
    python_field: Mapped[Optional[str]] = mapped_column(String(200), nullable=True, comment='Python字段名')
    
    # 序列化配置
    is_insert: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default=expression.true(), comment='是否为新增字段')
    is_edit: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default=expression.true(), comment='是否编辑字段')
    is_list: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default=expression.true(), comment='是否列表字段')
    is_query: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, server_default=expression.false(), comment='是否查询字段')
    query_type: Mapped[Optional[str]] = mapped_column(String(50), nullable=True, default=None, comment='查询方式')
    
    # 前端展示配置
    html_type: Mapped[Optional[str]] = mapped_column(String(100), nullable=True, default='input', comment='显示类型')
    dict_type: Mapped[Optional[str]] = mapped_column(String(200), nullable=True, default='', comment='字典类型')
    
    # 排序和扩展配置
    sort: Mapped[int] = mapped_column(Integer, nullable=False, default=0, comment='排序')
    
    # 归属关系
    table_id: Mapped[int] = mapped_column(
        Integer, 
        ForeignKey('gen_table.id', ondelete='CASCADE'), 
        nullable=False, 
        index=True,
        comment='归属表编号'
    )

    # 关系定义
    table: Mapped['GenTableModel'] = relationship('GenTableModel', back_populates='columns')
    
    @validates('column_name')
    def validate_column_name(self, key: str, column_name: str) -> str:
        """验证列名不为空"""
        if not column_name or not column_name.strip():
            raise ValueError('列名称不能为空')
        return column_name.strip()
    
    @validates('column_type')
    def validate_column_type(self, key: str, column_type: str) -> str:
        """验证列类型不为空"""
        if not column_type or not column_type.strip():
            raise ValueError('列类型不能为空')
        return column_type.strip()