X-Git-Url: https://apps.delaporte.us/gitweb/matrix-fridge-bot.git/blobdiff_plain/98abdc941c39a6dea3d3b5afb7cd3e13bcec5167..2f917aaa798317cb369c0226ff059bf83d01fbe6:/matrix_fridge_bot/tasks.py?ds=sidebyside

diff --git a/matrix_fridge_bot/tasks.py b/matrix_fridge_bot/tasks.py
index f95400b..c51eca1 100644
--- a/matrix_fridge_bot/tasks.py
+++ b/matrix_fridge_bot/tasks.py
@@ -7,43 +7,46 @@ 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.environ.get('MATRIX_DB_FILE', 'fridge.db')
 db_file = os.path.join(DATA_DIR, DB_FILE)
 
 DATA = Data(db_file)
 
 @task
 def add(c, task):
+    '''Add food to the fridge'''
     result = DATA.add(task)
     print(result)
 
-
 @task
-def check(c, task):
-    result = DATA.check(task)
+def setup(c):
+    '''One time database setup.'''
+    result = DATA.setup()
     print(result)
 
-@task
-def done(c, task):
-    result = DATA.done(task)
-    print(result)
 
-@task
-def undo(c, task):
-    result = DATA.undo(task)
-    print(result)
-
-@task
-def remove(c, task):
-    result = DATA.remove(task)
-    print(result)
 
 @task
 def list(c):
+    '''List good things to eat in the fridge'''
     result = DATA.list()
     for item in result:
         print(f'{item}')
 
+
+@task
+def toss(c):
+    result = DATA.to_toss()
+    for item in result:
+        print(f'{item}')
+
+
+@task
+def remove(c, item):
+    result = DATA.remove(item)
+    print(result)
+
+
 @task
 def filter(c, term):
     result = DATA.filter(term)