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

Fix authorization function so that it extracts user_record from update

Browse Source
This commit is contained in:
Davte 2020-04-20 18:20:44 +02:00
parent 91ab00eec6
commit 41bc5b68a4
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
davtelepot/__init__.py
View File

@ -14,7 +14,7 @@ __author__ = "Davide Testa"
__email__ = "davide@davte.it" __email__ = "davide@davte.it"
__credits__ = ["Marco Origlia", "Nick Lee @Nickoala"] __credits__ = ["Marco Origlia", "Nick Lee @Nickoala"]
__license__ = "GNU General Public License v3.0" __license__ = "GNU General Public License v3.0"
__version__ = "2.4.19" __version__ = "2.4.20"
__maintainer__ = "Davide Testa" __maintainer__ = "Davide Testa"
__contact__ = "t.me/davte" __contact__ = "t.me/davte"

10
davtelepot/authorization.py
View File

@ -235,6 +235,16 @@ def get_authorization_function(bot):
def is_authorized(update, user_record=None, authorization_level=2): def is_authorized(update, user_record=None, authorization_level=2):
"""Return True if user role is at least at `authorization_level`.""" """Return True if user role is at least at `authorization_level`."""
if user_record is None:
if (
isinstance(update, dict)
and 'from' in update
and isinstance(update['from'], dict)
and 'id' in update['from']
):
user_record = bot.db['users'].find_one(
telegram_id=update['from']['id']
)
user_role = bot.Role.get_user_role(user_record=user_record) user_role = bot.Role.get_user_role(user_record=user_record)
if user_role.code == 0: if user_role.code == 0:
return False return False