From 2a5ae51d71bd65c4eb165d9e80881094b6c84ef4 Mon Sep 17 00:00:00 2001 From: Ceda EI Date: Wed, 13 Mar 2019 19:49:15 +0530 Subject: [PATCH] Add get_quest, get_side_quest to player. --- questable.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/questable.py b/questable.py index 28e3565..d75c2ef 100644 --- a/questable.py +++ b/questable.py @@ -157,6 +157,28 @@ class player(): quests.append(q) return quests + def get_quest(self, qid): + cursor = self.DB.cursor() + query = ('SELECT chat_id, qid, name, importance, difficulty, ' + 'state FROM quests WHERE chat_id = ? AND qid = ?') + cursor.execute(query, (self.CHAT_ID, qid)) + row = cursor.fetchone() + if row is None: + return False + else: + return quest(self.DB, *row) + + def get_side_quest(self, qid): + cursor = self.DB.cursor() + query = ('SELECT chat_id, qid, name, importance, difficulty, ' + 'state FROM quests WHERE chat_id = ? AND qid = ?') + cursor.execute(query, (self.CHAT_ID, qid)) + row = cursor.fetchone() + if row is None: + return False + else: + return side_quest(self.DB, *row) + def get_tokens(self): cursor = self.DB.cursor() query = ('SELECT token FROM tokens WHERE chat_id=?')