init

backend/dvadmin/system/views/dept.py

@@ -0,0 +1,157 @@
+# -*- coding: utf-8 -*-
+
+"""
+@author: H0nGzA1
+@contact: QQ:2505811377
+@Remark: 部门管理
+"""
+from rest_framework import serializers
+from rest_framework.decorators import action
+from rest_framework.permissions import IsAuthenticated
+
+from dvadmin.system.models import Dept, RoleMenuButtonPermission
+from dvadmin.utils.json_response import DetailResponse, SuccessResponse
+from dvadmin.utils.permission import AnonymousUserPermission
+from dvadmin.utils.serializers import CustomModelSerializer
+from dvadmin.utils.viewset import CustomModelViewSet
+
+
+class DeptSerializer(CustomModelSerializer):
+    """
+    部门-序列化器
+    """
+    parent_name = serializers.CharField(read_only=True, source='parent.name')
+    status_label = serializers.SerializerMethodField()
+    has_children = serializers.SerializerMethodField()
+    hasChild = serializers.SerializerMethodField()
+
+    def get_hasChild(self, instance):
+        hasChild = Dept.objects.filter(parent=instance.id)
+        if hasChild:
+            return True
+        return False
+
+    def get_status_label(self, obj: Dept):
+        if obj.status:
+            return "启用"
+        return "禁用"
+
+    def get_has_children(self, obj: Dept):
+        return Dept.objects.filter(parent_id=obj.id).count()
+
+    class Meta:
+        model = Dept
+        fields = '__all__'
+        read_only_fields = ["id"]
+
+
+class DeptImportSerializer(CustomModelSerializer):
+    """
+    部门-导入-序列化器
+    """
+
+    class Meta:
+        model = Dept
+        fields = '__all__'
+        read_only_fields = ["id"]
+
+
+
+
+
+class DeptCreateUpdateSerializer(CustomModelSerializer):
+    """
+    部门管理 创建/更新时的列化器
+    """
+
+    def create(self, validated_data):
+        value = validated_data.get('parent',None)
+        if value is None:
+            validated_data['parent'] = self.request.user.dept
+        instance = super().create(validated_data)
+        instance.dept_belong_id = instance.id
+        instance.save()
+        return instance
+
+    class Meta:
+        model = Dept
+        fields = '__all__'
+
+
+class DeptViewSet(CustomModelViewSet):
+    """
+    部门管理接口
+    list:查询
+    create:新增
+    update:修改
+    retrieve:单例
+    destroy:删除
+    """
+    queryset = Dept.objects.all()
+    serializer_class = DeptSerializer
+    create_serializer_class = DeptCreateUpdateSerializer
+    update_serializer_class = DeptCreateUpdateSerializer
+    filter_fields = ['name', 'id', 'parent']
+    search_fields = []
+    # extra_filter_class = []
+    import_serializer_class = DeptImportSerializer
+    import_field_dict = {
+        "name": "部门名称",
+        "key": "部门标识",
+    }
+
+    def list(self, request, *args, **kwargs):
+        # 如果懒加载，则只返回父级
+        request.query_params._mutable = True
+        params = request.query_params
+        parent = params.get('parent', None)
+        page = params.get('page', None)
+        limit = params.get('limit', None)
+        if page:
+            del params['page']
+        if limit:
+            del params['limit']
+        if params:
+            if parent:
+                queryset = self.queryset.filter(status=True, parent=parent)
+            else:
+                queryset = self.queryset.filter(status=True)
+        else:
+            queryset = self.queryset.filter(status=True, parent__isnull=True)
+        queryset = self.filter_queryset(queryset)
+        serializer = DeptSerializer(queryset, many=True, request=request)
+        data = serializer.data
+        return SuccessResponse(data=data)
+
+    @action(methods=["GET"], detail=False, permission_classes=[IsAuthenticated], extra_filter_class=[])
+    def dept_lazy_tree(self, request, *args, **kwargs):
+        parent = self.request.query_params.get('parent')
+        is_superuser = request.user.is_superuser
+        if is_superuser:
+            queryset = Dept.objects.values('id', 'name', 'parent')
+        else:
+            role_ids = request.user.role.values_list('id',flat=True)
+            data_range = RoleMenuButtonPermission.objects.filter(role__in=role_ids).values_list('data_range', flat=True)
+            user_dept_id = request.user.dept.id
+            dept_list = [user_dept_id]
+            data_range_list = list(set(data_range))
+            for item in data_range_list:
+                if item in [0,2]:
+                    dept_list = [user_dept_id]
+                elif item == 1:
+                    dept_list = Dept.recursion_dept_info(dept_id=user_dept_id)
+                elif item == 3:
+                    dept_list = Dept.objects.values_list('id',flat=True)
+                elif item == 4:
+                    dept_list = request.user.role.values_list('dept',flat=True)
+                else:
+                    dept_list = []
+            queryset = Dept.objects.filter(id__in=dept_list).values('id', 'name', 'parent')
+        return DetailResponse(data=queryset, msg="获取成功")
+
+
+    @action(methods=["GET"], detail=False, permission_classes=[IsAuthenticated],extra_filter_class=[])
+    def all_dept(self, request, *args, **kwargs):
+        queryset = self.filter_queryset(self.get_queryset())
+        data = queryset.filter(status=True).order_by('sort').values('name', 'id', 'parent')
+        return DetailResponse(data=data, msg="获取成功")