Python自关联不用递归查询实现

根据ID查询所有的上级部门ID

def _get_parent_department_id(self, db, d_id):
    """
    根据ID查询所有的上级部门ID(依赖逻辑: 子部门的created_at必定大于其父部门的created_at)
    :param d_id 部门id:
    :return department_ids 所有上级部门id:
    """
    if d_id is None or d_id == "":
        return []
    department_ids = []
    parent_id = d_id
    all_department_list = db.query(DepartmentModel).order_by(DepartmentModel.created_at.desc()).all()
    for item in all_department_list:
        if item.id == parent_id:
            if item.parent_id == "":
                # 已找到最上层的顶级部门，结束循环
                break
            # 将当前部门的父部门id加入列表，并继续寻找父部门的父部门
            department_ids.append(item.parent_id)
            parent_id = item.parent_id
    return department_ids

根据ID查询所有的子孙级部门ID

def _get_child_department_id(self, db, d_id):
    """
    根据ID查询所有的子孙级部门ID
    :param d_id:
    :param db:
    :return:
    """
    child_department_ids = []
    all_department_list = db.query(DepartmentModel).order_by(DepartmentModel.created_at.asc())
    for item in all_department_list:
        if item.parent_id == d_id or item.parent_id in child_department_ids:
            child_department_ids.append(item.id)
    return child_department_ids