from django.http import HttpResponseRedirect, Http404 from django.shortcuts import render from django.views.generic import CreateView, TemplateView, DetailView from django.core.urlresolvers import reverse_lazy from camps.models import Camp from .models import Ticket, TicketType from .forms import TicketForm class CampTicketSaleCheck(object): def dispatch(self, *args, **kwargs): current_camp = Camp.objects.current() if current_camp and current_camp.ticket_sale_open: return super().dispatch(*args, **kwargs) raise Http404() class TicketIndexView(CampTicketSaleCheck, TemplateView): template_name = "tickets/index.html" def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['ticket_types'] = TicketType.objects.all() return context class TicketDetailView(CampTicketSaleCheck, DetailView): model = Ticket template_name = 'tickets/detail.html' context_object_name = 'ticket' class BuyTicketView(CampTicketSaleCheck, CreateView): model = Ticket template_name = "tickets/buy.html" form_class = TicketForm def get_form_kwargs(self): kwargs = super().get_form_kwargs() ticket_type = self.request.GET.get('ticket_type', None) if ticket_type: kwargs['initial'] = { 'ticket_type': ticket_type } return kwargs def form_valid(self, form): instance = form.save(commit=False) instance.user = self.request.user instance.save() return HttpResponseRedirect( reverse_lazy('tickets:detail', kwargs={ 'pk': str(instance.pk) }) )