X-Git-Url: https://apps.delaporte.us/gitweb/matrix-chore-bot.git/blobdiff_plain/99f1517e1c2cd574b1969a79d5631769f789debb..23985ed419254cf2f06730a172c6a9df6ef97155:/matrix_chore_bot/tasks.py

diff --git a/matrix_chore_bot/tasks.py b/matrix_chore_bot/tasks.py
index 5a4eaa6..f95400b 100644
--- 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())