From 6d67f050939d23e1bfbcb99606fbab34986e8905 Mon Sep 17 00:00:00 2001 From: Ceda EI Date: Wed, 13 Mar 2019 17:50:53 +0530 Subject: [PATCH] Add get_player_from_token --- questable.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/questable.py b/questable.py index c470948..28e3565 100644 --- a/questable.py +++ b/questable.py @@ -177,3 +177,14 @@ class player(): query = ('DELETE FROM tokens WHERE chat_id = ? AND token = ?') cursor.execute(query, (self.CHAT_ID, token)) self.DB.commit() + + +def get_player_from_token(db, token): + cursor = db.cursor() + query = "SELECT chat_id FROM tokens WHERE token=?" + cursor.execute(query, (token,)) + chat_id = cursor.fetchone() + if chat_id is None: + return False + else: + return player(db, chat_id[0])