from sqlalchemy import Column, Integer, String

from domain.models.base_model import Base, FullModelBase, OrderModelBase


class SysRoleModel(FullModelBase, OrderModelBase):
    """
    角色信息表
    """

    __tablename__ = "sys_role"

    role_name = Column(String(30), nullable=False, comment="角色名称")
    role_key = Column(String(100), nullable=False, comment="角色权限字符串")
    order_num = Column(Integer, nullable=False, comment="显示顺序")
    data_scope = Column(
        Integer,
        nullable=False,
        default=5,
        comment="数据范围（1：全部数据权限 2：自定数据权限 3：本部门数据权限 4：本部门及以下数据权限 5:自己的数据）",
    )
    menu_check_strictly = Column(Integer, default=1, comment="菜单树选择项是否关联显示")
    dept_check_strictly = Column(Integer, default=1, comment="部门树选择项是否关联显示")
    status = Column(
        Integer, nullable=False, default=0, comment="角色状态（0正常 1停用）"
    )
    remark = Column(String(500), default=None, comment="备注")


class SysRoleDeptModel(Base):
    """
    角色和部门关联表
    """

    __tablename__ = "sys_role_dept"

    role_id = Column(Integer, primary_key=True, nullable=False, comment="角色ID")
    dept_id = Column(Integer, primary_key=True, nullable=False, comment="部门ID")