ccvpn/tickets/views.py

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 ccvpn.common import get_client_ip
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 = get_client_ip(request)

    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':
        ctx = dict(
            staff_reply=request.user.has_perm('tickets.post_private_message'),
            ticket=ticket,
            ticket_messages=messages,
            form=ReplyForm(),
            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(
            staff_reply=request.user.has_perm('tickets.post_private_message'),
            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 = get_client_ip(request)

    msg.save()

    if not ticket.is_open:
        ticket.is_open = True
        ticket.save()

    ticket.notify_reply(msg)

    return redirect('tickets:view', id=ticket.id)