mirror of https://github.com/jumpserver/jumpserver
172 lines
5.1 KiB
Python
172 lines
5.1 KiB
Python
import json
|
|
from urllib.parse import urljoin
|
|
|
|
from django.conf import settings
|
|
from django.core.cache import cache
|
|
from django.forms import model_to_dict
|
|
from django.shortcuts import reverse
|
|
from django.template.loader import render_to_string
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from common.db.encoder import ModelJSONFieldEncoder
|
|
from common.utils import get_logger, random_string
|
|
from notifications.notifications import UserMessage
|
|
from . import const
|
|
from .models import Ticket
|
|
|
|
logger = get_logger(__file__)
|
|
|
|
|
|
class BaseTicketMessage(UserMessage):
|
|
title: ''
|
|
ticket: Ticket
|
|
content_title: str
|
|
|
|
@property
|
|
def ticket_detail_url(self):
|
|
tp = self.ticket.type
|
|
return urljoin(
|
|
settings.SITE_URL,
|
|
const.TICKET_DETAIL_URL.format(
|
|
id=str(self.ticket.id),
|
|
type=tp
|
|
)
|
|
)
|
|
|
|
@property
|
|
def content_title(self):
|
|
raise NotImplementedError
|
|
|
|
@property
|
|
def subject(self):
|
|
raise NotImplementedError
|
|
|
|
def get_html_msg(self) -> dict:
|
|
context = dict(
|
|
title=self.content_title,
|
|
content=self.content,
|
|
ticket_detail_url=self.ticket_detail_url
|
|
)
|
|
message = render_to_string('tickets/_msg_ticket.html', context)
|
|
return {
|
|
'subject': self.subject,
|
|
'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
return None
|
|
|
|
@property
|
|
def content(self):
|
|
content = [
|
|
{'title': _('Ticket basic info'), 'content': self.basic_items},
|
|
{'title': _('Ticket applied info'), 'content': self.spec_items},
|
|
]
|
|
return content
|
|
|
|
def _get_fields_items(self, item_names):
|
|
fields = self.ticket._meta._forward_fields_map
|
|
json_data = json.dumps(model_to_dict(self.ticket), cls=ModelJSONFieldEncoder)
|
|
data = json.loads(json_data)
|
|
items = []
|
|
|
|
for name in item_names:
|
|
field = fields[name]
|
|
item = {'name': name, 'title': field.verbose_name}
|
|
value = self.ticket.get_field_display(name, field, data)
|
|
if not value:
|
|
continue
|
|
item['value'] = value
|
|
items.append(item)
|
|
return items
|
|
|
|
@property
|
|
def basic_items(self):
|
|
item_names = ['serial_num', 'title', 'type', 'state', 'applicant', 'comment']
|
|
return self._get_fields_items(item_names)
|
|
|
|
@property
|
|
def spec_items(self):
|
|
fields = self.ticket._meta.local_fields + self.ticket._meta.local_many_to_many
|
|
excludes = ['ticket_ptr', 'flow']
|
|
item_names = [field.name for field in fields if field.name not in excludes]
|
|
return self._get_fields_items(item_names)
|
|
|
|
|
|
class TicketAppliedToAssigneeMessage(BaseTicketMessage):
|
|
def __init__(self, user, ticket):
|
|
self.token = random_string(32)
|
|
self.ticket = ticket
|
|
super().__init__(user)
|
|
|
|
@property
|
|
def content_title(self):
|
|
return _('Your has a new ticket, applicant - {}').format(self.ticket.applicant)
|
|
|
|
@property
|
|
def subject(self):
|
|
title = _('{}: New Ticket - {} ({})').format(
|
|
self.ticket.applicant,
|
|
self.ticket.title,
|
|
self.ticket.get_type_display()
|
|
)
|
|
return title
|
|
|
|
def get_ticket_approval_url(self):
|
|
url = reverse('tickets:direct-approve', kwargs={'token': self.token})
|
|
return urljoin(settings.SITE_URL, url)
|
|
|
|
def get_html_msg(self) -> dict:
|
|
context = dict(
|
|
title=self.content_title,
|
|
content=self.content,
|
|
ticket_detail_url=self.ticket_detail_url
|
|
)
|
|
|
|
ticket_approval_url = self.get_ticket_approval_url()
|
|
context.update({'ticket_approval_url': ticket_approval_url})
|
|
message = render_to_string('tickets/_msg_ticket.html', context)
|
|
cache.set(self.token, {
|
|
'ticket_id': self.ticket.id, 'approver_id': self.user.id,
|
|
'content': self.content,
|
|
}, 3600)
|
|
return {
|
|
'subject': self.subject, 'message': message
|
|
}
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
from .models import Ticket
|
|
from users.models import User
|
|
ticket = Ticket.objects.first()
|
|
user = User.objects.first()
|
|
return cls(user, ticket)
|
|
|
|
|
|
class TicketProcessedToApplicantMessage(BaseTicketMessage):
|
|
def __init__(self, user, ticket, processor):
|
|
self.ticket = ticket
|
|
self.processor = processor
|
|
super().__init__(user)
|
|
|
|
@property
|
|
def content_title(self):
|
|
return _('Your ticket has been processed, processor - {}').format(str(self.processor))
|
|
|
|
@property
|
|
def subject(self):
|
|
title = _('Ticket has processed - {} ({})').format(
|
|
self.ticket.title, self.ticket.get_type_display()
|
|
)
|
|
return title
|
|
|
|
@classmethod
|
|
def gen_test_msg(cls):
|
|
from .models import Ticket
|
|
from users.models import User
|
|
ticket = Ticket.objects.first()
|
|
user = User.objects.first()
|
|
processor = User.objects.last()
|
|
return cls(user, ticket, processor)
|