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