perf: 重构工单处理流程 (#5408)

* perf: 重构工单处理流程

* perf: 重构工单处理流程 (1)

* perf: 重构工单处理流程 (2)

* perf: 重构工单处理流程 (3)

* perf: 重构工单处理流程 (4)

* perf: 重构工单处理流程 (5)

* perf: 重构工单处理流程 (6)

Co-authored-by: Bai <bugatti_it@163.com>

apps/tickets/models/ticket.py

@@ -0,0 +1,199 @@
+# -*- coding: utf-8 -*-
+#
+import json
+import uuid
+from datetime import datetime
+from django.db import models
+from django.db.models import Q
+from django.utils.translation import ugettext_lazy as _
+from django.conf import settings
+
+from common.mixins.models import CommonModelMixin
+from orgs.mixins.models import OrgModelMixin
+from orgs.utils import tmp_to_root_org, tmp_to_org
+from tickets.const import TicketTypeChoices, TicketActionChoices, TicketStatusChoices
+from tickets.signals import post_change_ticket_action
+from tickets.handler import get_ticket_handler
+
+__all__ = ['Ticket', 'ModelJSONFieldEncoder']
+
+
+class ModelJSONFieldEncoder(json.JSONEncoder):
+    """ 解决一些类型的字段不能序列化的问题 """
+    def default(self, obj):
+        if isinstance(obj, datetime):
+            return obj.strftime(settings.DATETIME_DISPLAY_FORMAT)
+        if isinstance(obj, uuid.UUID):
+            return str(obj)
+        if isinstance(obj, type(_("ugettext_lazy"))):
+            return str(obj)
+        else:
+            return super().default(obj)
+
+
+class Ticket(CommonModelMixin, OrgModelMixin):
+    title = models.CharField(max_length=256, verbose_name=_("Title"))
+    type = models.CharField(
+        max_length=64, choices=TicketTypeChoices.choices,
+        default=TicketTypeChoices.general.value, verbose_name=_("Type")
+    )
+    meta = models.JSONField(encoder=ModelJSONFieldEncoder, default=dict, verbose_name=_("Meta"))
+    action = models.CharField(
+        choices=TicketActionChoices.choices, max_length=16,
+        default=TicketActionChoices.open.value, verbose_name=_("Action")
+    )
+    status = models.CharField(
+        max_length=16, choices=TicketStatusChoices.choices,
+        default=TicketStatusChoices.open.value, verbose_name=_("Status")
+    )
+    # 申请人
+    applicant = models.ForeignKey(
+        'users.User', related_name='applied_tickets', on_delete=models.SET_NULL, null=True,
+        verbose_name=_("Applicant")
+    )
+    applicant_display = models.CharField(
+        max_length=256, default='', verbose_name=_("Applicant display")
+    )
+    # 处理人
+    processor = models.ForeignKey(
+        'users.User', related_name='processed_tickets', on_delete=models.SET_NULL, null=True,
+        verbose_name=_("Processor")
+    )
+    processor_display = models.CharField(
+        max_length=256, blank=True, null=True, default='', verbose_name=_("Processor display")
+    )
+    # 受理人列表
+    assignees = models.ManyToManyField(
+        'users.User', related_name='assigned_tickets', verbose_name=_("Assignees")
+    )
+    assignees_display = models.JSONField(
+        encoder=ModelJSONFieldEncoder, default=list, verbose_name=_('Assignees display')
+    )
+    # 评论
+    comment = models.TextField(default='', blank=True, verbose_name=_('Comment'))
+
+    class Meta:
+        ordering = ('-date_created',)
+
+    def __str__(self):
+        return '{}({})'.format(self.title, self.applicant_display)
+
+    # type
+    @property
+    def type_apply_asset(self):
+        return self.type == TicketTypeChoices.apply_asset.value
+
+    @property
+    def type_apply_application(self):
+        return self.type == TicketTypeChoices.apply_application.value
+
+    @property
+    def type_login_confirm(self):
+        return self.type == TicketTypeChoices.login_confirm.value
+
+    # status
+    @property
+    def status_closed(self):
+        return self.status == TicketStatusChoices.closed.value
+    
+    @property
+    def status_open(self):
+        return self.status == TicketStatusChoices.open.value
+
+    def set_status_closed(self):
+        self.status = TicketStatusChoices.closed.value
+
+    # action
+    @property
+    def action_open(self):
+        return self.action == TicketActionChoices.open.value
+
+    @property
+    def action_approve(self):
+        return self.action == TicketActionChoices.approve.value
+
+    @property
+    def action_reject(self):
+        return self.action == TicketActionChoices.reject.value
+
+    @property
+    def action_close(self):
+        return self.action == TicketActionChoices.close.value
+
+    # action changed
+    def open(self, applicant):
+        self.applicant = applicant
+        self._change_action(action=TicketActionChoices.open.value)
+
+    def approve(self, processor):
+        self.processor = processor
+        self._change_action(action=TicketActionChoices.approve.value)
+
+    def reject(self, processor):
+        self.processor = processor
+        self._change_action(action=TicketActionChoices.reject.value)
+
+    def close(self, processor):
+        self.processor = processor
+        self._change_action(action=TicketActionChoices.close.value)
+
+    def _change_action(self, action):
+        self.action = action
+        self.save()
+        post_change_ticket_action.send(sender=self.__class__, ticket=self, action=action)
+
+    # ticket
+    def has_assignee(self, assignee):
+        return self.assignees.filter(id=assignee.id).exists()
+
+    @classmethod
+    def get_user_related_tickets(cls, user):
+        queries = None
+        tickets = cls.all()
+        if user.is_superuser:
+            pass
+        elif user.is_super_auditor:
+            pass
+        elif user.is_org_admin:
+            admin_orgs_id = [
+                str(org_id) for org_id in user.admin_orgs.values_list('id', flat=True)
+            ]
+            assigned_tickets_id = [
+                str(ticket_id) for ticket_id in user.assigned_tickets.values_list('id', flat=True)
+            ]
+            queries = Q(applicant=user)
+            queries |= Q(processor=user)
+            queries |= Q(org_id__in=admin_orgs_id)
+            queries |= Q(id__in=assigned_tickets_id)
+        elif user.is_org_auditor:
+            audit_orgs_id = [
+                str(org_id) for org_id in user.audit_orgs.values_list('id', flat=True)
+            ]
+            queries = Q(org_id__in=audit_orgs_id)
+        elif user.is_common_user:
+            queries = Q(applicant=user)
+        else:
+            tickets = cls.objects.none()
+        if queries:
+            tickets = tickets.filter(queries)
+        return tickets.distinct()
+
+    @classmethod
+    def all(cls):
+        with tmp_to_root_org():
+            return Ticket.objects.all()
+
+    def save(self, *args, **kwargs):
+        """ 确保保存的org_id的是自身的值 """
+        with tmp_to_org(self.org_id):
+            return super().save(*args, **kwargs)
+
+    @property
+    def handler(self):
+        return get_ticket_handler(ticket=self)
+
+    # body
+    @property
+    def body(self):
+        _body = self.handler.get_body()
+        return _body