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     print(result)
  46
  47
  48 @task
  49 def clear_all(c, store_id):
  50     Data().clear_list(store_id, clear_all=True)
  51
  52 @task
  53 def clear_list(c, store_id):
  54     Data().clear_list(store_id)
  55
  56
  57 @task
  58 def list_for_store(c, store_id):
  59     items = Data().get_shopping_list_for_store(store_id)
  60     for item in items:
  61         print(item)
  62
  63
  64 @task
  65 def list(c):
  66     '''Use previous store'''
  67     items = Data().get_last_shopping_list()
  68     for item in items:
  69         print(item)
  70
  71 @task
  72 def list_stores(c):
  73     items = Data().get_stores()
  74     for item in items:
  75         print(item)
  76
  77
  78 @task
  79 def list_unsorted(c):
  80     items = Data().get_unsorted_list()
  81     for item in items:
  82         print(item)
  83
  84
  85 @task
  86 def remove_item(c, item_id):
  87     Data().remove_item(item_id)
  88
  89 @task
  90 def remove_store(c, store_id):
  91     Data().remove_store(store_id)
  92