41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
|
from django.views.generic import TemplateView
|
||
|
|
from django.http import HttpResponseRedirect
|
||
|
|
|
||
|
|
class PollFormView(TemplateView):
|
||
|
|
template_name = 'poll/poll.html'
|
||
|
|
poll_argument = 'poll_id'
|
||
|
|
|
||
|
|
def set_poll(self, poll_id):
|
||
|
|
poll_id = poll_id
|
||
|
|
self.poll = self.poll_class.objects.get(pk=poll_id)
|
||
|
|
self.poll.vote_values = self.vote_values
|
||
|
|
|
||
|
|
def get_context_data(self, **kwargs):
|
||
|
|
context = super(PollFormView, self).get_context_data(**kwargs)
|
||
|
|
self.set_poll(self.kwargs['poll_id'])
|
||
|
|
context['poll'] = self.poll
|
||
|
|
if not 'forms' in context:
|
||
|
|
context['forms'] = context['poll'].get_vote_forms()
|
||
|
|
return context
|
||
|
|
|
||
|
|
def get_success_url(self):
|
||
|
|
return self.success_url
|
||
|
|
|
||
|
|
def post(self, request, *args, **kwargs):
|
||
|
|
context = self.get_context_data(**kwargs)
|
||
|
|
forms = self.poll.get_vote_forms(data=self.request.POST)
|
||
|
|
error = False
|
||
|
|
for form in forms:
|
||
|
|
if not form.is_valid():
|
||
|
|
error = True
|
||
|
|
if error:
|
||
|
|
return self.render_to_response(self.get_context_data(forms=forms))
|
||
|
|
|
||
|
|
for form in forms:
|
||
|
|
data = {}
|
||
|
|
for value in self.poll.vote_values:
|
||
|
|
data[value] = form.cleaned_data[value]
|
||
|
|
print data
|
||
|
|
self.poll.set_form_values(form.option, data)
|
||
|
|
return HttpResponseRedirect(self.get_success_url())
|