import sqlalchemy as sa

from sqlalchemy.orm import Mapped, mapped_column

from backend.common.model import DataClassBase, UniversalText, id_key


class GenColumn(DataClassBase):
    """代码生成模型列表"""

    __tablename__ = 'gen_column'

    id: Mapped[id_key] = mapped_column(init=False)
    name: Mapped[str] = mapped_column(sa.String(64), comment='列名称')
    comment: Mapped[str | None] = mapped_column(sa.String(256), default=None, comment='列描述')
    type: Mapped[str] = mapped_column(sa.String(32), default='String', comment='SQLA 模型列类型')
    pd_type: Mapped[str] = mapped_column(sa.String(32), default='str', comment='列类型对应的 pydantic 类型')
    default: Mapped[str | None] = mapped_column(UniversalText, default=None, comment='列默认值')
    sort: Mapped[int | None] = mapped_column(default=1, comment='列排序')
    length: Mapped[int] = mapped_column(default=0, comment='列长度')
    is_pk: Mapped[bool] = mapped_column(default=False, comment='是否主键')
    is_nullable: Mapped[bool] = mapped_column(default=False, comment='是否可为空')

    # 逻辑外键
    gen_business_id: Mapped[int] = mapped_column(sa.BigInteger, default=0, comment='代码生成业务ID')