import json from urllib.parse import urljoin from django.conf import settings from django.core.cache import cache from django.shortcuts import reverse from django.forms import model_to_dict from django.template.loader import render_to_string from django.utils.translation import ugettext_lazy as _ from notifications.notifications import UserMessage from common.utils import get_logger, random_string from common.db.encoder import ModelJSONFieldEncoder from .models import Ticket from . import const 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) 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 = None self.ticket = ticket super().__init__(user) @property def token(self): if self._token is None: self._token = random_string(32) return self._token @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, '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)