Update to print list after buying something

[matrix-shop-bot.git] / matrix_shop_bot / bot.py

diff --git a/matrix_shop_bot/bot.py b/matrix_shop_bot/bot.py
index 7c8d9a26878722360d846a271f0fca7228d446ad..387cf184af205789ef5e4a919701166600df73ed 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -30,12 +30,12 @@ async def do_shop(match):
       extra.help - show extended help
       add  - add an item to shop for
       add.store - add a store
+      stores - list all stores
+      at STORE_ID - (!shop at STORE_ID) list all items assigned to store and use this store for future commands
       assign ITEM_ID - assign item to last used store id
       buy - mark item as purchased
       clear STORE_ID - clear all purchased items for STORE_ID from database
       list - re-show list from last used store
-      stores - list all stores
-      shop.at STORE_ID - list all items assigned to store and use this store for future commands
     '''
     extended_help = '''
       assign.item.to.store ITEM_ID STORE_ID - plan to shop for at store for item
@@ -91,7 +91,10 @@ async def do_shop(match):
     if match.command('buy'):
         item_id = ' '.join(match.args()[1:])
         Data().purchase_item(item_id)
-        return ['Success']
+        results = []
+        for item in Data().get_last_shopping_list():
+            results.append(str(item))
+        return results
 
     if match.command('clear'):
         store_id = ' '.join(match.args()[1:])
@@ -105,7 +108,7 @@ async def do_shop(match):
         return ['Success']
 
 
-    if match.command('shop.at'):
+    if match.command('at'):
         store_id = ' '.join(match.args()[1])
         items = Data().get_shopping_list_for_store(store_id)
         results = ['Shopping List']