matrix_fridge_bot/tasks.py

   1 import os
   2
   3 from invoke import task
   4
   5 from .db import Data
   6
   7
   8 # Where to store data
   9 DATA_DIR = os.environ.get('MATRIX_DATA_DIR', '')
  10 DB_FILE = os.environ.get('MATRIX_DB_FILE', 'fridge.db')
  11 db_file = os.path.join(DATA_DIR, DB_FILE)
  12
  13 DATA = Data(db_file)
  14
  15 @task
  16 def add(c, task):
  17     '''Add food to the fridge'''
  18     result = DATA.add(task)
  19     print(result)
  20
  21
  22 @task
  23 def list(c):
  24     '''List good things to eat in the fridge'''
  25     result = DATA.list()
  26     for item in result:
  27         print(f'{item}')
  28
  29
  30 @task
  31 def to_toss(c):
  32     result = DATA.to_toss()
  33     for item in result:
  34         print(f'{item}')
  35
  36
  37 @task
  38 def remove(c, item):
  39     result = DATA.remove(item)
  40     print(result)
  41
  42
  43 @task
  44 def filter(c, term):
  45     result = DATA.filter(term)
  46     for item in result:
  47         print(f'{item}')
  48
  49 @task
  50 def debug(c):
  51     print(DATA.get_db_file())