Fix for clear command

[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..46ba972de0b86c0fbaefdd3c9d84b3b0b65d7374 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
@@ -89,25 +89,26 @@ async def do_shop(match):
         return results
 
     if match.command('buy'):
-        item_id = ' '.join(match.args()[1:])
-        Data().purchase_item(item_id)
-        return ['Success']
+        items = ' '.join(match.args()[1:])
+        results = Data().purchase_items(items)
+        # 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:])
-        Data().clear_list(store_id)
-        self.last_store_id = None
-        return ['Success']
+        store_name = ' '.join(match.args()[1:])
+        results = Data().clear_list(store_name)
+        return results
 
     if match.command('clear.all'):
-        store_id = ' '.join(match.args()[1:])
-        Data().clear_list(store_id, clear_all=True)
-        return ['Success']
-
+        store_name = ' '.join(match.args()[1:])
+        results = Data().clear_list(store_name, clear_all=True)
+        return results
 
-    if match.command('shop.at'):
-        store_id = ' '.join(match.args()[1])
-        items = Data().get_shopping_list_for_store(store_id)
+    if match.command('at'):
+        store_name = ' '.join(match.args()[1:])
+        items = Data().get_shopping_list(store_name)
         results = ['Shopping List']
         for item in items:
             results.append(str(item))