# -*- 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, Users from dvadmin.utils.filters import DataLevelPermissionsFilter from dvadmin.utils.json_response import DetailResponse, SuccessResponse, ErrorResponse 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() dept_user_count = serializers.SerializerMethodField() def get_dept_user_count(self, obj: Dept): return Users.objects.filter(dept=obj).count() 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 dept_obj = Dept.objects.filter(parent=self.request.user.dept).order_by('-sort').first() last_sort = dept_obj.sort if dept_obj else 0 validated_data['sort'] = last_sort + 1 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 and parent: queryset = self.queryset.filter(status=True, parent=parent) else: queryset = self.queryset.filter(status=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]) 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="获取成功") @action(methods=['POST'], detail=False, permission_classes=[IsAuthenticated]) def move_up(self, request): """部门上移""" dept_id = request.data.get('dept_id') try: dept = Dept.objects.get(id=dept_id) except Dept.DoesNotExist: return ErrorResponse(msg="部门不存在") previous_menu = Dept.objects.filter(sort__lt=dept.sort, parent=dept.parent).order_by('-sort').first() if previous_menu: previous_menu.sort, dept.sort = dept.sort, previous_menu.sort previous_menu.save() dept.save() return SuccessResponse(data=[], msg="上移成功") @action(methods=['POST'], detail=False, permission_classes=[IsAuthenticated]) def move_down(self, request): """部门下移""" dept_id = request.data['dept_id'] try: dept = Dept.objects.get(id=dept_id) except Dept.DoesNotExist: return ErrorResponse(msg="部门不存在") next_menu = Dept.objects.filter(sort__gt=dept.sort, parent=dept.parent).order_by('sort').first() if next_menu: next_menu.sort, dept.sort = dept.sort, next_menu.sort next_menu.save() dept.save() return SuccessResponse(data=[], msg="下移成功") @action(methods=['GET'], detail=False, permission_classes=[]) def dept_info(self, request): """部门信息""" def inner(did, li): sub = Dept.objects.filter(parent_id=did) if not sub.exists(): return li for i in sub: li.append(i.pk) inner(i, li) return li dept_id = request.query_params.get('dept_id') show_all = request.query_params.get('show_all') if dept_id is None: return ErrorResponse(msg="部门不存在") if not show_all: show_all = 0 if int(show_all): # 递归当前部门下的所有部门，查询用户 all_did = [dept_id] inner(dept_id, all_did) users = Users.objects.filter(dept_id__in=all_did) else: if dept_id != '': users = Users.objects.filter(dept_id=dept_id) else: users = Users.objects.none() dept_obj = Dept.objects.get(id=dept_id) if dept_id != '' else None sub_dept = Dept.objects.filter(parent_id=dept_obj.pk) if dept_id != '' else [] data = { 'dept_name': dept_obj and dept_obj.name, 'dept_user': users.count(), 'owner': dept_obj and dept_obj.owner, 'description': dept_obj and dept_obj.description, 'gender': { 'male': users.filter(gender=1).count(), 'female': users.filter(gender=2).count(), 'unknown': users.filter(gender=0).count(), }, 'sub_dept_map': [] } for dept in sub_dept: all_did = [dept.pk] inner(dept.pk, all_did) sub_data = { 'name': dept.name, 'count': Users.objects.filter(dept_id__in=all_did).count() } data['sub_dept_map'].append(sub_data) return SuccessResponse(data)