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

from django.db.models import Q
from rest_framework import status
from rest_framework.decorators import action
from rest_framework.permissions import IsAuthenticated
from rest_framework.response import Response
from rest_framework.viewsets import ModelViewSet
from erp_system.tasks import create_menus_permissions
from erp_system.models import Menu, PermissionsModel
from erp_system.serializers.menus_serializer import MenuSerializer


class MenuViewSet(ModelViewSet):
    """
    create:
    功能菜单--新增

    功能菜单新增, status: 201(成功), return: 新增功能菜单信息

    destroy:
    功能菜单--删除

    功能菜单删除, status: 204(成功), return: None

    update:
    功能菜单--修改,全局修改

    功能菜单修改, status: 200(成功), return: 修改后的功能菜单信息

    partial_update:
    功能菜单--局部修改

    功能菜单批修改 status: 200(成功), return: 修改后的功能菜单信息

    list:
    功能菜单--获取列表

    功能菜单列表信息, status: 200(成功), return: 功能菜单信息列表

    retrieve:
    查询某一个功能菜单

    查询指定ID的功能菜单, status: 200(成功), return: 用户功能菜单

    find_parent:
    查询所有的顶级功能菜单， 不需要传参。

    get_menus_by_permission:
    查询当前登录的用户，所有的，拥有GET权限的菜单列表，而且这些菜单列表必须是树形结构

    返回树形菜单列表, status: 200(成功)
    """
    serializer_class = MenuSerializer
    queryset = Menu.objects.all()
    # 必须是登录成功之后才能访问所有功能菜单的接口
    # permission_classes = [IsAuthenticated]

    @action(methods=['get'], detail=False)
    def find_parent(self, request, *args, **kwargs):
        lst = self.get_queryset().filter(parent__isnull=True)
        ser = self.get_serializer(instance=lst, many=True)
        return Response(ser.data)

    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        menu = serializer.save()
        # 发布异步的任务
        if menu and menu.id:
            create_menus_permissions.delay(menu.id)
        headers = self.get_success_headers(serializer.data)
        return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)

    @action(methods=['get'], detail=False)
    def get_menus_by_permission(self, request, *args, **kwargs):
        # 得到当前登录用户
        user = request.user
        # 查询该用户所拥有的权限ID列表
        permission_id = user.roles.values_list('permissions', flat=True).distinct()
        # 过滤那些GET的，或者是顶级菜单权限
        menu_ids = PermissionsModel.objects.filter(Q(method='GET') | Q(is_menu=True)).filter(id__in=permission_id)\
            .values_list('menu', flat=True)

        # 根据menu_id得到菜单接口对象
        menu_list = Menu.objects.filter(id__in=menu_ids, delete_flag=0).all()

        serializer = MenuSerializer(instance=menu_list, many=True)
        # 定义 Element
        tree_dict = {}  # 代表一个父菜单
        tree_data = []
        try:
            for item in serializer.data:
                tree_dict[item['id']] = item
            for i in tree_dict:
                if tree_dict[i]['parent']:
                    pid = tree_dict[i]['parent']
                    parent = tree_dict[pid]
                    parent.setdefault('children', []).append(tree_dict[i])
                else:
                    tree_data.append(tree_dict[i])
            results = tree_data
        except KeyError:
            results = serializer.data
        return Response(results)