matrix_shop_bot/tasks.py

   1 from invoke import task
   2 import logging
   3
   4 from .db import Data
   5
   6 logging.getLogger('invoke').setLevel(logging.WARNING)
   7 logging.basicConfig(filename=f'matrix-shop-bot.log', encoding='utf-8', level=logging.DEBUG)
   8
   9 @task
  10 def setup(c):
  11     result = Data().setup()
  12     print(result)
  13
  14 @task
  15 def add_item(c, name):
  16     result = Data().add_item(name)
  17     print(result)
  18
  19 @task
  20 def add_store(c, name):
  21     result = Data().add_store(name)
  22     print(result)
  23
  24
  25 @task
  26 def assign(c, item_id):
  27     result = Data().assign_last_store(item_id)
  28     print(result)
  29
  30
  31 @task
  32 def assign_to_store(c, item_id, store_id):
  33     result = Data().assign_store(store_id, item_id)
  34     print(result)
  35
  36 @task
  37 def assign_to_zone(c, item, zone):
  38     result = Data().assign_zone(item, zone)
  39     print(result)
  40
  41 @task
  42 def buy(c, items):
  43     result = Data().purchase_items(items)
  44     print(result)
  45
  46 @task
  47 def list_bought(c, store):
  48     result = Data().list_purchases(store)
  49     for item in result:
  50         print(str(item))
  51
  52 @task
  53 def clear(c, store_name, clear_all=False):
  54     results = Data().clear_list(store_name, clear_all)
  55     for item in results:
  56         print(str(item))
  57
  58
  59 @task
  60 def at(c, store_name):
  61     '''Select store to shop at'''
  62     items = Data().get_shopping_list(store_name)
  63     for item in items:
  64         print(item)
  65
  66 @task
  67 def list(c):
  68     '''Use previous store'''
  69     items = Data().get_last_shopping_list()
  70     for item in items:
  71         print(item)
  72
  73 @task
  74 def list_stores(c):
  75     items = Data().get_stores()
  76     for item in items:
  77         print(item)
  78
  79
  80 @task
  81 def list_unsorted(c):
  82     items = Data().get_unsorted_list()
  83     for item in items:
  84         print(item)
  85
  86
  87 @task
  88 def remove_item(c, item_id):
  89     Data().remove_item(item_id)
  90
  91 @task
  92 def remove_store(c, store_id):
  93     Data().remove_store(store_id)
  94
  95 @task
  96 def list_purchases(c, store):
  97     items = Data().list_purchases(store)
  98     for item in items:
  99         print(item)
 100