功能变化: 菜单管理按钮配置优化

backend/dvadmin/system/views/menu_button.py

@@ -6,7 +6,11 @@
 @Created on: 2021/6/3 003 0:30
 @Remark: 菜单按钮管理
 """
-from dvadmin.system.models import MenuButton
+from django.db.models import F
+from rest_framework.decorators import action
+
+from dvadmin.system.models import MenuButton, RoleMenuButtonPermission
+from dvadmin.utils.json_response import DetailResponse
 from dvadmin.utils.serializers import CustomModelSerializer
 from dvadmin.utils.viewset import CustomModelViewSet
 
@@ -57,3 +61,19 @@ class MenuButtonViewSet(CustomModelViewSet):
     create_serializer_class = MenuButtonCreateUpdateSerializer
     update_serializer_class = MenuButtonCreateUpdateSerializer
     extra_filter_class = []
+
+    @action(methods=['get'],detail=False)
+    def menu_button_all_permission(self,request):
+        """
+        获取所有的按钮权限
+        :param request:
+        :return:
+        """
+        is_superuser = request.user.is_superuser
+        is_admin = request.user.role.values_list('admin', flat=True)
+        if is_superuser or True in is_admin:
+            queryset = MenuButton.objects.values_list('menu_button__value',flat=True)
+        else:
+            role_id = request.user.role.values_list('id', flat=True)
+            queryset = RoleMenuButtonPermission.objects.filter(role__in=role_id).values_list('menu_button__value',flat=True).distinct()
+        return DetailResponse(data=queryset)