Mercurial > public > sg101
annotate polls/forms.py @ 1151:f56798c96ec2
WIP convert downloads to V3 design.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Tue, 13 Dec 2016 21:56:21 -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) |