66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
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)
|