100 lines
4.3 KiB
Python
100 lines
4.3 KiB
Python
from rest_framework.response import Response
|
|
from rest_framework import serializers
|
|
from rest_framework.mixins import CreateModelMixin, DestroyModelMixin, ListModelMixin, RetrieveModelMixin, UpdateModelMixin
|
|
from apps.wf.serializers import CustomFieldSerializer, StateSerializer, TicketCreateSerializer, TicketSerializer, TransitionSerializer, WorkflowSerializer
|
|
from django.shortcuts import render
|
|
from rest_framework.viewsets import GenericViewSet, ModelViewSet
|
|
from rest_framework.decorators import action
|
|
from apps.wf.models import CustomField, Ticket, Workflow, State, Transition
|
|
from apps.system.mixins import CreateUpdateCustomMixin, CreateUpdateModelAMixin, OptimizationMixin
|
|
from apps.wf.services import WfService
|
|
|
|
# Create your views here.
|
|
class WorkflowViewSet(CreateUpdateModelAMixin, ModelViewSet):
|
|
perms_map = {'get': '*', 'post': 'workflow_create',
|
|
'put': 'workflow_update', 'delete': 'workflow_delete'}
|
|
queryset = Workflow.objects.all()
|
|
serializer_class = WorkflowSerializer
|
|
search_fields = ['name', 'description']
|
|
filterset_fields = []
|
|
ordering_fields = ['create_time']
|
|
ordering = ['-create_time']
|
|
|
|
@action(methods=['get'], detail=True, perms_map={'get':'workflow_update'}, pagination_class=None, serializer_class=StateSerializer)
|
|
def states(self, request, pk=None):
|
|
"""
|
|
工作流下的状态节点
|
|
"""
|
|
wf = self.get_object()
|
|
serializer = self.serializer_class(instance=WfService.get_wf_states(wf), many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(methods=['get'], detail=True, perms_map={'get':'workflow_update'}, pagination_class=None, serializer_class=TransitionSerializer)
|
|
def transitions(self, request, pk=None):
|
|
"""
|
|
工作流下的流转规则
|
|
"""
|
|
wf = self.get_object()
|
|
serializer = self.serializer_class(instance=WfService.get_wf_transitions(wf), many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(methods=['get'], detail=True, perms_map={'get':'workflow_update'}, pagination_class=None, serializer_class=CustomFieldSerializer)
|
|
def customfields(self, request, pk=None):
|
|
"""
|
|
工作流下的自定义字段
|
|
"""
|
|
wf = self.get_object()
|
|
serializer = self.serializer_class(instance=CustomField.objects.filter(workflow=wf), many=True)
|
|
return Response(serializer.data)
|
|
|
|
@action(methods=['get'], detail=True, perms_map={'get':'workflow_init'})
|
|
def init(self, request, pk=None):
|
|
"""
|
|
新建工单初始化
|
|
"""
|
|
ret={}
|
|
wf = self.get_object()
|
|
start_state = WfService.get_wf_start_state(wf)
|
|
transitions = WfService.get_state_transitions(start_state)
|
|
ret['workflow'] = pk
|
|
return Response(ret)
|
|
|
|
class StateViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, DestroyModelMixin, GenericViewSet):
|
|
perms_map = {'*':'*'}
|
|
queryset = State.objects.all()
|
|
serializer_class = StateSerializer
|
|
search_fields = ['name']
|
|
filterset_fields = ['workflow']
|
|
ordering = ['sort']
|
|
|
|
class TransitionViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, DestroyModelMixin, GenericViewSet):
|
|
perms_map = {'*':'*'}
|
|
queryset = Transition.objects.all()
|
|
serializer_class = TransitionSerializer
|
|
search_fields = ['name']
|
|
filterset_fields = ['workflow']
|
|
ordering = ['id']
|
|
|
|
class CustomFieldViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, DestroyModelMixin, GenericViewSet):
|
|
perms_map = {'*':'*'}
|
|
queryset = CustomField.objects.all()
|
|
serializer_class = CustomFieldSerializer
|
|
search_fields = ['field_name']
|
|
filterset_fields = ['workflow', 'field_type']
|
|
ordering = ['sort']
|
|
|
|
class TicketViewSet(OptimizationMixin, CreateUpdateCustomMixin, CreateModelMixin, ListModelMixin, RetrieveModelMixin, GenericViewSet):
|
|
perms_map = {'*':'*'}
|
|
queryset = Ticket.objects.all()
|
|
serializer_class = TicketSerializer
|
|
search_fields = ['title']
|
|
filterset_fields = ['workflow', 'state']
|
|
ordering = ['-create_time']
|
|
|
|
def get_serializer_class(self):
|
|
if self.action == 'create':
|
|
return TicketCreateSerializer
|
|
return super().get_serializer_class()
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
return super().create(request, *args, **kwargs) |