Refactored how we save a vote.

polls/forms.py

@@ -15,3 +15,8 @@ def __init__(self, *args, **kwargs):
         # This has to be done here (instead of declaratively) because the
         # ``Poll`` instance will change from request to request.
         self.fields['choice'] = forms.ModelChoiceField(queryset=Choice.objects.filter(poll=self.instance.pk), empty_label=None, widget=forms.RadioSelect)
+
+    def save(self):
+        choice = self.cleaned_data['choice']
+        choice.record_vote()
+        return choice

polls/models.py

@@ -21,3 +21,7 @@ class Choice(models.Model):
 
     def __unicode__(self):
         return self.choice
+
+    def record_vote(self):
+        self.votes += 1
+        self.save()

polls/views.py

@@ -18,10 +18,7 @@ def detail(request, poll_id):
         form = PollForm(request.POST, instance=p)
 
         if form.is_valid():
-            choice = form.cleaned_data['choice']
-            choice.votes += 1
-            choice.save()
-
+            form.save()
             return HttpResponseRedirect(reverse('polls_results', kwargs={'poll_id': p.id}))
     else:
         form = PollForm(instance=p)