annotate polls/forms.py @ 861:e4f8d87c3d30
Configure Markdown logger to reduce noise in logs.
Markdown is logging at the INFO level whenever it loads an extension.
This looks like it has been fixed in master at GitHub. But until then
we will explicitly configure the MARKDOWN logger to log at WARNING
or higher.
| author |
Brian Neal <bgneal@gmail.com> |
| date |
Mon, 01 Dec 2014 18:36:27 -0600 |
| 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)
|
