count intimacy, untested

1 files changed, 12 insertions, 1 deletions

diff --git a/catdb.py b/catdb.py
index 14843f8..e3bf921 100644
--- a/catdb.py
+++ b/catdb.py
@@ -29,7 +29,7 @@ class DBHandler():
     def query(self,str):
         conn = sqlite3.connect(self.DB)
         cur = conn.cursor()
-        res = cur.execute(str)
+        res = cur.execute(str).fetchall()
         conn.close()
         return res
 
@@ -42,3 +42,14 @@ class DBHandler():
         except Exception as e:
             print("ERROR ","failed to insert event to the db, ignoring ",e)
 
+    def count_interaction(self):
+        try:
+            res = self.query(QUERY_COUNT_INTERACTION)
+            return dict(res)
+        except:
+            print("query failed")
+        return None
+
+
+
+