def get_remaining_votes(session, poll):
"""Get the remaining votes for a poll."""
if not poll_has_limited_votes(poll) \
or not poll.should_show_result() \
or poll.anonymous:
return []
user_vote_count = func.sum(Vote.vote_count).label('user_vote_count')
remaining_user_votes = session.query(User.name, user_vote_count) \
.join(Vote) \
.filter(Vote.poll == poll) \
.group_by(User.name) \
.having(user_vote_count < poll.number_of_votes) \
.order_by(User.name) \
.all()
if len(remaining_user_votes) == 0:
return []
lines = []
lines.append(i18n.t('poll.remaining_votes', locale=poll.locale))
for user_votes in remaining_user_votes:
lines.append(
i18n.t('poll.remaining_votes_user',
locale=poll.locale,
name=user_votes[0],
count=poll.number_of_votes - user_votes[1]))
return lines
def __init__(self, session, poll):
self.total_user_count = session.query(User.id) \
.join(Vote) \
.join(PollOption) \
.filter(PollOption.poll == poll) \
.group_by(User.id) \
.count()
# Flags
self.anonymous = poll.anonymous
self.show_results = poll.should_show_result()
self.show_percentage = poll.show_percentage
self.limited_votes = poll_has_limited_votes(poll)
def get_poll_text(session, poll, show_warning):
"""Create the text of the poll."""
total_user_count = session.query(User.id) \
.join(Vote) \
.join(PollOption) \
.filter(PollOption.poll == poll) \
.group_by(User.id) \
.count()
# Name and description
lines = []
lines.append(f'✉️ *{poll.name}*')
if poll.description is not None:
lines.append(f'_{poll.description}_')
# Anonymity information
if not poll.results_visible and not poll.should_show_result() or \
poll.anonymous and not poll.closed:
lines.append('')
if poll.anonymous and not poll.closed:
lines.append(f"_{i18n.t('poll.anonymous', locale=poll.locale)}_")
if not poll.results_visible and not poll.should_show_result():
lines.append(
f"_{i18n.t('poll.results_not_visible', locale=poll.locale)}_")
# Sort the options accordingly to the polls settings
options = get_sorted_options(poll, total_user_count)
# All options with their respective people percentage
for index, option in enumerate(options):
lines.append('')
lines.append(get_option_line(session, option, index))
if option.description is not None:
lines.append(f'┆ _{option.description}_')
if poll.should_show_result() and poll.show_percentage:
lines.append(get_percentage_line(option, total_user_count))
# Add the names of the voters to the respective options
if poll.should_show_result() and not poll.anonymous and len(
option.votes) > 0:
# Sort the votes accordingly to the poll's settings
votes = get_sorted_votes(poll, option.votes)
for index, vote in enumerate(votes):
vote_line = get_vote_line(poll, option, vote, index)
lines.append(vote_line)
lines.append('')
if poll_has_limited_votes(poll):
lines.append(
i18n.t('poll.vote_times',
locale=poll.locale,
amount=poll.number_of_votes))
# Total user count information
information_line = get_vote_information_line(poll, total_user_count)
if information_line is not None:
lines.append(information_line)
if not poll.anonymous:
remaining_votes = get_remaining_votes(session, poll)
lines += remaining_votes
if poll.due_date is not None:
lines.append(
i18n.t('poll.due',
locale=poll.locale,
date=poll.get_formatted_due_date()))
# Notify users that poll is closed
if poll.closed:
lines.append(i18n.t('poll.closed', locale=poll.locale))
if show_warning:
lines.append(i18n.t('poll.too_many_votes', locale=poll.locale))
return '\n'.join(lines)