from django.utils.translation import ugettext as _ from common.utils import get_logger from tickets.utils import ( send_ticket_processed_mail_to_applicant, send_ticket_applied_mail_to_assignees ) from tickets.const import TicketAction logger = get_logger(__name__) class BaseHandler(object): def __init__(self, ticket): self.ticket = ticket # on action def _on_open(self): self.ticket.applicant_display = str(self.ticket.applicant) meta_display = getattr(self, '_construct_meta_display_of_open', lambda: {})() self.ticket.meta.update(meta_display) self.ticket.save() self._send_applied_mail_to_assignees() def _on_approve(self): if self.ticket.approval_step != len(self.ticket.process_map): self._send_processed_mail_to_applicant(self.ticket.processor) self.ticket.approval_step += 1 self.ticket.create_related_node() self._send_applied_mail_to_assignees() is_finished = False else: self.ticket.set_state_approve() self.ticket.set_status_closed() self._send_processed_mail_to_applicant(self.ticket.processor) is_finished = True self.ticket.save() return is_finished def _on_reject(self): self.ticket.set_state_reject() self.ticket.set_status_closed() self.__on_process(self.ticket.processor) def _on_close(self): self.ticket.set_state_closed() self.ticket.set_status_closed() self.__on_process(self.ticket.processor) def __on_process(self, processor): self._send_processed_mail_to_applicant(processor) self.ticket.save() def dispatch(self, action): processor = self.ticket.processor current_node = self.ticket.current_node.first() self.ticket.process_map[self.ticket.approval_step - 1].update({ 'approval_date': str(current_node.date_updated), 'state': current_node.state, 'processor': processor.id if processor else '', 'processor_display': str(processor) if processor else '', }) self.ticket.save() self._create_comment_on_action(action) method = getattr(self, f'_on_{action}', lambda: None) return method() # email def _send_applied_mail_to_assignees(self): assignees = self.ticket.current_node.first().ticket_assignees.all() assignees_display = ', '.join([str(i.assignee) for i in assignees]) logger.debug('Send applied email to assignees: {}'.format(assignees_display)) send_ticket_applied_mail_to_assignees(self.ticket) def _send_processed_mail_to_applicant(self, processor): logger.debug('Send processed mail to applicant: {}'.format(self.ticket.applicant_display)) send_ticket_processed_mail_to_applicant(self.ticket, processor) # comments def _create_comment_on_action(self, action): user = self.ticket.processor # 打开或关闭工单,备注显示是自己,其他是受理人 if action == TicketAction.open or action == TicketAction.close: user = self.ticket.applicant user_display = str(user) action_display = getattr(TicketAction, action).label data = { 'body': _('{} {} the ticket').format(user_display, action_display), 'user': user, 'user_display': user_display } return self.ticket.comments.create(**data) # body body_html_format = ''' {}:
{}
''' def get_body(self): old_body = self.ticket.meta.get('body') if old_body: # 之前版本的body return old_body basic_body = self._construct_basic_body() meta_body = self._construct_meta_body() return basic_body + meta_body def _construct_basic_body(self): basic_body = ''' {}: {} {}: {} {}: {} {}: {} '''.format( _('Ticket title'), self.ticket.title, _('Ticket type'), self.ticket.get_type_display(), _('Ticket status'), self.ticket.get_status_display(), _('Ticket applicant'), self.ticket.applicant_display, ).strip() body = self.body_html_format.format(_("Ticket basic info"), basic_body) return body def _construct_meta_body(self): body = '' open_body = self._base_construct_meta_body_of_open().strip() body += open_body return body def _base_construct_meta_body_of_open(self): meta_body_of_open = getattr( self, '_construct_meta_body_of_open', lambda: _('No content') )().strip() body = self.body_html_format.format(_('Ticket applied info'), meta_body_of_open) return body