136 lines
5.0 KiB
Python
136 lines
5.0 KiB
Python
import sys
|
|
|
|
from django.contrib import messages
|
|
from django.db import transaction
|
|
from django.utils.translation import ungettext, ugettext_lazy as _
|
|
|
|
from orchestra.admin.decorators import action_with_confirmation
|
|
|
|
from .forms import ChangeReasonForm
|
|
from .helpers import markdown_formated_changes
|
|
from .models import Queue, Ticket
|
|
|
|
|
|
def change_ticket_state_factory(action, final_state):
|
|
context = {
|
|
'action': action,
|
|
'form': ChangeReasonForm()
|
|
}
|
|
@transaction.atomic
|
|
@action_with_confirmation(action_name=action, extra_context=context)
|
|
def change_ticket_state(modeladmin, request, queryset, action=action, final_state=final_state):
|
|
form = ChangeReasonForm(request.POST)
|
|
if form.is_valid():
|
|
reason = form.cleaned_data['reason']
|
|
for ticket in queryset:
|
|
if ticket.state != final_state:
|
|
changes = {
|
|
'state': (ticket.state, final_state)
|
|
}
|
|
is_read = ticket.is_read_by(request.user)
|
|
getattr(ticket, action)()
|
|
msg = _("Marked as %s") % final_state.lower()
|
|
modeladmin.log_change(request, ticket, msg)
|
|
content = markdown_formated_changes(changes)
|
|
content += reason
|
|
ticket.messages.create(content=content, author=request.user)
|
|
if is_read and not ticket.is_read_by(request.user):
|
|
ticket.mark_as_read_by(request.user)
|
|
context = {
|
|
'count': queryset.count(),
|
|
'state': final_state.lower()
|
|
}
|
|
msg = _("%s selected tickets are now %s.") % context
|
|
modeladmin.message_user(request, msg)
|
|
else:
|
|
context['form'] = form
|
|
# action_with_confirmation must display form validation errors
|
|
return True
|
|
change_ticket_state.url_name = action
|
|
change_ticket_state.verbose_name = action
|
|
change_ticket_state.short_description = _('%s selected tickets') % action.capitalize()
|
|
change_ticket_state.description = _('Mark ticket as %s.') % final_state.lower()
|
|
change_ticket_state.__name__ = action
|
|
return change_ticket_state
|
|
|
|
|
|
action_map = {
|
|
Ticket.RESOLVED: 'resolve',
|
|
Ticket.REJECTED: 'reject',
|
|
Ticket.CLOSED: 'close'
|
|
}
|
|
|
|
|
|
thismodule = sys.modules[__name__]
|
|
for state, name in action_map.items():
|
|
action = change_ticket_state_factory(name, state)
|
|
setattr(thismodule, '%s_tickets' % name, action)
|
|
|
|
|
|
@transaction.atomic
|
|
def take_tickets(modeladmin, request, queryset):
|
|
for ticket in queryset:
|
|
if ticket.owner != request.user:
|
|
changes = {
|
|
'owner': (ticket.owner, request.user)
|
|
}
|
|
is_read = ticket.is_read_by(request.user)
|
|
ticket.take(request.user)
|
|
modeladmin.log_change(request, ticket, _("Taken"))
|
|
content = markdown_formated_changes(changes)
|
|
ticket.messages.create(content=content, author=request.user)
|
|
if is_read and not ticket.is_read_by(request.user):
|
|
ticket.mark_as_read_by(request.user)
|
|
modeladmin.log_change(request, ticket, 'Taken')
|
|
context = {
|
|
'count': queryset.count(),
|
|
'user': request.user
|
|
}
|
|
msg = _("%(count)s selected tickets are now owned by %(user)s.") % context
|
|
modeladmin.message_user(request, msg)
|
|
take_tickets.url_name = 'take'
|
|
take_tickets.short_description = _("Take selected tickets")
|
|
take_tickets.description = _("Make yourself owner of the ticket.")
|
|
|
|
|
|
@transaction.atomic
|
|
def mark_as_unread(modeladmin, request, queryset):
|
|
""" Mark a tickets as unread """
|
|
for ticket in queryset:
|
|
ticket.mark_as_unread_by(request.user)
|
|
modeladmin.log_change(request, ticket, 'Marked as unread')
|
|
num = len(queryset)
|
|
msg = ungettext(
|
|
_("Selected ticket has been marked as unread."),
|
|
_("%i selected tickets have been marked as unread.") % num,
|
|
num)
|
|
modeladmin.message_user(request, msg)
|
|
|
|
|
|
@transaction.atomic
|
|
def mark_as_read(modeladmin, request, queryset):
|
|
""" Mark a tickets as unread """
|
|
for ticket in queryset:
|
|
ticket.mark_as_read_by(request.user)
|
|
modeladmin.log_change(request, ticket, 'Marked as read')
|
|
num = len(queryset)
|
|
msg = ungettext(
|
|
_("Selected ticket has been marked as read."),
|
|
_("%i selected tickets have been marked as read.") % num,
|
|
num)
|
|
modeladmin.message_user(request, msg)
|
|
|
|
|
|
@transaction.atomic
|
|
def set_default_queue(modeladmin, request, queryset):
|
|
""" Set a queue as default issues queue """
|
|
if queryset.count() != 1:
|
|
messages.warning(request, _("Please, select only one queue."))
|
|
return
|
|
Queue.objects.filter(default=True).update(default=False)
|
|
queue = queryset.get()
|
|
queue.default = True
|
|
queue.save(update_fields=['default'])
|
|
modeladmin.log_change(request, queue, _("Chosen as default."))
|
|
messages.info(request, _("Chosen '%s' as default queue.") % queue)
|