matrix_shop_bot/tasks.py

   1 from invoke import task
   2
   3 from .db import Data
   4
   5 @task
   6 def setup(c):
   7     result = Data().setup()
   8     print(result)
   9
  10 @task
  11 def add_item(c, name):
  12     result = Data().add_item(name)
  13     print(result)
  14
  15 @task
  16 def add_store(c, name):
  17     result = Data().add_store(name)
  18     print(result)
  19
  20
  21 @task
  22 def list(c, store_id):
  23     items = Data().get_shopping_list(store_id)
  24     for item in items:
  25         print(item)
  26
  27
  28 @task
  29 def list_unsorted(c, store_id):
  30     items = Data().list_unsorted()
  31     for item in items:
  32         print(item)
  33
  34 @task
  35 def assign(c, store_id, item_id):
  36     result = Data().assign_store(store_id, item_id)
  37     print(result)
  38