init

backend/dvadmin/system/views/role.py

@@ -0,0 +1,98 @@
+# -*- coding: utf-8 -*-
+
+"""
+@author: 猿小天
+@contact: QQ:1638245306
+@Created on: 2021/6/3 003 0:30
+@Remark: 角色管理
+"""
+from rest_framework import serializers
+from rest_framework.decorators import action
+from rest_framework.permissions import IsAuthenticated
+
+from dvadmin.system.models import Role, Menu, MenuButton, Dept
+from dvadmin.system.views.dept import DeptSerializer
+from dvadmin.system.views.menu import MenuSerializer
+from dvadmin.system.views.menu_button import MenuButtonSerializer
+from dvadmin.utils.crud_mixin import FastCrudMixin
+from dvadmin.utils.json_response import SuccessResponse, DetailResponse
+from dvadmin.utils.serializers import CustomModelSerializer
+from dvadmin.utils.validator import CustomUniqueValidator
+from dvadmin.utils.viewset import CustomModelViewSet
+
+
+class RoleSerializer(CustomModelSerializer):
+    """
+    角色-序列化器
+    """
+
+    class Meta:
+        model = Role
+        fields = "__all__"
+        read_only_fields = ["id"]
+
+
+
+
+
+class RoleCreateUpdateSerializer(CustomModelSerializer):
+    """
+    角色管理 创建/更新时的列化器
+    """
+    menu = MenuSerializer(many=True, read_only=True)
+    dept = DeptSerializer(many=True, read_only=True)
+    permission = MenuButtonSerializer(many=True, read_only=True)
+    key = serializers.CharField(max_length=50,
+                                validators=[CustomUniqueValidator(queryset=Role.objects.all(), message="权限字符必须唯一")])
+    name = serializers.CharField(max_length=50, validators=[CustomUniqueValidator(queryset=Role.objects.all())])
+
+    def validate(self, attrs: dict):
+        return super().validate(attrs)
+
+    def save(self, **kwargs):
+        is_superuser = self.request.user.is_superuser
+        if not is_superuser:
+            self.validated_data.pop('admin')
+        data = super().save(**kwargs)
+        return data
+
+    class Meta:
+        model = Role
+        fields = '__all__'
+
+
+class MenuPermissonSerializer(CustomModelSerializer):
+    """
+    菜单的按钮权限
+    """
+    menuPermission = serializers.SerializerMethodField()
+
+    def get_menuPermission(self, instance):
+        is_superuser = self.request.user.is_superuser
+        if is_superuser:
+            queryset = MenuButton.objects.filter(menu__id=instance.id)
+        else:
+            menu_permission_id_list = self.request.user.role.values_list('permission',flat=True)
+            queryset = MenuButton.objects.filter(id__in=menu_permission_id_list,menu__id=instance.id)
+        serializer = MenuButtonSerializer(queryset,many=True, read_only=True)
+        return serializer.data
+
+    class Meta:
+        model = Menu
+        fields = ['id', 'parent', 'name', 'menuPermission']
+
+
+class RoleViewSet(CustomModelViewSet,FastCrudMixin):
+    """
+    角色管理接口
+    list:查询
+    create:新增
+    update:修改
+    retrieve:单例
+    destroy:删除
+    """
+    queryset = Role.objects.all()
+    serializer_class = RoleSerializer
+    create_serializer_class = RoleCreateUpdateSerializer
+    update_serializer_class = RoleCreateUpdateSerializer
+    search_fields = ['name', 'key']