Mercurial > public > sg101
view polls/forms.py @ 1160:b84ce2d7e93d
New template for displaying django messages.
New common template for rendering django messages in the new V3 Foundation
style.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Tue, 24 Jan 2017 19:22:29 -0600 (2017-01-25) |
parents | ee87ea74d46b |
children |
line wrap: on
line source
"""Forms for the Polls application.""" from django import forms from django.db.models import F from polls.models import Choice class VoteForm(forms.Form): """Form for voting in a poll.""" choices = forms.ModelChoiceField(label='', empty_label=None, queryset=Choice.objects.none(), widget=forms.RadioSelect) def __init__(self, poll, *args, **kwargs): self.user = kwargs.pop('user', None) self.user_choice = kwargs.pop('user_choice', None) super(VoteForm, self).__init__(*args, **kwargs) self.fields['choices'].queryset = poll.choice_set.all() def clean(self): if self.user_choice: raise forms.ValidationError("You've already voted in this poll!") return self.cleaned_data def save(self): choice = self.cleaned_data['choices'] Choice.objects.filter(id=choice.id).update(votes=F('votes') + 1) choice.voters.add(self.user)