mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
970 B
Python
33 lines
970 B
Python
from django import forms
|
|
from voice.models import Vote, Feature
|
|
|
|
class VoteForm(forms.ModelForm):
|
|
|
|
class Meta:
|
|
model = Vote
|
|
exclude = ('used_twitter', 'used_facebook',)
|
|
|
|
def clean(self):
|
|
cleaned_data = self.cleaned_data
|
|
feature_id = cleaned_data.get('feature')
|
|
voter = cleaned_data.get('voter')
|
|
|
|
vote = Vote.objects.filter(feature=feature_id, voter=voter)
|
|
if vote.exists():
|
|
error = '%s has already voted on this feature' % voter
|
|
self._errors['feature'] = self._errors.get('feature', [])
|
|
self._errors['feature'].append(error)
|
|
raise forms.ValidationError(error)
|
|
|
|
return cleaned_data
|
|
|
|
class FeatureForm(forms.ModelForm):
|
|
|
|
description = forms.CharField(widget=forms.Textarea(
|
|
attrs={'class':'xxlarge', 'rows':3}))
|
|
voter = forms.EmailField()
|
|
|
|
class Meta:
|
|
model = Feature
|
|
exclude = ('votes_needed', 'created', 'state',)
|