def create_poll(session, poll, user, chat, message=None):
"""Finish the poll creation."""
poll.created = True
user.expected_input = None
user.current_poll = None
if message:
message = message.edit_text(
get_poll_text(session, poll),
parse_mode='markdown',
reply_markup=get_management_keyboard(poll),
disable_web_page_preview=True,
)
else:
message = chat.send_message(
get_poll_text(session, poll),
parse_mode='markdown',
reply_markup=get_management_keyboard(poll),
disable_web_page_preview=True,
)
reference = Reference(poll,
admin_chat_id=chat.id,
admin_message_id=message.message_id)
session.add(reference)
session.commit()
def clone_poll(session, context, poll):
"""Clone this poll."""
new_poll = poll.clone(session)
session.commit()
context.tg_chat.send_message(
get_poll_management_text(session, new_poll),
parse_mode='markdown',
reply_markup=get_management_keyboard(new_poll))
context.query.answer(i18n.t('callback.cloned', locale=poll.user.locale))
def create_poll(session, poll, user, chat, message=None):
"""Finish the poll creation."""
poll.created = True
user.expected_input = None
user.current_poll = None
text = get_poll_text(session, poll)
if len(text) > 4000:
error_message = i18n.t("misc.over_4000", locale=user.locale)
message = chat.send_message(error_message, parse_mode="markdown")
session.delete(poll)
return
if message:
message = message.edit_text(
text,
parse_mode="markdown",
reply_markup=get_management_keyboard(poll),
disable_web_page_preview=True,
)
else:
message = chat.send_message(
text,
parse_mode="markdown",
reply_markup=get_management_keyboard(poll),
disable_web_page_preview=True,
)
if len(text) > 3000:
error_message = i18n.t("misc.over_3000", locale=user.locale)
message = chat.send_message(error_message, parse_mode="markdown")
reference = Reference(poll,
ReferenceType.admin.name,
user=user,
message_id=message.message_id)
session.add(reference)
session.commit()
increase_stat(session, "created_polls")
increase_user_stat(session, user, "created_polls")
def clone_poll(session, context, poll):
"""Clone this poll."""
new_poll = poll.clone(session)
session.commit()
context.tg_chat.send_message(
get_poll_text(session, new_poll),
parse_mode='markdown',
reply_markup=get_management_keyboard(new_poll),
disable_web_page_preview=True,
)
return i18n.t('callback.cloned', locale=poll.user.locale)
def show_menu(session, context, poll):
"""Replace the current message with the main poll menu."""
message = context.query.message
message.edit_text(
get_poll_management_text(session, poll),
parse_mode='markdown',
reply_markup=get_management_keyboard(poll),
)
reference = Reference(poll,
admin_chat_id=message.chat.id,
admin_message_id=message.message_id)
session.add(reference)
session.commit()
def send_updates(session, bot, poll, show_warning=False):
"""Actually update all messages."""
for reference in poll.references:
try:
# Admin poll management interface
if reference.admin_message_id is not None and not poll.in_settings:
text, keyboard = get_poll_text_and_vote_keyboard(session, poll, show_warning, show_back=True)
if poll.user.expected_input != ExpectedInput.votes.name:
keyboard = get_management_keyboard(poll)
try:
bot.edit_message_text(
text,
chat_id=reference.admin_chat_id,
message_id=reference.admin_message_id,
reply_markup=keyboard,
parse_mode='markdown',
disable_web_page_preview=True,
)
except Unauthorized as e:
if e.message == 'Forbidden: user is deactivated':
session.delete(reference)
# Edit message via inline_message_id
elif reference.inline_message_id is not None:
# Create text and keyboard
text, keyboard = get_poll_text_and_vote_keyboard(session, poll, show_warning)
bot.edit_message_text(
text,
inline_message_id=reference.inline_message_id,
reply_markup=keyboard,
parse_mode='markdown',
disable_web_page_preview=True,
)
except BadRequest as e:
if e.message.startswith('Message_id_invalid') or \
e.message.startswith("Message can't be edited") or \
e.message.startswith("Message to edit not found") or \
e.message.startswith("Chat not found"):
session.delete(reference)
session.commit()
elif e.message.startswith('Message is not modified'):
pass
else:
raise
def show_menu(session, context, poll):
"""Replace the current message with the main poll menu."""
message = context.query.message
message.edit_text(
get_poll_text(session, poll),
parse_mode='markdown',
reply_markup=get_management_keyboard(poll),
disable_web_page_preview=True,
)
reference = Reference(
poll,
admin_user=context.user,
admin_message_id=message.message_id
)
session.add(reference)
session.commit()
def go_back(session, context, poll):
"""Go back to the original step."""
if context.callback_result == CallbackResult.main_menu:
text = get_poll_text(session, poll)
keyboard = get_management_keyboard(poll)
poll.in_settings = False
elif context.callback_result == CallbackResult.settings:
text = get_settings_text(poll)
keyboard = get_settings_keyboard(poll)
context.query.message.edit_text(
text,
parse_mode='markdown',
reply_markup=keyboard,
disable_web_page_preview=True,
)
# Reset the expected input from the previous option
context.user.expected_input = None
def send_updates(session, bot, poll, show_warning=False):
"""Actually update all messages."""
for reference in poll.references:
try:
# Admin poll management interface
if reference.admin_message_id is not None and not poll.in_settings:
if poll.user.expected_input == ExpectedInput.votes.name:
keyboard = get_vote_keyboard(poll, show_back=True)
else:
keyboard = get_management_keyboard(poll)
text = get_poll_management_text(session, poll, show_warning)
bot.edit_message_text(
text,
chat_id=reference.admin_chat_id,
message_id=reference.admin_message_id,
reply_markup=keyboard,
parse_mode='markdown',
)
# Edit message via inline_message_id
elif reference.inline_message_id is not None:
# Create text and keyboard
text = get_poll_text(session, poll, show_warning)
keyboard = get_vote_keyboard(poll)
bot.edit_message_text(
text,
inline_message_id=reference.inline_message_id,
reply_markup=keyboard,
parse_mode='markdown',
)
except BadRequest as e:
if e.message.startswith('Message_id_invalid') or \
e.message.startswith("Message can't be edited"):
session.delete(reference)
session.commit()
elif e.message.startswith('Message is not modified'):
pass
else:
raise
def update_reference(session, bot, poll, reference, show_warning=False):
try:
# Admin poll management interface
if reference.type == ReferenceType.admin.name and not poll.in_settings:
text, keyboard = get_poll_text_and_vote_keyboard(
session,
poll,
user=poll.user,
show_warning=show_warning,
show_back=True)
if poll.user.expected_input != ExpectedInput.votes.name:
keyboard = get_management_keyboard(poll)
bot.edit_message_text(
text,
chat_id=reference.user.id,
message_id=reference.message_id,
reply_markup=keyboard,
parse_mode="markdown",
disable_web_page_preview=True,
)
# User that votes in private chat (priority vote)
elif reference.type == ReferenceType.private_vote.name:
text, keyboard = get_poll_text_and_vote_keyboard(
session,
poll,
user=reference.user,
show_warning=show_warning,
)
bot.edit_message_text(
text,
chat_id=reference.user.id,
message_id=reference.message_id,
reply_markup=keyboard,
parse_mode="markdown",
disable_web_page_preview=True,
)
# Edit message created via inline query
elif reference.type == ReferenceType.inline.name:
# Create text and keyboard
text, keyboard = get_poll_text_and_vote_keyboard(
session, poll, show_warning=show_warning)
bot.edit_message_text(
text,
inline_message_id=reference.bot_inline_message_id,
reply_markup=keyboard,
parse_mode="markdown",
disable_web_page_preview=True,
)
except BadRequest as e:
if (e.message.startswith("Message_id_invalid")
or e.message.startswith("Message can't be edited")
or e.message.startswith("Message to edit not found")
or e.message.startswith("Chat not found")
or e.message.startswith("Can't access the chat")):
session.delete(reference)
session.commit()
elif e.message.startswith("Message is not modified"):
pass
else:
raise
except Unauthorized:
session.delete(reference)
session.commit()
except TimedOut:
# Ignore timeouts during updates for now
pass
