158 lines
5.2 KiB
Python
158 lines
5.2 KiB
Python
# -*- 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="获取成功")
|