mirror of https://github.com/jumpserver/jumpserver
241 lines
8.3 KiB
Python
241 lines
8.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
from django.db import models
|
|
from django.db.models import Q
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from common.mixins.models import CommonModelMixin
|
|
from common.db.encoder import ModelJSONFieldEncoder
|
|
from orgs.mixins.models import OrgModelMixin
|
|
from orgs.utils import tmp_to_root_org, tmp_to_org
|
|
from tickets.const import TicketType, TicketStatus, TicketState, TicketApprovalLevel, ProcessStatus, TicketAction
|
|
from tickets.signals import post_change_ticket_action
|
|
from tickets.handler import get_ticket_handler
|
|
from tickets.errors import AlreadyClosed
|
|
|
|
__all__ = ['Ticket']
|
|
|
|
|
|
class TicketStep(CommonModelMixin):
|
|
ticket = models.ForeignKey(
|
|
'Ticket', related_name='ticket_steps', on_delete=models.CASCADE, verbose_name='Ticket'
|
|
)
|
|
level = models.SmallIntegerField(
|
|
default=TicketApprovalLevel.one, choices=TicketApprovalLevel.choices,
|
|
verbose_name=_('Approve level')
|
|
)
|
|
state = models.CharField(choices=ProcessStatus.choices, max_length=64, default=ProcessStatus.notified)
|
|
|
|
|
|
class TicketAssignee(CommonModelMixin):
|
|
assignee = models.ForeignKey(
|
|
'users.User', related_name='ticket_assignees', on_delete=models.CASCADE, verbose_name='Assignee'
|
|
)
|
|
state = models.CharField(choices=ProcessStatus.choices, max_length=64, default=ProcessStatus.notified)
|
|
step = models.ForeignKey('tickets.TicketStep', related_name='ticket_assignees', on_delete=models.CASCADE)
|
|
|
|
class Meta:
|
|
verbose_name = _('Ticket assignee')
|
|
|
|
def __str__(self):
|
|
return '{0.assignee.name}({0.assignee.username})_{0.step}'.format(self)
|
|
|
|
|
|
class Ticket(CommonModelMixin, OrgModelMixin):
|
|
title = models.CharField(max_length=256, verbose_name=_("Title"))
|
|
type = models.CharField(
|
|
max_length=64, choices=TicketType.choices,
|
|
default=TicketType.general, verbose_name=_("Type")
|
|
)
|
|
meta = models.JSONField(encoder=ModelJSONFieldEncoder, default=dict, verbose_name=_("Meta"))
|
|
state = models.CharField(
|
|
max_length=16, choices=TicketState.choices,
|
|
default=TicketState.open, verbose_name=_("State")
|
|
)
|
|
status = models.CharField(
|
|
max_length=16, choices=TicketStatus.choices,
|
|
default=TicketStatus.open, verbose_name=_("Status")
|
|
)
|
|
approval_step = models.SmallIntegerField(
|
|
default=TicketApprovalLevel.one, choices=TicketApprovalLevel.choices,
|
|
verbose_name=_('Approval step')
|
|
)
|
|
# 申请人
|
|
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"))
|
|
process_map = models.JSONField(encoder=ModelJSONFieldEncoder, default=list, verbose_name=_("Process"))
|
|
# 评论
|
|
comment = models.TextField(default='', blank=True, verbose_name=_('Comment'))
|
|
flow = models.ForeignKey(
|
|
'TicketFlow', related_name='tickets', on_delete=models.SET_NULL, null=True,
|
|
verbose_name=_("TicketFlow")
|
|
)
|
|
|
|
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 == TicketType.apply_asset.value
|
|
|
|
@property
|
|
def type_apply_application(self):
|
|
return self.type == TicketType.apply_application.value
|
|
|
|
@property
|
|
def type_login_confirm(self):
|
|
return self.type == TicketType.login_confirm.value
|
|
|
|
# status
|
|
@property
|
|
def status_open(self):
|
|
return self.status == TicketStatus.open.value
|
|
|
|
@property
|
|
def status_closed(self):
|
|
return self.status == TicketStatus.closed.value
|
|
|
|
@property
|
|
def state_open(self):
|
|
return self.state == TicketState.open.value
|
|
|
|
@property
|
|
def state_approve(self):
|
|
return self.state == TicketState.approved.value
|
|
|
|
@property
|
|
def state_reject(self):
|
|
return self.state == TicketState.rejected.value
|
|
|
|
@property
|
|
def state_close(self):
|
|
return self.state == TicketState.closed.value
|
|
|
|
@property
|
|
def current_node(self):
|
|
return self.ticket_steps.filter(level=self.approval_step)
|
|
|
|
@property
|
|
def processor(self):
|
|
processor = self.current_node.first().ticket_assignees.exclude(state=ProcessStatus.notified).first()
|
|
return processor.assignee if processor else None
|
|
|
|
def set_state_approve(self):
|
|
self.state = TicketState.approved
|
|
|
|
def set_state_reject(self):
|
|
self.state = TicketState.rejected
|
|
|
|
def set_state_closed(self):
|
|
self.state = TicketState.closed
|
|
|
|
def set_status_closed(self):
|
|
self.status = TicketStatus.closed
|
|
|
|
def create_related_node(self):
|
|
approval_rule = self.get_current_ticket_flow_approve()
|
|
ticket_step = TicketStep.objects.create(ticket=self, level=self.approval_step)
|
|
ticket_assignees = []
|
|
assignees = approval_rule.assignees.all()
|
|
for assignee in assignees:
|
|
ticket_assignees.append(TicketAssignee(step=ticket_step, assignee=assignee))
|
|
TicketAssignee.objects.bulk_create(ticket_assignees)
|
|
|
|
def create_process_map(self):
|
|
approval_rules = self.flow.rules.order_by('level')
|
|
nodes = list()
|
|
for node in approval_rules:
|
|
nodes.append(
|
|
{
|
|
'approval_level': node.level,
|
|
'state': ProcessStatus.notified,
|
|
'assignees': [i for i in node.assignees.values_list('id', flat=True)],
|
|
'assignees_display': node.assignees_display
|
|
}
|
|
)
|
|
return nodes
|
|
|
|
# TODO 兼容不存在流的工单
|
|
def create_process_map_and_node(self, assignees):
|
|
self.process_map = [{
|
|
'approval_level': 1,
|
|
'state': 'notified',
|
|
'assignees': [assignee.id for assignee in assignees],
|
|
'assignees_display': [str(assignee) for assignee in assignees]
|
|
}, ]
|
|
self.save()
|
|
ticket_step = TicketStep.objects.create(ticket=self, level=1)
|
|
ticket_assignees = []
|
|
for assignee in assignees:
|
|
ticket_assignees.append(TicketAssignee(step=ticket_step, assignee=assignee))
|
|
TicketAssignee.objects.bulk_create(ticket_assignees)
|
|
|
|
# action changed
|
|
def open(self, applicant):
|
|
self.applicant = applicant
|
|
self._change_action(TicketAction.open)
|
|
|
|
def update_current_step_state_and_assignee(self, processor, state):
|
|
if self.status_closed:
|
|
raise AlreadyClosed
|
|
self.state = state
|
|
current_node = self.current_node
|
|
current_node.update(state=state)
|
|
current_node.first().ticket_assignees.filter(assignee=processor).update(state=state)
|
|
|
|
def approve(self, processor):
|
|
self.update_current_step_state_and_assignee(processor, TicketState.approved)
|
|
self._change_action(TicketAction.approve)
|
|
|
|
def reject(self, processor):
|
|
self.update_current_step_state_and_assignee(processor, TicketState.rejected)
|
|
self._change_action(TicketAction.reject)
|
|
|
|
def close(self, processor):
|
|
self.update_current_step_state_and_assignee(processor, TicketState.closed)
|
|
self._change_action(TicketAction.close)
|
|
|
|
def _change_action(self, action):
|
|
self.save()
|
|
post_change_ticket_action.send(sender=self.__class__, ticket=self, action=action)
|
|
|
|
# ticket
|
|
def has_assignee(self, assignee):
|
|
return self.ticket_steps.filter(ticket_assignees__assignee=assignee, level=self.approval_step).exists()
|
|
|
|
@classmethod
|
|
def get_user_related_tickets(cls, user):
|
|
queries = Q(applicant=user) | Q(ticket_steps__ticket_assignees__assignee=user)
|
|
tickets = cls.all().filter(queries).distinct()
|
|
return tickets
|
|
|
|
def get_current_ticket_flow_approve(self):
|
|
return self.flow.rules.filter(level=self.approval_step).first()
|
|
|
|
@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
|