From f5f81f5a93acddbc2362c228c0ebc75cbe0e2010 Mon Sep 17 00:00:00 2001 From: Ceda EI Date: Mon, 11 Mar 2019 18:17:01 +0530 Subject: [PATCH] Add delete_token() --- questable.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/questable.py b/questable.py index 40791b8..c470948 100644 --- a/questable.py +++ b/questable.py @@ -169,4 +169,11 @@ class player(): token = str(uuid.uuid4()) query = ('INSERT INTO tokens(chat_id, token) values(?, ?)') cursor.execute(query, (self.CHAT_ID, token)) + self.DB.commit() return token + + def delete_token(self, token): + cursor = self.DB.cursor() + query = ('DELETE FROM tokens WHERE chat_id = ? AND token = ?') + cursor.execute(query, (self.CHAT_ID, token)) + self.DB.commit()