From 879423f7c79a6d61ef93f181d7a62e9133cc9d34 Mon Sep 17 00:00:00 2001 From: H4CK3R-01 Date: Mon, 28 Mar 2022 17:51:10 +0200 Subject: [PATCH] Use user id instead of email --- api/helper_functions.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/api/helper_functions.py b/api/helper_functions.py index 7f25059..4ebf10e 100644 --- a/api/helper_functions.py +++ b/api/helper_functions.py @@ -41,12 +41,17 @@ def get_email_from_token_data(): if token is not None: if ':' in token: # Maybe bot token, check if token valid and return username after ":" then - email = token.split(":")[1] + telegram_user_id = token.split(":")[1] token = token.split(":")[0] try: if jwt.decode(token, os.getenv('SECRET_KEY'), algorithms=["HS256"])['email'] == os.getenv("BOT_EMAIL"): - return email + res = db.session.query(User).filter_by(telegram_user_id=telegram_user_id).first() + + if res is not None: + return res.as_dict()['email'] + else: + return None else: return None except jwt.PyJWTError: