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 387cf184af205789ef5e4a919701166600df73ed..46ba972de0b86c0fbaefdd3c9d84b3b0b65d7374 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -89,28 +89,26 @@ async def do_shop(match):
         return results
 
     if match.command('buy'):
         return results
 
     if match.command('buy'):

-        item_id = ' '.join(match.args()[1:])
-        Data().purchase_item(item_id)
-        results = []
-        for item in Data().get_last_shopping_list():
-            results.append(str(item))
+        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'):
         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'):
 
     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('at'):
 
     if match.command('at'):

-        store_id = ' '.join(match.args()[1])
-        items = Data().get_shopping_list_for_store(store_id)
+        store_name = ' '.join(match.args()[1:])
+        items = Data().get_shopping_list(store_name)

         results = ['Shopping List']
         for item in items:
             results.append(str(item))
         results = ['Shopping List']
         for item in items:
             results.append(str(item))