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 @task
  21 def buy(c, item_id):
  22     result = Data().purchase_item(item_id)
  23     print(result)
  24
  25 @task
  26 def list(c, store_id):
  27     items = Data().get_shopping_list(store_id)
  28     for item in items:
  29         print(item)
  30
  31 @task
  32 def list_stores(c):
  33     items = Data().get_stores()
  34     for item in items:
  35         print(item)
  36
  37 @task
  38 def clear_list(c, store_id):
  39     Data().clear_list(store_id)
  40
  41 @task
  42 def clear_all(c, store_id):
  43     Data().clear_list(store_id, clear_all=True)
  44
  45 @task
  46 def list_unsorted(c):
  47     items = Data().get_unsorted_list()
  48     for item in items:
  49         print(item)
  50
  51 @task
  52 def assign(c, store_id, item_id):
  53     result = Data().assign_store(store_id, item_id)
  54     print(result)
  55
  56