Add invoke debug task to check db

diff --git a/matrix_chore_bot/db.py b/matrix_chore_bot/db.py
index 1a07bec5af581eae1c1f12d5be2420265443ac5c..a0eae22f210f4374073a8ae41fa8c43a3053b57f 100644 (file)
--- a/matrix_chore_bot/db.py
+++ b/matrix_chore_bot/db.py
@@ -35,14 +35,14 @@ def get_chore(session, chore_id):
     return session.exec(cmd).first()
 
 class Data():
     return session.exec(cmd).first()
 
 class Data():

-    def __init__(self, db_file=None):
-
-        if not db_file:
-            db_file = DB_FILE
+    def __init__(self, db_file):

 
         self.file = db_file
         self.engine = self.get_engine()
 
 
         self.file = db_file
         self.engine = self.get_engine()
 

+    def get_db_file(self):
+        return self.file
+

     def get_engine(self):
         return create_engine(f"sqlite:///{self.file}")
 
     def get_engine(self):
         return create_engine(f"sqlite:///{self.file}")
 

diff --git a/matrix_chore_bot/tasks.py b/matrix_chore_bot/tasks.py
index 5a4eaa6dadfa183cb28b81872912007f8266ac1e..f95400b6ef15ce35e39c54553c3c3ec95c6cb9cd 100644 (file)
--- a/matrix_chore_bot/tasks.py
+++ b/matrix_chore_bot/tasks.py
@@ -1,42 +1,55 @@
+import os
+

 from invoke import task
 
 from .db import Data
 
 from invoke import task
 
 from .db import Data
 

+
+# Where to store data
+DATA_DIR = os.environ.get('MATRIX_DATA_DIR', '')
+DB_FILE = os.environ.get('MATRIX_DB_FILE', 'chore.db')
+db_file = os.path.join(DATA_DIR, DB_FILE)
+
+DATA = Data(db_file)
+

 @task
 def add(c, task):
 @task
 def add(c, task):

-    result = Data().add(task)
+    result = DATA.add(task)

     print(result)
 
 
 @task
 def check(c, task):
     print(result)
 
 
 @task
 def check(c, task):

-    result = Data().check(task)
+    result = DATA.check(task)

     print(result)
 
 @task
 def done(c, task):
     print(result)
 
 @task
 def done(c, task):

-    result = Data().done(task)
+    result = DATA.done(task)

     print(result)
 
 @task
 def undo(c, task):
     print(result)
 
 @task
 def undo(c, task):

-    result = Data().undo(task)
+    result = DATA.undo(task)

     print(result)
 
 @task
 def remove(c, task):
     print(result)
 
 @task
 def remove(c, task):

-    result = Data().remove(task)
+    result = DATA.remove(task)

     print(result)
 
 @task
 def list(c):
     print(result)
 
 @task
 def list(c):

-    result = Data().list()
+    result = DATA.list()

     for item in result:
         print(f'{item}')
 
 @task
 def filter(c, term):
     for item in result:
         print(f'{item}')
 
 @task
 def filter(c, term):

-    result = Data().filter(term)
+    result = DATA.filter(term)

     for item in result:
         print(f'{item}')
 
     for item in result:
         print(f'{item}')
 

+@task
+def debug(c):
+    print(DATA.get_db_file())