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, item_id):
  23     result = Data().assign_last_store(item_id)
  24     print(result)
  25
  26
  27 @task
  28 def assign_to_store(c, item_id, store_id):
  29     result = Data().assign_store(store_id, item_id)
  30     print(result)
  31
  32
  33 @task
  34 def buy(c, item_id):
  35     result = Data().purchase_item(item_id)
  36     print(result)
  37
  38 @task
  39 def clear_all(c, store_id):
  40     Data().clear_list(store_id, clear_all=True)
  41
  42 @task
  43 def clear_list(c, store_id):
  44     Data().clear_list(store_id)
  45
  46
  47 @task
  48 def list_for_store(c, store_id):
  49     items = Data().get_shopping_list_for_store(store_id)
  50     for item in items:
  51         print(item)
  52
  53
  54 @task
  55 def list(c):
  56     '''Use previous store'''
  57     items = Data().get_last_shopping_list()
  58     for item in items:
  59         print(item)
  60
  61 @task
  62 def list_stores(c):
  63     items = Data().get_stores()
  64     for item in items:
  65         print(item)
  66
  67
  68 @task
  69 def list_unsorted(c):
  70     items = Data().get_unsorted_list()
  71     for item in items:
  72         print(item)
  73
  74
  75 @task
  76 def remove_item(c, item_id):
  77     Data().remove_item(item_id)
  78
  79 @task
  80 def remove_store(c, store_id):
  81     Data().remove_store(store_id)
  82