X-Git-Url: https://apps.delaporte.us/gitweb/matrix-shop-bot.git/blobdiff_plain/8b639943d820c97787fed92731f0ec8d13e9ac7e..refs/heads/main:/matrix_shop_bot/tasks.py diff --git a/matrix_shop_bot/tasks.py b/matrix_shop_bot/tasks.py index 6b999bf..580eeba 100644 --- a/matrix_shop_bot/tasks.py +++ b/matrix_shop_bot/tasks.py @@ -33,6 +33,10 @@ def assign_to_store(c, item_id, store_id): result = Data().assign_store(store_id, item_id) print(result) +@task +def item_to_zone(c, item, zone): + result = Data().assign_zone(item, zone) + print(result) @task def buy(c, items): @@ -46,12 +50,10 @@ def list_bought(c, store): 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 @@ -82,6 +84,14 @@ def list_unsorted(c): print(item) +@task +def rename_item(c, old, new): + Data().rename_item(old, new) + +@task +def rename_store(c, old, new): + Data().rename_store(old, new) + @task def remove_item(c, item_id): Data().remove_item(item_id) @@ -90,3 +100,9 @@ def remove_item(c, item_id): def remove_store(c, store_id): Data().remove_store(store_id) +@task +def list_purchases(c, store): + items = Data().list_purchases(store) + for item in items: + print(item) +