修改info 获取权限及菜单

backend/system/views/menu.py

@@ -69,7 +69,7 @@ class MenuMetaViewSet(viewsets.ModelViewSet):
 
 class MenuViewSet(CustomModelViewSet):
     """菜单管理视图集"""
-    queryset = Menu.objects.filter(pid__isnull=True).order_by('id', 'status')
+    queryset = Menu.objects.filter(pid__isnull=True).order_by('sort', 'id', 'status')
     serializer_class = MenuSerializer
     filter_backends = [DjangoFilterBackend, SearchFilter, OrderingFilter]
     filterset_fields = ['status', 'type', 'pid', 'name']
@@ -91,7 +91,10 @@ class MenuViewSet(CustomModelViewSet):
     def name_search(self, request):
         name = request.GET.get('name')
         pk = request.GET.get('id', None)
+        pid = request.GET.get('pid', None)
         queryset = Menu.objects.all()
+        if pid:
+            queryset = queryset.filter(pid=pid)
         if pk:
             queryset = queryset.exclude(pk=pk)
         if name:

backend/system/views/user.py

@@ -6,7 +6,8 @@ from rest_framework.response import Response
 from rest_framework.views import APIView
 from django.contrib.auth.hashers import make_password
 
-from system.models import User
+from system.models import User, Menu
+from system.views.menu import MenuSerializer
 
 from utils.serializers import CustomModelSerializer
 from utils.custom_model_viewSet import CustomModelViewSet
@@ -68,10 +69,22 @@ class UserInfo(APIView):
         user = self.request.user
         user_data = UserSerializer(user).data
         if user.is_superuser:
-            user_data['roles'] = ['admin']
+            roles = ['admin']
+            menus = Menu.objects.filter(pid__isnull=True).order_by('sort')
+            permissions = Menu.objects.filter(type='button').order_by('sort').values_list('auth_code', flat=True)
+        else:
+            roles = user.get_role_name
+            menus = Menu.objects.filter(pid__isnull=True, role__users=user).order_by('sort').distinct()
+            permissions = Menu.objects.filter(type='button', role__users=user).order_by('sort').distinct().values_list('auth_code', flat=True)
+        menus_data = MenuSerializer(menus, many=True).data
         return Response({
             "code": 0,
-            "data": user_data,
+            "data": {
+                "menus": menus_data,
+                "permissions": permissions,
+                "roles": roles,
+                "user": user_data,
+            },
             "error": None,
             "message": "ok"
         })