bornhack-website/src/tickets/views.py

import logging

from django.contrib import messages
from django.contrib.auth.mixins import LoginRequiredMixin
from django.views.generic.detail import SingleObjectMixin
from django.views.generic import (
    DetailView,
    UpdateView,
    ListView,
    View
)
from django.http import (
    HttpResponse,
    Http404
)

from .models import ShopTicket
logger = logging.getLogger("bornhack.%s" % __name__)


class ShopTicketListView(LoginRequiredMixin, ListView):
    model = ShopTicket
    template_name = 'ticket_list.html'
    context_object_name = 'tickets'

    def get_queryset(self):
        tickets = super(ShopTicketListView, self).get_queryset()
        user = self.request.user
        return tickets.filter(order__user=user)


class ShopTicketDownloadView(LoginRequiredMixin, SingleObjectMixin, View):
    model = ShopTicket

    def dispatch(self, request, *args, **kwargs):
        if not request.user == self.get_object().order.user:
            raise Http404("Ticket not found")

        return super().dispatch(request, *args, **kwargs)

    def get(self, request, *args, **kwargs):
        response = HttpResponse(content_type='application/pdf')
        response['Content-Disposition'] = 'attachment; filename="{type}_ticket_{pk}.pdf"'.format(
            type=self.get_object().shortname,
            pk=self.get_object().pk
        )
        response.write(self.get_object().generate_pdf().getvalue())
        return response


class ShopTicketDetailView(LoginRequiredMixin, UpdateView, DetailView):
    model = ShopTicket
    template_name = 'ticket_detail.html'
    context_object_name = 'ticket'
    fields = ['name', 'email']

    def form_valid(self, form):
        messages.info(self.request, 'Ticket updated!')
        return super().form_valid(form)

    def dispatch(self, request, *args, **kwargs):
        ticket = self.get_object()
        if ticket.order.user != request.user:
            raise Http404("Ticket not found")
        return super().dispatch(request, *args, **kwargs)