def on_callback_query(query: tg.CallbackQuery):
query_id = query.id
telegram_user_id = query.from_user.id
chat = bot_backend.get_chat_by_telegram_id(telegram_user_id, True,
Chat.Sources.CALLBACK)
data = parse_get_mode_control_button_data(query.data)
if data[0] not in CALLBACK_ACTIONS:
bot.answer_callback_query(query_id, t('bot.callback.error'))
return
session_id = int(data[1])
session = bot_backend.get_session_by_id(session_id)
if session is None:
bot.answer_callback_query(query_id, t('bot.callback.error'))
return
if session.chat.id != chat.id:
bot.answer_callback_query(query_id, t('bot.callback.error'))
return
if data[0] == CALLBACK_ACTION_CHANGE_MODE:
mode = data[2]
if mode not in [
Session.Modes.HORIZONTAL, Session.Modes.VERTICAL,
Session.Modes.TWO_ROWS, Session.Modes.SQUARE
]:
mode = Session.Modes.HORIZONTAL
bot_backend.change_mode(session, mode)
bot_backend.push_history_item(session)
send_combined_image(session)
def on_cancel_command(message):
telegram_chat_id = message.chat.id
chat = bot_backend.get_chat_by_telegram_id(telegram_chat_id, True,
Chat.Sources.CANCEL)
if bot_backend.chat_has_active_session(chat):
session = bot_backend.get_active_session(chat)
bot_backend.cancel_session(session)
bot.send_message(telegram_chat_id, t('bot.messages.cancel'))
def on_soedini_command(message: tg.Message):
telegram_chat_id = message.chat.id
chat = bot_backend.get_chat_by_telegram_id(telegram_chat_id, True,
Chat.Sources.SOEDINI)
if not bot_backend.chat_has_active_session(chat):
bot.send_message(chat.telegram_id, t('bot.state.error'))
return
session = bot_backend.get_active_session(chat)
photos = session.photo_set.all()
if len(photos) == 0:
bot.send_message(chat.telegram_id, t('bot.state.error'))
return
bot_backend.finish_session(session)
bot_backend.push_history_item(session)
send_combined_image(session)
def on_photo_message(message: tg.Message):
telegram_chat_id = message.chat.id
if len(message.photo) == 0:
return
file_id = message.photo[-1].file_id
chat = bot_backend.get_chat_by_telegram_id(telegram_chat_id, True,
Chat.Sources.PHOTO)
new_session_created = False
if not bot_backend.chat_has_active_session(chat):
active_session = bot_backend.create_session(chat)
new_session_created = True
else:
active_session = bot_backend.get_active_session(chat)
bot_backend.add_photo_to_session(active_session, file_id)
if new_session_created:
bot.send_message(chat.telegram_id, t('bot.messages.continue'))
def on_help_command(message: tg.Message):
telegram_chat_id = message.chat.id
chat = bot_backend.get_chat_by_telegram_id(telegram_chat_id, True,
Chat.Sources.HELP)
bot.send_message(chat.telegram_id, t('bot.messages.start'))