96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
import logging
|
|
import os
|
|
|
|
from django.core.exceptions import SuspiciousOperation
|
|
from django.http import HttpResponseRedirect
|
|
from django.urls import reverse_lazy
|
|
from django.views import generic
|
|
|
|
from . import converters
|
|
from . import forms
|
|
from . import models
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class HomeView(generic.TemplateView):
|
|
template_name = 'dav_events/home.html'
|
|
|
|
|
|
class ImprintView(generic.TemplateView):
|
|
template_name = 'dav_events/imprint.html'
|
|
|
|
|
|
class EventListView(generic.ListView):
|
|
model = models.Event
|
|
queryset = models.Event.objects.all()
|
|
|
|
|
|
class EventDetailView(generic.DetailView):
|
|
model = models.Event
|
|
|
|
|
|
class EventCreateView(generic.FormView):
|
|
initial_form_class = forms.ModeForm
|
|
template_dir = os.path.join('dav_events', 'event_create')
|
|
default_template_name = 'default.html'
|
|
abort_url = reverse_lazy('dav_events:home')
|
|
|
|
def get_template_names(self):
|
|
form = self.get_form()
|
|
form_name = form.form_name
|
|
template = os.path.join(self.template_dir, '{}.html'.format(form_name))
|
|
default_template = os.path.join(self.template_dir, self.default_template_name)
|
|
return [
|
|
template, default_template
|
|
]
|
|
|
|
def get_form_class(self, form_name=None):
|
|
if form_name is not None:
|
|
form_class = getattr(forms, form_name)
|
|
elif 'dav_events_event_create_next_form_name' in self.request.session:
|
|
form_name = self.request.session['dav_events_event_create_next_form_name']
|
|
form_class = getattr(forms, form_name)
|
|
if not issubclass(form_class, forms.ChainedForm):
|
|
raise SuspiciousOperation('Invalid next form: {}'.format(form_name))
|
|
else:
|
|
form_class = self.initial_form_class
|
|
|
|
return form_class
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super(EventCreateView, self).get_context_data(**kwargs)
|
|
context['abort_url'] = self.abort_url
|
|
return context
|
|
|
|
def form_valid(self, form):
|
|
previous_data = converters.SessionDict(
|
|
self.request.session.get('dav_events_event_create_previous_data', dict()))
|
|
previous_data.update(form.cleaned_data)
|
|
|
|
event = converters.event_factory(previous_data)
|
|
|
|
next_form_name = form.next_form_name
|
|
if next_form_name:
|
|
self.request.session['dav_events_event_create_previous_data'] = previous_data.session_data
|
|
self.request.session['dav_events_event_create_next_form_name'] = next_form_name
|
|
next_form_class = self.get_form_class(next_form_name)
|
|
next_form = next_form_class(previous_data=previous_data)
|
|
return self.render_to_response(self.get_context_data(form=next_form, event=event))
|
|
else:
|
|
event.save()
|
|
self.clean_session_data()
|
|
return HttpResponseRedirect(reverse_lazy('dav_events:home'))
|
|
|
|
def clean_session_data(self, session=None):
|
|
if session is None:
|
|
session = self.request.session
|
|
if 'dav_events_event_create_next_form_name' in session:
|
|
del session['dav_events_event_create_next_form_name']
|
|
if 'dav_events_event_create_previous_data' in session:
|
|
del session['dav_events_event_create_previous_data']
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.clean_session_data(request.session)
|
|
return super(EventCreateView, self).get(request, *args, **kwargs)
|