OpenSlides/openslides/poll/views.py

from django.forms.models import modelform_factory
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404

from openslides.utils.views import TemplateView, FormMixin


class PollFormView(FormMixin, TemplateView):
    poll_class = None

    def get(self, request, *args, **kwargs):
        self.poll = self.get_object()
        return super(PollFormView, self).get(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        self.poll = self.get_object()
        option_forms = self.poll.get_vote_forms(data=self.request.POST)

        FormClass = self.get_modelform_class()
        pollform = FormClass(data=self.request.POST, instance=self.poll,
                             prefix='pollform')

        error = False
        for form in option_forms:
            if not form.is_valid():
                error = True

        if not pollform.is_valid():
            error = True

        if error:
            response = self.render_to_response(self.get_context_data(
                forms=option_forms,
                pollform=pollform))
        else:
            for form in option_forms:
                data = {}
                for value in self.poll.get_vote_values():
                    data[value] = form.cleaned_data[value]
                self.poll.set_vote_objects_with_values(form.option, data)

            pollform.save()
            response = HttpResponseRedirect(self.get_success_url())
        return response

    def get_poll_class(self):
        if self.poll_class is not None:
            return self.poll_class
        else:
            raise NotImplementedError(
                'No poll class defined. Either provide a poll_class or define '
                'a get_poll_class method.')

    def get_object(self):
        """
        Returns the poll object. Raises Http404 if the poll does not exist.
        """
        try:
            obj = self._object
        except AttributeError:
            queryset = self.get_poll_class().objects.filter(pk=self.kwargs['poll_id'])
            obj = get_object_or_404(queryset)
            self._object = obj
        return obj

    def get_context_data(self, **kwargs):
        context = super(PollFormView, self).get_context_data(**kwargs)
        context['poll'] = self.poll
        if 'forms' in kwargs:
            context['forms'] = kwargs['forms']
            context['pollform'] = kwargs['pollform']
        else:
            context['forms'] = self.poll.get_vote_forms()
            FormClass = self.get_modelform_class()
            context['pollform'] = FormClass(instance=self.poll,
                                            prefix='pollform')
        return context

    def get_modelform_class(self):
        fields = []
        self.poll.append_pollform_fields(fields)
        return modelform_factory(type(self.poll), fields=fields)