|
|
from django.utils.decorators import method_decorator
|
|
|
from drf_yasg import openapi
|
|
|
from drf_yasg.utils import swagger_auto_schema
|
|
|
from rest_framework.decorators import action
|
|
|
from rest_framework.permissions import IsAuthenticated
|
|
|
from rest_framework.viewsets import ModelViewSet
|
|
|
|
|
|
from ERP_5.utils.base_views import MultipleDestroyMixin
|
|
|
from ERP_5.utils.paginations import GlobalPagination
|
|
|
from erp_system.models import DeptModel
|
|
|
from erp_system.serializers.dept_serializer import DeptSerializer
|
|
|
|
|
|
# 定义好了一个swagger的参数
|
|
|
pid_param = openapi.Parameter(name='pid', in_=openapi.IN_QUERY, description='父部门的ID', required=False,
|
|
|
type=openapi.TYPE_INTEGER)
|
|
|
|
|
|
|
|
|
@method_decorator(name='list', decorator=swagger_auto_schema(
|
|
|
manual_parameters=[pid_param],
|
|
|
operation_description="查询部门的树形结构数据, pid=0或者不传pid表示查询顶级机构的树。")
|
|
|
)
|
|
|
class DeptViewSet(ModelViewSet, MultipleDestroyMixin):
|
|
|
"""
|
|
|
create:
|
|
|
部门机构--新增
|
|
|
|
|
|
部门机构新增, status: 201(成功), return: 新增部门机构信息
|
|
|
|
|
|
destroy:
|
|
|
部门机构--删除
|
|
|
|
|
|
部门机构删除, status: 204(成功), return: None
|
|
|
|
|
|
multiple_delete:
|
|
|
部门机构--批量删除,必传参数：ids=[1,2,3,4...]
|
|
|
|
|
|
部门机构批量删除, status: 204(成功), return: None
|
|
|
|
|
|
update:
|
|
|
部门机构--修改,不能修改部门机构，只是修改部门机构名字。
|
|
|
|
|
|
部门机构修改, status: 200(成功), return: 修改后的部门机构信息
|
|
|
|
|
|
partial_update:
|
|
|
部门机构--局部修改(部门机构批量授权),只能真的某一个部门机构一次性授权，之前的的授权会被覆盖
|
|
|
|
|
|
部门机构批量授权, status: 200(成功), return: 修改后的部门机构信息
|
|
|
|
|
|
list:
|
|
|
部门机构--获取列表, 如果传参pid=0或者pid不传，则查询所有顶级部门列表， 如果传参pid=具体的值，则查询某个父部门下的所有子部门
|
|
|
|
|
|
部门机构列表信息, status: 200(成功), return: 部门机构信息列表
|
|
|
|
|
|
retrieve:
|
|
|
查询某一个部门机构
|
|
|
|
|
|
查询指定ID的部门机构, status: 200(成功), return: 用户部门机构
|
|
|
"""
|
|
|
|
|
|
def get_queryset(self):
|
|
|
if self.action == 'list':
|
|
|
q = self.request.query_params.get('pid', None)
|
|
|
if q: # 参数中有pid
|
|
|
pid = int(q)
|
|
|
if pid == 0:
|
|
|
return DeptModel.objects.filter(parent__isnull=True).all()
|
|
|
else:
|
|
|
return DeptModel.objects.filter(parent__id=pid).all()
|
|
|
else:
|
|
|
return DeptModel.objects.filter(parent__isnull=True).all()
|
|
|
else:
|
|
|
return DeptModel.objects.all()
|
|
|
|
|
|
queryset = DeptModel.objects.all()
|
|
|
serializer_class = DeptSerializer
|
|
|
pagination_class = GlobalPagination
|
|
|
|
|
|
# permission_classes = [IsAuthenticated]
|
|
|
|
|
|
# @swagger_auto_schema(method='GET', manual_parameters=[pid_param])
|
|
|
# @action(methods=['GET'], detail=False)
|
|
|
# def find(self, request, *args, **kwargs):
|
|
|
# return super(DeptViewSet, self).list(request, *args, **kwargs)
|
