From a3f50b53341d432ded04b9dc35cf032f50fc3f61 Mon Sep 17 00:00:00 2001
From: Ceda EI <ceda_ei@protonmail.com>
Date: Sat, 27 Oct 2018 16:59:01 +0530
Subject: [PATCH] Add get_from_db to base_quest. Add other functions.

Add get_quest, add_quest, get_side_quest, add_side_quest for easier
way to add / retrieve quests and side quests.
---
 questable.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/questable.py b/questable.py
index f81297c..7290eb7 100644
--- a/questable.py
+++ b/questable.py
@@ -37,6 +37,14 @@ class base_quest():
                                self.state, self.CHAT_ID, self.QID))
         self.DB.commit()
 
+    def get_from_db(self):
+        cursor = self.DB.cursor()
+        query = (f'SELECT name, importance, difficulty, date, state FROM '
+                 '{self.TABLE} where chat_id=? AND qid=?')
+        cursor.execute(query, self.CHAT_ID, self.QID)
+        output = cursor.fetchone()
+        self.name, self.imp, self.diff, self.date, self.state = output
+
 
 class quest(base_quest):
     TABLE = "quests"
@@ -44,3 +52,29 @@ class quest(base_quest):
 
 class side_quest(base_quest):
     TABLE = "side_quests"
+
+
+def get_quest(db, chat_id, qid):
+    q = quest(db, chat_id, qid)
+    q.get_from_db()
+    return q
+
+
+def add_quest(self, db, chat_id, qid, name=None, imp=None, diff=None,
+              state=None, date=None):
+    q = quest(self, db, chat_id, qid, name, imp, diff, state, date)
+    q.add_to_db()
+    return q
+
+
+def get_side_quest(db, chat_id, qid):
+    q = side_quest(db, chat_id, qid)
+    q.get_from_db()
+    return q
+
+
+def add_side_quest(self, db, chat_id, qid, name=None, imp=None, diff=None,
+                   state=None, date=None):
+    q = side_quest(self, db, chat_id, qid, name, imp, diff, state, date)
+    q.add_to_db()
+    return q