ERP6/ERP_5/apps/erp_system/views/permission_views.py

from drf_yasg import openapi
from drf_yasg.utils import swagger_auto_schema
from rest_framework.decorators import action
from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet

from erp_system.models import PermissionsModel, RolesModel
from erp_system.serializers.permission_serializer import PermissionsSerializer

import logging

logger = logging.getLogger('my')

class PermissionViewSet(ModelViewSet):
    """
     create:
    权限--新增

    权限新增, status: 201(成功), return: 新增权限信息

    destroy:
    权限--删除

    权限删除, status: 204(成功), return: None

    multiple_delete:
    权限--批量删除,必传参数：ids=[1,2,3,4...]

    权限批量删除, status: 204(成功), return: None

    update:
    权限--修改

    权限修改, status: 200(成功), return: 修改增权限信息

    partial_update:
    权限--局部修改

    权限局部修改, status: 200(成功), return: 修改增权限信息

    list:
    权限--获取所有的权限

    find_permissions_by_menu:
    获取当前菜单的所有权限设置 列表,必传参数：menu_id

    例如：http://127.0.0.1:8000/permissions/find_permissions_by_menu/?menu_id=2
    权限列表信息, status: 200(成功), return: 权限信息列表

    find_permissions:
    打开授权页面的时候，首先请求的接口；获取当前角色已经授予的权限ID列表和整个项目的所有权限(树形)列表,必传参数：rid(角色ID)

    例如：http://127.0.0.1:8000/permissions/find_permissions/?rid=2
    树形权限列表信息, status: 200(成功), return: 权限信息列表
    retrieve:
    查询某一个指定的权限信息

    查询指定ID的权限, status: 200(成功), return: 权限信息

    """

    queryset = PermissionsModel.objects.all()
    serializer_class = PermissionsSerializer

    # 定义好了一个swagger的参数
    menu_param = openapi.Parameter(name='menu_id', in_=openapi.IN_QUERY, description='所选的菜单ID',
                                   type=openapi.TYPE_INTEGER)

    @swagger_auto_schema(method='GET', manual_parameters=[menu_param], operation_description='单一授权或者取消单一权限')
    @action(methods=['GET'], detail=False)
    def find_permissions_by_menu(self, request, *args, **kwargs):
        menu_id = request.query_params.get('menu_id')
        permission_list = PermissionsModel.objects.filter(menu__id=menu_id).all()
        return Response(data=self.get_serializer(instance=permission_list, many=True).data)

    # 定义好了一个swagger的参数
    menu_param2 = openapi.Parameter(name='rid', in_=openapi.IN_QUERY, description='所选的角色ID',
                                    type=openapi.TYPE_INTEGER)

    @swagger_auto_schema(method='GET', manual_parameters=[menu_param2])
    @action(methods=['GET'], detail=False)
    def find_permissions(self, request, *args, **kwargs):
        #  获取当前角色已经授予的权限ID列表和整个项目的所有权限(树形)列表
        result = {}  # 返回的整个字典数据，
        rid = request.query_params.get('rid', None)
        if rid:  # 查询当前角色的所有已经授权的ID
            ids = RolesModel.objects.filter(id=rid).values_list('permissions', flat=True).distinct()
            result['ids'] = ids  # 存入返回字典中

        # 整个ERP系统中，所有权限列表，而且要树形展示
        permissions_list = PermissionsModel.objects.values('id', 'name', 'menu__parent__id', 'menu__name', 'menu__id')
        tree_dict = {}  # 一个父节点权限（里面是树形嵌套）
        tree_data = []  # 所有权限 ，
        """
        [
            {父节点1的内容，children=[{子节点内容}，{子节点内容}...]},
            {父节点2的内容，children=[{子节点内容}，{子节点内容}...]},
            ...
        ]
        """
        # 第一步构建一颗二级树
        for item in permissions_list:
            tree_dict[item['menu__id']] = item
        for i in tree_dict:
            if tree_dict[i]['menu__parent__id']:  # 表示子权限
                pid = tree_dict[i]['menu__parent__id']
                parent = tree_dict[pid]
                child = dict()  # 创建一个空字典(二级)，第二级只要：menu__id, menu__name, permissions(权限列表)
                child['menu__id'] = tree_dict[i]['menu__id']
                child['menu__name'] = tree_dict[i]['menu__name']
                child.setdefault('permissions', [])
                parent.setdefault('children', []).append(child)
            else:
                tree_data.append(tree_dict[i])

        # 第二步：在前面二级树形结构的基础上，添加子权限。得到三级树
        for parent in tree_data:
            if 'children' in parent:
                for child in parent['children']:
                    for node in permissions_list:
                        if node['menu__parent__id'] and node['menu__id'] == child['menu__id']:
                            child['permissions'].append(node)
        logger.info(tree_data)
        result['tree_data'] = tree_data  # 把三级树结构，加入到结果中
        return Response(result)