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_for_store(c, store_id):
  43     items = Data().get_shopping_list_for_store(store_id)
  44     for item in items:
  45         print(item)
  46
  47
  48 @task
  49 def list(c):
  50     '''Use previous store'''
  51     items = Data().get_last_shopping_list()
  52     for item in items:
  53         print(item)
  54
  55 @task
  56 def list_stores(c):
  57     items = Data().get_stores()
  58     for item in items:
  59         print(item)
  60
  61
  62 @task
  63 def list_unsorted(c):
  64     items = Data().get_unsorted_list()
  65     for item in items:
  66         print(item)
  67
  68
  69 @task
  70 def remove_item(c, item_id):
  71     Data().remove_item(item_id)
  72
  73 @task
  74 def remove_store(c, store_id):
  75     Data().remove_store(store_id)
  76