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_str=None):
  43
  44     store_id = int(store_id_str) if store_id_str else None
  45     items = Data().get_shopping_list(store_id)
  46     for item in items:
  47         print(item)
  48
  49 @task
  50 def list_stores(c):
  51     items = Data().get_stores()
  52     for item in items:
  53         print(item)
  54
  55
  56 @task
  57 def list_unsorted(c):
  58     items = Data().get_unsorted_list()
  59     for item in items:
  60         print(item)
  61
  62
  63 @task
  64 def remove_item(c, item_id):
  65     Data().remove_item(item_id)
  66
  67 @task
  68 def remove_store(c, store_id):
  69     Data().remove_store(store_id)
  70