from pear_admin.extensions import db

from ._base import BaseORM


class DepartmentORM(BaseORM):
    __tablename__ = "ums_department"
    id = db.Column(db.Integer, primary_key=True, comment="部门 ID")
    name = db.Column(db.String(50), comment="部门名称")
    leader = db.Column(db.String(50), comment="负责人")
    enable = db.Column(db.Boolean, comment="状态 (1 开启，0 关闭)", default=True)

    users = db.relationship("UserORM", backref="department")

    pid = db.Column(db.Integer, db.ForeignKey("ums_department.id"), default=1)
    parent = db.relationship(
        "DepartmentORM", back_populates="children", remote_side=[id]
    )  # 自关联
    children = db.relationship("DepartmentORM", back_populates="parent")