You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

171 lines
5.0 KiB
Python

from django.shortcuts import render, redirect, get_object_or_404
from django.contrib.auth.decorators import login_required
from django.http.response import HttpResponseNotFound
from django.utils import timezone
from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
from django.utils.translation import ugettext as _
from .models import Ticket, TicketMessage
from .forms import NewTicketForm, ReplyForm, StaffReplyForm
def common_context(request):
context = {
'open_n': Ticket.objects.filter(is_open=True, user=request.user).count(),
'closed_n': Ticket.objects.filter(is_open=False, user=request.user).count(),
}
if request.user.has_perm('tickets.view_any_ticket'):
context.update({
'all_open_n': Ticket.objects.filter(is_open=True).count(),
'all_closed_n': Ticket.objects.filter(is_open=False).count(),
})
return context
@login_required
def index(request, f=None, all=False):
tickets = Ticket.objects
if f == 'closed':
tickets = tickets.filter(is_open=False)
else:
tickets = tickets.filter(is_open=True)
if all is False or not request.user.has_perm('tickets.view_any_ticket'):
tickets = tickets.filter(user=request.user)
single_user = True
else:
single_user = False
paginator = Paginator(tickets, 20)
page = request.GET.get('page')
try:
tickets = paginator.page(page)
except PageNotAnInteger:
tickets = paginator.page(1)
except EmptyPage:
tickets = paginator.page(paginator.num_pages)
context = dict(
tickets=tickets,
filter=f,
single_user=single_user,
title=_("Tickets"),
)
context.update(common_context(request))
if not f:
return render(request, 'tickets/index.html', context)
else:
return render(request, 'tickets/list.html', context)
@login_required
def new(request):
context = common_context(request)
context['title'] = _("New Ticket")
if request.method != 'POST':
context['form'] = NewTicketForm()
return render(request, 'tickets/new.html', context)
form = NewTicketForm(request.POST)
if not form.is_valid():
context['form'] = form
return render(request, 'tickets/new.html', context)
ticket = Ticket(category=form.cleaned_data['category'],
subject=form.cleaned_data['subject'],
user=request.user)
ticket.save()
firstmsg = TicketMessage(ticket=ticket, user=request.user,
message=form.cleaned_data['message'])
if not request.user.is_staff:
firstmsg.remote_addr = request.META['REMOTE_ADDR']
firstmsg.save()
ticket.notify_new(firstmsg)
return redirect('tickets:view', id=ticket.id)
@login_required
def view(request, id):
ticket = get_object_or_404(Ticket, id=id)
view_any_ticket = request.user.has_perm('tickets.view_any_ticket')
reply_any_ticket = request.user.has_perm('tickets.reply_any_ticket')
if not view_any_ticket and ticket.user != request.user:
return HttpResponseNotFound()
if request.user.has_perm('tickets.view_private_message'):
messages = ticket.message_set.all()
else:
messages = ticket.message_set.filter(staff_only=False)
if request.method != 'POST':
if request.user.is_staff:
form = StaffReplyForm()
else:
form = ReplyForm()
ctx = dict(
ticket=ticket,
ticket_messages=messages,
form=form,
title=_("Ticket:") + " " + ticket.subject,
)
ctx.update(common_context(request))
return render(request, 'tickets/view.html', ctx)
if not reply_any_ticket and ticket.user != request.user:
return HttpResponseNotFound()
if request.POST.get('close') or request.POST.get('button_close'):
ticket.is_open = False
ticket.closed = timezone.now()
ticket.save()
return redirect('tickets:view', id=ticket.id)
if request.POST.get('reopen') or request.POST.get('button_reopen'):
ticket.is_open = True
ticket.save()
return redirect('tickets:view', id=ticket.id)
if request.user.has_perm('tickets.post_private_message'):
form = StaffReplyForm(request.POST)
else:
form = ReplyForm(request.POST)
if not form.is_valid():
ctx = dict(
ticket=ticket,
ticket_messages=messages,
form=form,
title=_("Ticket:") + " " + ticket.subject,
)
ctx.update(common_context(request))
return render(request, 'tickets/view.html', ctx)
msg = TicketMessage(ticket=ticket, user=request.user,
**form.cleaned_data)
if not request.user.is_staff:
msg.remote_addr = request.META['REMOTE_ADDR']
msg.save()
if not ticket.is_open:
ticket.is_open = True
ticket.save()
ticket.notify_reply(msg)
return redirect('tickets:view', id=ticket.id)