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

Provide a chat_id getter and a method to change it.

Browse Source
This commit is contained in:
Davte 2019-07-01 20:20:12 +02:00
parent 55d872fc21
commit 783040bcd0
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
davtelepot/bot.py
View File

@ -112,6 +112,14 @@ class Bot(TelegramBot):
self._under_maintenance = False self._under_maintenance = False
self._allowed_during_maintenance = [] self._allowed_during_maintenance = []
self._maintenance_message = None self._maintenance_message = None
# Default chat_id getter: same chat as update
self.get_chat_id = lambda update: (
update['message']['chat']['id']
if 'message' in update and 'chat' in update['message']
else update['chat']['id']
if 'chat' in update
else None
)
# Message to be returned if user is not allowed to call method # Message to be returned if user is not allowed to call method
self._authorization_denied_message = None self._authorization_denied_message = None
# Default authorization function (always return True) # Default authorization function (always return True)
@ -664,6 +672,15 @@ class Bot(TelegramBot):
""" """
self.authorization_function = authorization_function self.authorization_function = authorization_function
def set_chat_id_getter(self, getter):
"""Set chat_id getter.
It must be a function that takes an update and returns the proper
chat_id.
"""
assert callable(getter), "Chat id getter must be a function!"
self.get_chat_id = getter
@staticmethod @staticmethod
def get_identifier_from_update_or_user_id(user_id=None, update=None): def get_identifier_from_update_or_user_id(user_id=None, update=None):
"""Get telegram id of user given an update. """Get telegram id of user given an update.