diff options
Diffstat (limited to 'polls')
| -rw-r--r-- | polls/feeds.py | 3 | ||||
| -rw-r--r-- | polls/models.py | 5 |
2 files changed, 5 insertions, 3 deletions
diff --git a/polls/feeds.py b/polls/feeds.py index c2c79be..2d52dc7 100644 --- a/polls/feeds.py +++ b/polls/feeds.py @@ -26,9 +26,6 @@ from django.utils.translation import gettext_lazy as _ from papillon.settings import BASE_SITE from papillon.polls.models import Poll, Vote, Voter -class VoterFeedObject: - def __init__(self, voter): - self.voter = voter class PollLatestEntries(Feed): def get_object(self, poll_url): diff --git a/polls/models.py b/polls/models.py index 4a7f584..531c1e3 100644 --- a/polls/models.py +++ b/polls/models.py @@ -81,6 +81,11 @@ class Poll(models.Model): comment.delete() self.delete() + def getChoices(self): + """ + Get choices associated to this vote""" + return Choice.objects.filter(poll=self) + class Admin: pass class Meta: |