Better parent-child relationship

[downdawn]

An example

class Department(Base):
    """ 部门 """
    __tablename__ = 'sys_department'

    id: Mapped[id_key] = mapped_column(init=False)
    name: Mapped[str] = mapped_column(String(32), unique=True, comment='部门名称')
    sort: Mapped[int] = mapped_column(comment='排序')
    status: Mapped[int] = mapped_column(default=1, comment='状态')
    parent_id: Mapped[int | None] = mapped_column(ForeignKey('sys_department.id'), default=None, comment='父级部门id')

    def as_dict(self):
        return {c.name: getattr(self, c.name) for c in self.__mapper__.columns}

    async def get_dept_tree():
        async with async_db_session() as db:
            dept_list = await DeptDao.get_all_dept(db)
            return get_descendant(dept_list, None)

    async def get_all_dept(self, db: AsyncSession):
        data = await db.execute(select(self.model).order_by(self.model.sort))
        return data.scalars().all()

def get_descendant(nodes, parent_id):
    """递归获取子级"""
    result = []
    for node in nodes:
        if node.parent_id == parent_id:
            child = get_descendant(nodes, node.id)
            _data = node.as_dict()
            if child and child is not None:
                _data['children'] = child
            result.append(_data)
    if result:
        result = sorted(result, key=lambda x: x.get('sort'))
        return result

But the best should still be set in the model, it may be so, no further exploration

children: Mapped[list['Department']] = relationship('Department', init=False, backref='parent', passive_deletes=True)

[wu-clan]

Work in progress #101.

[wu-clan]

Now available for review