一级菜单 二级菜单 三级菜单处理逻辑

一级二级三级菜单处理逻辑

    def get_role_permission(cls, permission_ids, is_admin=False):
        # 一次性全部把数据取出来
        permission_objs = Permission.objects.all()

        permission_datas = PermissionSerializer(permission_objs, many=True).data
        permission_datas_dict = {permission_data.get("id"): permission_data for permission_data in permission_datas}

        # 一级菜单
        first_menu_objs = list()
        first_ids = list()
        for permission_obj in permission_objs:
            if permission_obj.parent_id == 0:  # 父id为0的表示一级菜单
                permission_id = permission_obj.id
                first_data = permission_datas_dict.get(permission_id)
                first_menu_objs.append(first_data)  # 一级菜单的信息
                # 一级菜单的id
                first_ids.append(permission_id)

        # 二级菜单
        second_ids = list()
        second_parents_dict = {first_id: list() for first_id in first_ids}
        for permission_obj in permission_objs:
            parent_id = permission_obj.parent_id
            if parent_id in first_ids:  # 父id在一级菜单id列表里的为二级菜单
                permission_id = permission_obj.id
                second_data = permission_datas_dict.get(permission_id)
                second_list = second_parents_dict.get(parent_id)  # 根据父id获取二级list
                second_list.append(second_data)
                # 二级菜单的id
                second_ids.append(permission_id)

        for first_id in first_ids:  # 一级菜单存储二级菜单的数据
            first_data = permission_datas_dict.get(first_id)  # 取一级菜单的信息，根据一级父id
            first_data["child"] = second_parents_dict.get(first_id)  # 一级菜单存储二级菜单的数据

        # 三级菜单
        third_parents_dict = {second_id: list() for second_id in second_ids}
        for permission_obj in permission_objs:
            parent_id = permission_obj.parent_id
            if parent_id in second_ids:
                permission_id = permission_obj.id
                third_data = permission_datas_dict.get(permission_id)
                third_list = third_parents_dict.get(parent_id)
                if is_admin:
                    third_list.append(third_data)
                else:
                    if permission_id in permission_ids:
                        third_list.append(third_data)

        for second_id in second_ids:  # 二级菜单存储三级菜单的数据，根据二级父id
            second_data = permission_datas_dict.get(second_id)
            second_data["child"] = third_parents_dict.get(second_id)

        return first_menu_objs