X-Git-Url: https://us-delaporte.pagekite.me/gitweb/matrix-shop-bot.git/blobdiff_plain/dbe3765edb809f29f17baf9f638089eb40f4eda7..b988f8c87df51797f52d20ae91c14c739459b758:/matrix_shop_bot/tasks.py?ds=sidebyside diff --git a/matrix_shop_bot/tasks.py b/matrix_shop_bot/tasks.py index f4775fa..8d1125e 100644 --- a/matrix_shop_bot/tasks.py +++ b/matrix_shop_bot/tasks.py @@ -42,25 +42,23 @@ def buy(c, items): @task def list_bought(c, store): result = Data().list_purchases(store) - print(result) - + for item in result: + print(str(item)) @task -def clear_all(c, store_id): - Data().clear_list(store_id, clear_all=True) - -@task -def clear_list(c, store_id): - Data().clear_list(store_id) +def clear(c, store_name, clear_all=False): + results = Data().clear_list(store_name, clear_all) + for item in results: + print(str(item)) @task -def list_for_store(c, store_id): - items = Data().get_shopping_list_for_store(store_id) +def at(c, store_name): + '''Select store to shop at''' + items = Data().get_shopping_list(store_name) for item in items: print(item) - @task def list(c): '''Use previous store'''