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
  37 @task
  38 def buy(c, items):
  39     result = Data().purchase_items(items)
  40     print(result)
  41
  42 @task
  43 def list_bought(c, store):
  44     result = Data().list_purchases(store)
  45     for item in result:
  46         print(str(item))
  47
  48 @task
  49 def clear(c, store_name, clear_all=False):
  50     results = Data().clear_list(store_name, clear_all)
  51     for item in results:
  52         print(str(item))
  53
  54
  55 @task
  56 def at(c, store_name):
  57     '''Select store to shop at'''
  58     items = Data().get_shopping_list(store_name)
  59     for item in items:
  60         print(item)
  61
  62 @task
  63 def list(c):
  64     '''Use previous store'''
  65     items = Data().get_last_shopping_list()
  66     for item in items:
  67         print(item)
  68
  69 @task
  70 def list_stores(c):
  71     items = Data().get_stores()
  72     for item in items:
  73         print(item)
  74
  75
  76 @task
  77 def list_unsorted(c):
  78     items = Data().get_unsorted_list()
  79     for item in items:
  80         print(item)
  81
  82
  83 @task
  84 def remove_item(c, item_id):
  85     Data().remove_item(item_id)
  86
  87 @task
  88 def remove_store(c, store_id):
  89     Data().remove_store(store_id)
  90