Log

[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
 
 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())