annotate polls/forms.py @ 955:71a671dab55d
First commit of whitelisting image hosts.
This is behind a feature flag courtesy of waffle.
author |
Brian Neal <bgneal@gmail.com> |
date |
Wed, 03 Jun 2015 21:13:08 -0500 |
parents |
ee87ea74d46b |
children |
|
rev |
line source |
gremmie@1
|
1 """Forms for the Polls application."""
|
gremmie@1
|
2
|
gremmie@1
|
3 from django import forms
|
bgneal@439
|
4 from django.db.models import F
|
gremmie@1
|
5
|
gremmie@1
|
6 from polls.models import Choice
|
gremmie@1
|
7
|
gremmie@1
|
8
|
gremmie@1
|
9 class VoteForm(forms.Form):
|
bgneal@439
|
10 """Form for voting in a poll."""
|
bgneal@439
|
11 choices = forms.ModelChoiceField(label='', empty_label=None,
|
bgneal@439
|
12 queryset=Choice.objects.none(), widget=forms.RadioSelect)
|
gremmie@1
|
13
|
bgneal@439
|
14 def __init__(self, poll, *args, **kwargs):
|
bgneal@439
|
15 self.user = kwargs.pop('user', None)
|
bgneal@439
|
16 self.user_choice = kwargs.pop('user_choice', None)
|
bgneal@439
|
17 super(VoteForm, self).__init__(*args, **kwargs)
|
bgneal@439
|
18 self.fields['choices'].queryset = poll.choice_set.all()
|
gremmie@1
|
19
|
bgneal@439
|
20 def clean(self):
|
bgneal@439
|
21 if self.user_choice:
|
bgneal@439
|
22 raise forms.ValidationError("You've already voted in this poll!")
|
bgneal@439
|
23 return self.cleaned_data
|
bgneal@439
|
24
|
bgneal@439
|
25 def save(self):
|
bgneal@439
|
26 choice = self.cleaned_data['choices']
|
bgneal@439
|
27 Choice.objects.filter(id=choice.id).update(votes=F('votes') + 1)
|
bgneal@439
|
28 choice.voters.add(self.user)
|