from django.utils import timezone from django.db import transaction from rest_framework.views import APIView from apps.system.models import User from apps.wf.filters import TicketFilterSet from rest_framework.response import Response from rest_framework.mixins import CreateModelMixin, DestroyModelMixin, ListModelMixin, \ RetrieveModelMixin, UpdateModelMixin from apps.wf.serializers import CustomFieldCreateUpdateSerializer, CustomFieldSerializer, StateSerializer, \ TicketAddNodeEndSerializer, TicketAddNodeSerializer, TicketCloseSerializer, \ TicketCreateSerializer, TicketDestorySerializer, TicketFlowSerializer, \ TicketHandleSerializer, TicketRetreatSerializer, \ TicketSerializer, TransitionSerializer, WorkflowSerializer, \ TicketListSerializer, TicketDetailSerializer from rest_framework.viewsets import GenericViewSet, ModelViewSet from rest_framework.decorators import action from apps.wf.models import CustomField, Ticket, Workflow, State, Transition, TicketFlow from apps.utils.mixins import CreateUpdateCustomMixin, CreateUpdateModelAMixin from apps.wf.services import WfService from rest_framework.exceptions import APIException from rest_framework import status from django.db.models import Count from .scripts import GetParticipants # Create your views here. class FromCodeListView(APIView): def get(self, request, format=None): """ 获取处理人代码列表 """ return Response(GetParticipants.all_funcs) 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_worlflow_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_workflow_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, is_deleted=False).order_by('sort'), 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_workflow_start_state(wf) transitions = WfService.get_state_transitions(start_state) ret['workflow'] = pk ret['transitions'] = TransitionSerializer(instance=transitions, many=True).data field_list = CustomFieldSerializer(instance=WfService.get_workflow_custom_fields(wf), many=True).data for i in field_list: if i['field_key'] in start_state.state_fields: i['field_attribute'] = start_state.state_fields[i['field_key']] else: i['field_attribute'] = State.STATE_FIELD_READONLY ret['field_list'] = field_list return Response(ret) class StateViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, DestroyModelMixin, GenericViewSet): perms_map = {'get': '*', 'post': 'workflow:update', 'put': 'workflow:update', 'delete': 'workflow:update'} queryset = State.objects.all() serializer_class = StateSerializer search_fields = ['name'] filterset_fields = ['workflow'] ordering = ['sort'] class TransitionViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, DestroyModelMixin, GenericViewSet): perms_map = {'get': '*', 'post': 'workflow:update', 'put': 'workflow:update', 'delete': 'workflow:update'} queryset = Transition.objects.all() serializer_class = TransitionSerializer search_fields = ['name'] filterset_fields = ['workflow'] ordering = ['id'] class CustomFieldViewSet(CreateModelMixin, UpdateModelMixin, RetrieveModelMixin, DestroyModelMixin, GenericViewSet): perms_map = {'get': '*', 'post': 'workflow:update', 'put': 'workflow:update', 'delete': 'workflow:update'} queryset = CustomField.objects.all() serializer_class = CustomFieldSerializer search_fields = ['field_name'] filterset_fields = ['workflow', 'field_type'] ordering = ['sort'] def get_serializer_class(self): if self.action in ['create', 'update']: return CustomFieldCreateUpdateSerializer return super().get_serializer_class() class TicketViewSet(CreateUpdateCustomMixin, CreateModelMixin, ListModelMixin, RetrieveModelMixin, GenericViewSet): perms_map = {'get': '*', 'post': 'ticket:create'} queryset = Ticket.objects.all() serializer_class = TicketSerializer search_fields = ['title'] filterset_class = TicketFilterSet ordering = ['-create_time'] def get_serializer_class(self): if self.action == 'create': return TicketCreateSerializer elif self.action == 'handle': return TicketHandleSerializer elif self.action == 'retreat': return TicketRetreatSerializer elif self.action == 'list': return TicketListSerializer elif self.action == 'retrieve': return TicketDetailSerializer return super().get_serializer_class() def filter_queryset(self, queryset): if not self.detail and not self.request.query_params.get('category', None): raise APIException('请指定查询分类') return super().filter_queryset(queryset) @transaction.atomic def create(self, request, *args, **kwargs): """ 新建工单 """ rdata = request.data serializer = self.get_serializer(data=rdata) serializer.is_valid(raise_exception=True) vdata = serializer.validated_data # 校验之后的数据 start_state = WfService.get_workflow_start_state(vdata['workflow']) transition = vdata.pop('transition') ticket_data = vdata['ticket_data'] save_ticket_data = {} # 校验必填项 if transition.field_require_check: for key, value in start_state.state_fields.items(): if int(value) == State.STATE_FIELD_REQUIRED: if key not in ticket_data and not ticket_data[key]: raise APIException('字段{}必填'.format(key)) save_ticket_data[key] = ticket_data[key] elif int(value) == State.STATE_FIELD_OPTIONAL: save_ticket_data[key] = ticket_data[key] ticket = serializer.save(state=start_state, create_by=request.user, create_time=timezone.now(), act_state=Ticket.TICKET_ACT_STATE_DRAFT, belong_dept=request.user.dept, ticket_data=save_ticket_data) # 先创建出来 # 更新title和sn title = vdata.get('title', '') title_template = ticket.workflow.title_template if title_template: all_ticket_data = {**rdata, **ticket_data} title = title_template.format(**all_ticket_data) sn = WfService.get_ticket_sn(ticket.workflow) # 流水号 ticket.sn = sn ticket.title = title ticket.save() ticket = WfService.handle_ticket(ticket=ticket, transition=transition, new_ticket_data=ticket_data, handler=request.user, created=True) return Response(TicketSerializer(instance=ticket).data) @action(methods=['get'], detail=False, perms_map={'get': '*'}) def duty_agg(self, request, pk=None): """ 工单待办聚合 """ ret = {} queryset = Ticket.objects.filter(participant__contains=request.user.id, is_deleted=False)\ .exclude(act_state__in=[Ticket.TICKET_ACT_STATE_FINISH, Ticket.TICKET_ACT_STATE_CLOSED]) ret['total_count'] = queryset.count() ret['details'] = list(queryset.values('workflow', 'workflow__name').annotate(count=Count('workflow'))) return Response(ret) @action(methods=['post'], detail=True, perms_map={'post': '*'}) @transaction.atomic def handle(self, request, pk=None): """ 处理工单 """ ticket = self.get_object() serializer = TicketHandleSerializer(data=request.data) serializer.is_valid(raise_exception=True) vdata = serializer.validated_data new_ticket_data = ticket.ticket_data new_ticket_data.update(**vdata['ticket_data']) ticket = WfService.handle_ticket(ticket=ticket, transition=vdata['transition'], new_ticket_data=new_ticket_data, handler=request.user, suggestion=vdata['suggestion']) return Response(TicketSerializer(instance=ticket).data) @action(methods=['get'], detail=True, perms_map={'get': '*'}) def flowsteps(self, request, pk=None): """ 工单流转step, 用于显示当前状态的step图(线性结构) """ ticket = self.get_object() steps = WfService.get_ticket_steps(ticket) return Response(StateSerializer(instance=steps, many=True).data) @action(methods=['get'], detail=True, perms_map={'get': '*'}) def flowlogs(self, request, pk=None): """ 工单流转记录 """ ticket = self.get_object() flowlogs = TicketFlow.objects.filter(ticket=ticket).order_by('-create_time') serializer = TicketFlowSerializer(instance=flowlogs, many=True) return Response(serializer.data) @action(methods=['get'], detail=True, perms_map={'get': '*'}) def transitions(self, request, pk=None): """ 获取工单可执行的操作 """ ticket = self.get_object() transitions = WfService.get_ticket_transitions(ticket) return Response(TransitionSerializer(instance=transitions, many=True).data) @action(methods=['post'], detail=True, perms_map={'post': '*'}) def accpet(self, request, pk=None): """ 接单,当工单当前处理人实际为多个人时(角色、部门、多人都有可能， 注意角色和部门有可能实际只有一人) """ ticket = self.get_object() result = WfService.ticket_handle_permission_check(ticket, request.user) if result.get('need_accept', False): ticket.participant_type = State.PARTICIPANT_TYPE_PERSONAL ticket.participant = request.user.id ticket.save() # 接单日志 # 更新工单流转记录 TicketFlow.objects.create(ticket=ticket, state=ticket.state, ticket_data=WfService.get_ticket_all_field_value(ticket), suggestion='', participant_type=State.PARTICIPANT_TYPE_PERSONAL, intervene_type=Transition.TRANSITION_ATTRIBUTE_TYPE_ACCEPT, participant=request.user, transition=None) return Response() else: raise APIException('无需接单') @action(methods=['post'], detail=True, perms_map={'post': '*'}) def retreat(self, request, pk=None): """ 撤回工单，允许创建人在指定状态撤回工单至初始状态，状态设置中开启允许撤回 """ ticket = self.get_object() if ticket.create_by != request.user: raise APIException('非创建人不可撤回') if not ticket.state.enable_retreat: raise APIException('该状态不可撤回') start_state = WfService.get_workflow_start_state(ticket.workflow) ticket.state = start_state ticket.participant_type = State.PARTICIPANT_TYPE_PERSONAL ticket.participant = request.user.id ticket.act_state = Ticket.TICKET_ACT_STATE_RETREAT ticket.save() # 更新流转记录 suggestion = request.data.get('suggestion', '') # 撤回原因 TicketFlow.objects.create(ticket=ticket, state=ticket.state, ticket_data=WfService.get_ticket_all_field_value(ticket), suggestion=suggestion, participant_type=State.PARTICIPANT_TYPE_PERSONAL, intervene_type=Transition.TRANSITION_INTERVENE_TYPE_RETREAT, participant=request.user, transition=None) return Response() @action(methods=['post'], detail=True, perms_map={'post': '*'}, serializer_class=TicketAddNodeSerializer) def add_node(self, request, pk=None): """ 加签 """ ticket = self.get_object() data = request.data add_user = User.objects.get(pk=data['toadd_user']) ticket.participant_type = State.PARTICIPANT_TYPE_PERSONAL ticket.participant = add_user.id ticket.in_add_node = True ticket.add_node_man = request.user ticket.save() # 更新流转记录 suggestion = request.data.get('suggestion', '') # 加签说明 TicketFlow.objects.create(ticket=ticket, state=ticket.state, ticket_data=WfService.get_ticket_all_field_value(ticket), suggestion=suggestion, participant_type=State.PARTICIPANT_TYPE_PERSONAL, intervene_type=Transition.TRANSITION_INTERVENE_TYPE_ADD_NODE, participant=request.user, transition=None) return Response() @action(methods=['post'], detail=True, perms_map={'post': '*'}, serializer_class=TicketAddNodeEndSerializer) def add_node_end(self, request, pk=None): """ 加签完成 """ ticket = self.get_object() ticket.participant_type = State.PARTICIPANT_TYPE_PERSONAL ticket.in_add_node = False ticket.participant = ticket.add_node_man.id ticket.add_node_man = None ticket.save() # 更新流转记录 suggestion = request.data.get('suggestion', '') # 加签意见 TicketFlow.objects.create(ticket=ticket, state=ticket.state, ticket_data=WfService.get_ticket_all_field_value(ticket), suggestion=suggestion, participant_type=State.PARTICIPANT_TYPE_PERSONAL, intervene_type=Transition.TRANSITION_INTERVENE_TYPE_ADD_NODE_END, participant=request.user, transition=None) return Response() @action(methods=['post'], detail=True, perms_map={'post': '*'}, serializer_class=TicketCloseSerializer) def close(self, request, pk=None): """ 关闭工单(创建人在初始状态) """ ticket = self.get_object() if ticket.state.type == State.STATE_TYPE_START and ticket.create_by == request.user: end_state = WfService.get_workflow_end_state(ticket.workflow) ticket.state = end_state ticket.participant_type = 0 ticket.participant = 0 ticket.act_state = Ticket.TICKET_ACT_STATE_CLOSED ticket.save() # 更新流转记录 suggestion = request.data.get('suggestion', '') # 关闭原因 TicketFlow.objects.create(ticket=ticket, state=ticket.state, ticket_data=WfService.get_ticket_all_field_value(ticket), suggestion=suggestion, participant_type=State.PARTICIPANT_TYPE_PERSONAL, intervene_type=Transition.TRANSITION_INTERVENE_TYPE_CLOSE, participant=request.user, transition=None) return Response() else: return Response('工单不可关闭', status=status.HTTP_400_BAD_REQUEST) @action(methods=['post'], detail=False, perms_map={'post': 'ticket:destorys'}, serializer_class=TicketDestorySerializer) def destorys(self, request, pk=None): """ 批量物理删除 """ Ticket.objects.filter(id__in=request.data.get('ids', [])).delete(soft=False) return Response() class TicketFlowViewSet(ListModelMixin, RetrieveModelMixin, GenericViewSet): """ 工单日志 """ perms_map = {'get': '*'} queryset = TicketFlow.objects.all() serializer_class = TicketFlowSerializer search_fields = ['suggestion'] filterset_fields = ['ticket'] ordering = ['-create_time']