50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from app import db
|
|
from app.models.poll import Poll, PollAnswer
|
|
from collections import Counter
|
|
|
|
class SimplePoll(Poll):
|
|
"""Poll with only one answer allowed"""
|
|
|
|
__tablename__ = 'simplepoll'
|
|
|
|
# Names of templates
|
|
template = 'simplepoll.html'
|
|
|
|
__mapper_args__ = {
|
|
'polymorphic_identity': __tablename__,
|
|
}
|
|
|
|
def __init__(self, author, title, choices, **kwargs):
|
|
choices = [Choice(i, t) for i, t in enumerate(choices)]
|
|
super().__init__(author, title, choices, **kwargs)
|
|
|
|
# Mandatory methods
|
|
def vote(self, user, request):
|
|
try:
|
|
choice_id = int(request.form['pollanwsers'])
|
|
except (KeyError, ValueError):
|
|
return None
|
|
|
|
answer = PollAnswer(self, user, choice_id)
|
|
return answer
|
|
|
|
@property
|
|
def results(self):
|
|
values = {c: 0 for c in self.choices}
|
|
counter = Counter(values)
|
|
answers = [self.choice_from_id(a.answer) for a in self.answers]
|
|
counter.update(answers)
|
|
return counter
|
|
|
|
# Custom method
|
|
def choice_from_id(self, id):
|
|
for c in self.choices:
|
|
if c.id == id:
|
|
return c
|
|
return None
|
|
|
|
class Choice():
|
|
def __init__(self, id, title):
|
|
self.id = id
|
|
self.title = title
|