feat: base 添加ticketmixin可集成到viewset下以支持工作流

apps/wf/mixins.py

@@ -1,11 +1,18 @@
 from apps.wf.models import Workflow, Ticket, State
 from rest_framework.exceptions import ParseError
 from apps.wf.services import WfService
+from apps.system.models import User
 
 class TicketMixin:
+    """
+    可挂载到正常model,使其支持工作流
+    model添加ticket字段
+    serializer添加ticket_
+    该处会修改perform_create和perform_update方法,注意!
+    """
     workflow_key = None
     ticket_auto_submit_on_update = True
-    ticket_data_save_keys = []
+    ticket_data_save_fields = []
 
     def get_workflow_key(self, instance):
         return self.workflow_key
@@ -13,19 +20,58 @@ class TicketMixin:
     def should_create_ticket(self, instance):
         return True
     
+    def gen_ticket_data(self, instance):
+        ticket_data = {"t_model": instance.__class__.__name__, "t_id": instance.id}
+        if self.ticket_data_save_fields:
+            for field in self.ticket_data_save_fields:
+                if '.' in field:
+                    attr_list  = field.split('.')
+                    expr = instance
+                    for a in attr_list:
+                        expr = getattr(expr, a)
+                    ticket_data[field] = expr
+                else:
+                    ticket_data[field] = getattr(instance, field)
+        return ticket_data
+
+    def perform_update(self, serializer):
+        ins = serializer.save()
+        if ins.ticket and self.ticket_auto_submit_on_update:
+            source_state:State = ins.ticket.state
+            if source_state.type != State.STATE_TYPE_START:
+                raise ParseError('该工单已开始流转,不可修改')
+            transition = WfService.get_state_transitions(source_state).first()
+            ticket_data = self.gen_ticket_data(ins)
+            WfService.handle_ticket(ticket=ins.ticket, transition=transition, new_ticket_data=ticket_data, 
+                                    handler=self.request.user, oinfo=self.request.data)
+    
     def perform_create(self, serializer):
         ins = serializer.save()
-        if self.workflow_key:
+        handler:User = self.request.user
+        if self.should_create_ticket(ins):
+            workflow_key = self.get_workflow_key(ins)
+            if not workflow_key:
+                raise ParseError('工作流异常:必须赋值workflow_key')
             try:
-                wf = Workflow.objects.get(key=self.workflow_key)
+                wf = Workflow.objects.get(key=workflow_key)
             except Exception as e:
-                raise ParseError(f'工作流{self.workflow_key}异常:{e}')
+                raise ParseError(f'工作流{workflow_key}异常:{e}')
+            
+            # 开始创建工单
             source_state: State = WfService.get_workflow_start_state(wf)
             transitions = WfService.get_state_transitions(source_state)
             if transitions.count() == 1:
                 transition = transitions.first()
-                
+                ticket_data = self.gen_ticket_data(ins)
+                WfService.handle_ticket(ticket=None, transition=transition, new_ticket_data=ticket_data, 
+                                        handler=handler, oinfo=self.request.data)
             else:
-                raise ParseError(f'工作流{self.workflow_key}异常:起始状态{source_state}有多个后续状态;不可直接创建')
+                raise ParseError(f'工作流{workflow_key}异常:有多个后续状态;不可处理')
 
-            
+    def perform_destroy(self, instance):
+        ticket = instance.ticket
+        if ticket and ticket.state.type != State.STATE_TYPE_START:
+            raise ParseError('该工单已开始流转,不可删除')
+        instance.delete()
+        ticket.delete()
+        

apps/wf/services.py

@@ -332,7 +332,10 @@ class WfService(object):
             title_template = ticket.workflow.title_template
             if title_template:
                 all_ticket_data = {**oinfo, **new_ticket_data}
-                ticket_title = title_template.format(**all_ticket_data)
+                try:
+                    ticket_title = title_template.format(**all_ticket_data)
+                except KeyError as e:
+                    raise ParseError(f"工单标题模板中存在未定义的变量:{e}")
             sn = WfService.get_ticket_sn(ticket.workflow)  # 流水号
             ticket.sn = sn
             ticket.title = ticket_title