annotate polls/forms.py @ 661:15dbe0ccda95
Prevent exceptions when viewing downloads in the admin when the file
doesn't exist on the filesystem. This is usually seen in development
but can also happen in production if the file is missing.
| author |
Brian Neal <bgneal@gmail.com> |
| date |
Tue, 14 May 2013 21:02:47 -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)
|
