X-Git-Url: https://apps.delaporte.us/gitweb/matrix-shop-bot.git/blobdiff_plain/b634b095ba15a57b290abbf3577b175d0d281aff..4503755f5b9624a373db24ab8072532c2a97590a:/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'''