Add invoke debug task to check db

[matrix-chore-bot.git] / matrix_chore_bot / tasks.py

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
 
+
+# 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):
-    result = Data().add(task)
+    result = DATA.add(task)
     print(result)
 
 
 @task
 def check(c, task):
-    result = Data().check(task)
+    result = DATA.check(task)
     print(result)
 
 @task
 def done(c, task):
-    result = Data().done(task)
+    result = DATA.done(task)
     print(result)
 
 @task
 def undo(c, task):
-    result = Data().undo(task)
+    result = DATA.undo(task)
     print(result)
 
 @task
 def remove(c, task):
-    result = Data().remove(task)
+    result = DATA.remove(task)
     print(result)
 
 @task
 def list(c):
-    result = Data().list()
+    result = DATA.list()
     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}')
 
+@task
+def debug(c):
+    print(DATA.get_db_file())