Davte/davtelepot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Pass language to update handlers

Browse Source
This commit is contained in:
Davte
2020-05-06 00:39:23 +02:00
parent dddaf20ea2
commit a2c83f8d9c
1 changed files with 56 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
davtelepot/bot.py
View File

@@ -17,11 +17,11 @@ Usage
database_url='my_other_db') database_url='my_other_db')
@long_polling_bot.command('/foo') @long_polling_bot.command('/foo')
async def foo_command(bot, update, user_record): async def foo_command(bot, update, user_record, language):
return "Bar!" return "Bar!"
@webhook_bot.command('/bar') @webhook_bot.command('/bar')
async def bar_command(bot, update, user_record): async def bar_command(bot, update, user_record, language):
return "Foo!" return "Foo!"
exit_state = Bot.run( exit_state = Bot.run(
@@ -579,17 +579,19 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
def administrators(self): def administrators(self):
return self._get_administrators(self) return self._get_administrators(self)
async def message_router(self, update, user_record): async def message_router(self, update, user_record, language):
"""Route Telegram `message` update to appropriate message handler.""" """Route Telegram `message` update to appropriate message handler."""
for key, value in update.items(): for key, value in update.items():
if key in self.message_handlers: if key in self.message_handlers:
return await self.message_handlers[key](update, user_record) return await self.message_handlers[key](update=update,
user_record=user_record,
language=language)
logging.error( logging.error(
f"The following message update was received: {update}\n" f"The following message update was received: {update}\n"
"However, this message type is unknown." "However, this message type is unknown."
) )
async def edited_message_handler(self, update, user_record): async def edited_message_handler(self, update, user_record, language=None):
"""Handle Telegram `edited_message` update.""" """Handle Telegram `edited_message` update."""
logging.info( logging.info(
f"The following update was received: {update}\n" f"The following update was received: {update}\n"
@@ -597,7 +599,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def channel_post_handler(self, update, user_record): async def channel_post_handler(self, update, user_record, language=None):
"""Handle Telegram `channel_post` update.""" """Handle Telegram `channel_post` update."""
logging.info( logging.info(
f"The following update was received: {update}\n" f"The following update was received: {update}\n"
@@ -605,7 +607,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def edited_channel_post_handler(self, update, user_record): async def edited_channel_post_handler(self, update, user_record, language=None):
"""Handle Telegram `edited_channel_post` update.""" """Handle Telegram `edited_channel_post` update."""
logging.info( logging.info(
f"The following update was received: {update}\n" f"The following update was received: {update}\n"
@@ -613,7 +615,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def inline_query_handler(self, update, user_record): async def inline_query_handler(self, update, user_record, language=None):
"""Handle Telegram `inline_query` update. """Handle Telegram `inline_query` update.
Answer it with results or log errors. Answer it with results or log errors.
@@ -648,7 +650,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
logging.info("Error answering inline query\n{}".format(e)) logging.info("Error answering inline query\n{}".format(e))
return return
async def chosen_inline_result_handler(self, update, user_record): async def chosen_inline_result_handler(self, update, user_record, language=None):
"""Handle Telegram `chosen_inline_result` update.""" """Handle Telegram `chosen_inline_result` update."""
if user_record is not None: if user_record is not None:
user_id = user_record['telegram_id'] user_id = user_record['telegram_id']
@@ -678,7 +680,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
self.chosen_inline_result_handlers[user_id][result_id] = handler self.chosen_inline_result_handlers[user_id][result_id] = handler
return return
async def callback_query_handler(self, update, user_record): async def callback_query_handler(self, update, user_record, language=None):
"""Handle Telegram `callback_query` update. """Handle Telegram `callback_query` update.
A callback query is sent when users press inline keyboard buttons. A callback query is sent when users press inline keyboard buttons.
@@ -733,7 +735,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
logging.error(e) logging.error(e)
return return
async def shipping_query_handler(self, update, user_record): async def shipping_query_handler(self, update, user_record, language=None):
"""Handle Telegram `shipping_query` update.""" """Handle Telegram `shipping_query` update."""
logging.info( logging.info(
f"The following update was received: {update}\n" f"The following update was received: {update}\n"
@@ -741,7 +743,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def pre_checkout_query_handler(self, update, user_record): async def pre_checkout_query_handler(self, update, user_record, language=None):
"""Handle Telegram `pre_checkout_query` update.""" """Handle Telegram `pre_checkout_query` update."""
logging.info( logging.info(
f"The following update was received: {update}\n" f"The following update was received: {update}\n"
@@ -749,7 +751,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def poll_handler(self, update, user_record): async def poll_handler(self, update, user_record, language=None):
"""Handle Telegram `poll` update.""" """Handle Telegram `poll` update."""
logging.info( logging.info(
f"The following update was received: {update}\n" f"The following update was received: {update}\n"
@@ -757,7 +759,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def text_message_handler(self, update, user_record): async def text_message_handler(self, update, user_record, language=None):
"""Handle `text` message update.""" """Handle `text` message update."""
replier, reply = None, None replier, reply = None, None
text = update['text'].lower() text = update['text'].lower()
@@ -817,56 +819,56 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def audio_file_handler(self, update, user_record): async def audio_file_handler(self, update, user_record, language=None):
"""Handle `audio` file update.""" """Handle `audio` file update."""
logging.info( logging.info(
"A audio file update was received, " "A audio file update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def document_message_handler(self, update, user_record): async def document_message_handler(self, update, user_record, language=None):
"""Handle `document` message update.""" """Handle `document` message update."""
logging.info( logging.info(
"A document message update was received, " "A document message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def animation_message_handler(self, update, user_record): async def animation_message_handler(self, update, user_record, language=None):
"""Handle `animation` message update.""" """Handle `animation` message update."""
logging.info( logging.info(
"A animation message update was received, " "A animation message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def game_message_handler(self, update, user_record): async def game_message_handler(self, update, user_record, language=None):
"""Handle `game` message update.""" """Handle `game` message update."""
logging.info( logging.info(
"A game message update was received, " "A game message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def photo_message_handler(self, update, user_record): async def photo_message_handler(self, update, user_record, language=None):
"""Handle `photo` message update.""" """Handle `photo` message update."""
logging.info( logging.info(
"A photo message update was received, " "A photo message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def sticker_message_handler(self, update, user_record): async def sticker_message_handler(self, update, user_record, language=None):
"""Handle `sticker` message update.""" """Handle `sticker` message update."""
logging.info( logging.info(
"A sticker message update was received, " "A sticker message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def video_message_handler(self, update, user_record): async def video_message_handler(self, update, user_record, language=None):
"""Handle `video` message update.""" """Handle `video` message update."""
logging.info( logging.info(
"A video message update was received, " "A video message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def voice_message_handler(self, update, user_record): async def voice_message_handler(self, update, user_record, language=None):
"""Handle `voice` message update.""" """Handle `voice` message update."""
replier, reply = None, None replier, reply = None, None
user_id = update['from']['id'] if 'from' in update else None user_id = update['from']['id'] if 'from' in update else None
@@ -896,21 +898,21 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def video_note_message_handler(self, update, user_record): async def video_note_message_handler(self, update, user_record, language=None):
"""Handle `video_note` message update.""" """Handle `video_note` message update."""
logging.info( logging.info(
"A video_note message update was received, " "A video_note message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def contact_message_handler(self, update, user_record): async def contact_message_handler(self, update, user_record, language=None):
"""Handle `contact` message update.""" """Handle `contact` message update."""
logging.info( logging.info(
"A contact message update was received, " "A contact message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def location_message_handler(self, update, user_record): async def location_message_handler(self, update, user_record, language=None):
"""Handle `location` message update.""" """Handle `location` message update."""
replier, reply = None, None replier, reply = None, None
user_id = update['from']['id'] if 'from' in update else None user_id = update['from']['id'] if 'from' in update else None
@@ -940,56 +942,56 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
return return
async def venue_message_handler(self, update, user_record): async def venue_message_handler(self, update, user_record, language=None):
"""Handle `venue` message update.""" """Handle `venue` message update."""
logging.info( logging.info(
"A venue message update was received, " "A venue message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def poll_message_handler(self, update, user_record): async def poll_message_handler(self, update, user_record, language=None):
"""Handle `poll` message update.""" """Handle `poll` message update."""
logging.info( logging.info(
"A poll message update was received, " "A poll message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def new_chat_members_message_handler(self, update, user_record): async def new_chat_members_message_handler(self, update, user_record, language=None):
"""Handle `new_chat_members` message update.""" """Handle `new_chat_members` message update."""
logging.info( logging.info(
"A new_chat_members message update was received, " "A new_chat_members message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def left_chat_member_message_handler(self, update, user_record): async def left_chat_member_message_handler(self, update, user_record, language=None):
"""Handle `left_chat_member` message update.""" """Handle `left_chat_member` message update."""
logging.info( logging.info(
"A left_chat_member message update was received, " "A left_chat_member message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def new_chat_title_message_handler(self, update, user_record): async def new_chat_title_message_handler(self, update, user_record, language=None):
"""Handle `new_chat_title` message update.""" """Handle `new_chat_title` message update."""
logging.info( logging.info(
"A new_chat_title message update was received, " "A new_chat_title message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def new_chat_photo_message_handler(self, update, user_record): async def new_chat_photo_message_handler(self, update, user_record, language=None):
"""Handle `new_chat_photo` message update.""" """Handle `new_chat_photo` message update."""
logging.info( logging.info(
"A new_chat_photo message update was received, " "A new_chat_photo message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def delete_chat_photo_message_handler(self, update, user_record): async def delete_chat_photo_message_handler(self, update, user_record, language=None):
"""Handle `delete_chat_photo` message update.""" """Handle `delete_chat_photo` message update."""
logging.info( logging.info(
"A delete_chat_photo message update was received, " "A delete_chat_photo message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def group_chat_created_message_handler(self, update, user_record): async def group_chat_created_message_handler(self, update, user_record, language=None):
"""Handle `group_chat_created` message update.""" """Handle `group_chat_created` message update."""
logging.info( logging.info(
"A group_chat_created message update was received, " "A group_chat_created message update was received, "
@@ -1004,63 +1006,63 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def channel_chat_created_message_handler(self, update, user_record): async def channel_chat_created_message_handler(self, update, user_record, language=None):
"""Handle `channel_chat_created` message update.""" """Handle `channel_chat_created` message update."""
logging.info( logging.info(
"A channel_chat_created message update was received, " "A channel_chat_created message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def migrate_to_chat_id_message_handler(self, update, user_record): async def migrate_to_chat_id_message_handler(self, update, user_record, language=None):
"""Handle `migrate_to_chat_id` message update.""" """Handle `migrate_to_chat_id` message update."""
logging.info( logging.info(
"A migrate_to_chat_id message update was received, " "A migrate_to_chat_id message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def migrate_from_chat_id_message_handler(self, update, user_record): async def migrate_from_chat_id_message_handler(self, update, user_record, language=None):
"""Handle `migrate_from_chat_id` message update.""" """Handle `migrate_from_chat_id` message update."""
logging.info( logging.info(
"A migrate_from_chat_id message update was received, " "A migrate_from_chat_id message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def pinned_message_message_handler(self, update, user_record): async def pinned_message_message_handler(self, update, user_record, language=None):
"""Handle `pinned_message` message update.""" """Handle `pinned_message` message update."""
logging.info( logging.info(
"A pinned_message message update was received, " "A pinned_message message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def invoice_message_handler(self, update, user_record): async def invoice_message_handler(self, update, user_record, language=None):
"""Handle `invoice` message update.""" """Handle `invoice` message update."""
logging.info( logging.info(
"A invoice message update was received, " "A invoice message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def successful_payment_message_handler(self, update, user_record): async def successful_payment_message_handler(self, update, user_record, language=None):
"""Handle `successful_payment` message update.""" """Handle `successful_payment` message update."""
logging.info( logging.info(
"A successful_payment message update was received, " "A successful_payment message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def connected_website_message_handler(self, update, user_record): async def connected_website_message_handler(self, update, user_record, language=None):
"""Handle `connected_website` message update.""" """Handle `connected_website` message update."""
logging.info( logging.info(
"A connected_website message update was received, " "A connected_website message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def passport_data_message_handler(self, update, user_record): async def passport_data_message_handler(self, update, user_record, language=None):
"""Handle `passport_data` message update.""" """Handle `passport_data` message update."""
logging.info( logging.info(
"A passport_data message update was received, " "A passport_data message update was received, "
"but this handler does nothing yet." "but this handler does nothing yet."
) )
async def dice_handler(self, update, user_record): async def dice_handler(self, update, user_record, language=None):
"""Handle `dice` message update.""" """Handle `dice` message update."""
logging.info( logging.info(
"A dice message update was received, " "A dice message update was received, "
@@ -2013,7 +2015,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
""" """
self._allowed_during_maintenance.append(criterion) self._allowed_during_maintenance.append(criterion)
async def handle_update_during_maintenance(self, update, user_record=None): async def handle_update_during_maintenance(self, update, user_record=None, language=None):
"""Handle an update while bot is under maintenance. """Handle an update while bot is under maintenance.
Handle all types of updates. Handle all types of updates.
@@ -2106,7 +2108,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
Decorate command handlers like this: Decorate command handlers like this:
``` ```
@bot.command('/my_command', ['Button'], True, "My command", 'user') @bot.command('/my_command', ['Button'], True, "My command", 'user')
async def command_handler(bot, update, user_record): async def command_handler(bot, update, user_record, language):
return "Result" return "Result"
``` ```
When a message text starts with `/command[@bot_name]`, or with an When a message text starts with `/command[@bot_name]`, or with an
@@ -2165,7 +2167,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
command = command.strip('/ ').lower() command = command.strip('/ ').lower()
def command_decorator(command_handler): def command_decorator(command_handler):
async def decorated_command_handler(bot, update, user_record): async def decorated_command_handler(bot, update, user_record, language=None):
logging.info( logging.info(
f"Command `{command}@{bot.name}` called by " f"Command `{command}@{bot.name}` called by "
f"`{update['from'] if 'from' in update else update['chat']}`" f"`{update['from'] if 'from' in update else update['chat']}`"
@@ -2223,7 +2225,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
return 'from' in update return 'from' in update
@bot.parser(custom_criteria, authorization_level='user') @bot.parser(custom_criteria, authorization_level='user')
async def text_parser(bot, update, user_record): async def text_parser(bot, update, user_record, language):
return "Result" return "Result"
``` ```
If condition evaluates True when run on a message text If condition evaluates True when run on a message text
@@ -2241,7 +2243,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
def parser_decorator(parser): def parser_decorator(parser):
async def decorated_parser(bot, update, user_record): async def decorated_parser(bot, update, user_record, language=None):
logging.info( logging.info(
f"Text message update matching condition " f"Text message update matching condition "
f"`{condition.__name__}@{bot.name}` from " f"`{condition.__name__}@{bot.name}` from "
@@ -2310,7 +2312,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
``` ```
@bot.button('a_prefix:///', description="A button", @bot.button('a_prefix:///', description="A button",
authorization_level='user') authorization_level='user')
async def button_handler(bot, update, user_record, data): async def button_handler(bot, update, user_record, language, data):
return "Result" return "Result"
``` ```
`separator` will be used to parse callback data received when a button `separator` will be used to parse callback data received when a button
@@ -2325,7 +2327,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
def button_decorator(handler): def button_decorator(handler):
async def decorated_button_handler(bot, update, user_record): async def decorated_button_handler(bot, update, user_record, language=None):
logging.info( logging.info(
f"Button `{update['data']}`@{bot.name} pressed by " f"Button `{update['data']}`@{bot.name} pressed by "
f"`{update['from']}`" f"`{update['from']}`"
@@ -2382,7 +2384,7 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
) )
def query_decorator(handler): def query_decorator(handler):
async def decorated_query_handler(bot, update, user_record): async def decorated_query_handler(bot, update, user_record, language=None):
logging.info( logging.info(
f"Inline query matching condition " f"Inline query matching condition "
f"`{condition.__name__}@{bot.name}` from " f"`{condition.__name__}@{bot.name}` from "
@@ -2881,9 +2883,12 @@ class Bot(TelegramBot, ObjectWithDatabase, MultiLanguageObject):
for key, value in update.items(): for key, value in update.items():
if key in self.routing_table: if key in self.routing_table:
user_record = self.get_user_record(update=value) user_record = self.get_user_record(update=value)
language = self.get_language(update=update,
user_record=user_record)
return await self.routing_table[key]( return await self.routing_table[key](
update=value, update=value,
user_record=user_record user_record=user_record,
language=language
) )
logging.error(f"Unknown type of update.\n{update}") logging.error(f"Unknown type of update.\n{update}")