from sqlalchemy import Boolean, ForeignKey, Integer, String, Text, UniqueConstraint
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.core.base_model import ModelMixin, TenantMixin


class DictTypeModel(ModelMixin, TenantMixin):
    """
    字典类型表

    __platform_data_shared__ = True 表示 tenant_id=1 的平台字典对
    所有租户可读，但只有平台管理员可写。
    """

    __tablename__: str = "sys_dict_type"
    __table_args__ = (UniqueConstraint("tenant_id", "dict_type"), {"comment": "字典类型表"})
    __loader_options__: list[str] = ["dict_data_list"]
    __platform_data_shared__: bool = True

    dict_name: Mapped[str] = mapped_column(String(64), nullable=False, comment="字典名称")
    dict_type: Mapped[str] = mapped_column(String(255), nullable=False, comment="字典类型")
    status: Mapped[int] = mapped_column(Integer, default=0, nullable=False, comment="状态(0:启动 1:停用)", index=True)
    description: Mapped[str | None] = mapped_column(Text, default=None, nullable=True, comment="备注")
    dict_data_list: Mapped[list["DictDataModel"]] = relationship(
        "DictDataModel",
        back_populates="dict_type_obj",
        cascade="all, delete-orphan",
    )


class DictDataModel(ModelMixin, TenantMixin):
    """
    字典数据表

    与 DictTypeModel 相同：tenant_id=1 的平台字典数据对
    所有租户可读，但只有平台管理员可写。
    """

    __tablename__: str = "sys_dict_data"
    __table_args__ = (
        UniqueConstraint("tenant_id", "dict_type_id", "dict_value", name="uq_dict_data_value"),
        {"comment": "字典数据表"},
    )
    __loader_options__: list[str] = ["dict_type_obj"]
    __platform_data_shared__: bool = True

    status: Mapped[int] = mapped_column(Integer, default=0, nullable=False, comment="状态(0:启动 1:停用)", index=True)
    description: Mapped[str | None] = mapped_column(Text, default=None, nullable=True, comment="备注")
    dict_sort: Mapped[int] = mapped_column(Integer, nullable=False, default=0, comment="字典排序")
    dict_label: Mapped[str] = mapped_column(String(255), nullable=False, comment="字典标签")
    dict_value: Mapped[str] = mapped_column(String(255), nullable=False, comment="字典键值")
    css_class: Mapped[str | None] = mapped_column(String(255), nullable=True, comment="样式属性（其他样式扩展）")
    list_class: Mapped[str | None] = mapped_column(String(255), nullable=True, comment="表格回显样式")
    is_default: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, comment="是否默认(True是 False否)")
    dict_type: Mapped[str] = mapped_column(String(255), nullable=False, comment="字典类型")

    # 添加外键关系，同时保留dict_type字段用于业务查询
    dict_type_id: Mapped[int] = mapped_column(
        Integer,
        ForeignKey("sys_dict_type.id", ondelete="CASCADE"),
        nullable=False,
        comment="字典类型ID",
    )

    # 关系定义
    dict_type_obj: Mapped[DictTypeModel] = relationship("DictTypeModel", back_populates="dict_data_list")