annotate gpp/polls/forms.py @ 505:a5d11471d031
Refactor the logic in the rate limiter decorator. Check to see if the request was ajax, as the ajax view always returns 200. Have to decode the JSON response to see if an error occurred or not.
| author |
Brian Neal <bgneal@gmail.com> |
| date |
Sat, 03 Dec 2011 19:13:38 +0000 |
| parents |
1f139de929c4 |
| 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)
|
