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 4e766036820c13ffe6eb7cbf1f78e381e9793b66..46ba972de0b86c0fbaefdd3c9d84b3b0b65d7374 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -97,19 +97,17 @@ async def do_shop(match):
         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('at'):
-        store_name = ' '.join(match.args()[1])
+        store_name = ' '.join(match.args()[1:])
         items = Data().get_shopping_list(store_name)
         results = ['Shopping List']
         for item in items: