Fixup

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

diff --git a/matrix_shop_bot/bot.py b/matrix_shop_bot/bot.py
index 8a25f537632431c5e0fd22c475f5bd2f323716df..5c8e6c5eae83f7ffbbc319bf217c2373244e8e14 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -4,7 +4,7 @@ import os
 
 import simplematrixbotlib as botlib
 
-from .db import Data, pretty_print
+from .db import Data
 
 MODULE='shop'
 
@@ -30,8 +30,14 @@ async def do_shop(match):
       add  - add an item to shop for
       add-store - add a store
       assign STORE_ID ITEM_ID - plan to shop for at store for item
+      buy - mark item as purchased
+      clear STORE_ID - clear all purchased items for STORE_ID from database
+      clear-all STORE_ID - clear **all** items for STORE_ID from database
       list STORE_ID - list all items assigned to store
+      list-stores - list all stores
       list-unsorted - list all items not assigned to any store
+      remove ITEM_ID - remove a shopping item from the datatbase
+      remove-store STORE_ID - remove a store from the datatbase (requires clearing the list first)
       setup - run one time database setup
     '''
 
@@ -39,6 +45,13 @@ async def do_shop(match):
         result = Data().setup()
         return [result]
 
+    if match.command('list-stores'):
+        items = Data().get_stores()
+        results = ['List of Stores']
+        for item in items:
+            results.append(str(item))
+        return results
+
     if match.command('add'):
         new = ' '.join(match.args()[1:])
         result = Data().add_item(new)
@@ -49,24 +62,57 @@ async def do_shop(match):
         result = Data().add_store(new)
         return [f'Store added with id {result}']
 
+    if match.command('assign'):
+        store_id = int(match.args()[1])
+        item_id  = int(match.args()[2])
+        Data().assign_store(store_id, item_id)
+
+        # Now print the list for the assigned store
+        items = Data().get_shopping_list(store_id)
+        results = ['Shopping List']
+        for item in items:
+            results.append(str(item))
+        return results
+
+    if match.command('buy'):
+        item_id = ' '.join(match.args()[1:])
+        Data().purchase_item(item_id)
+        return ['Success']
+
+    if match.command('clear'):
+        store_id = ' '.join(match.args()[1:])
+        Data().clear_list(store_id)
+        return ['Success']
+
+    if match.command('clear-all'):
+        store_id = ' '.join(match.args()[1:])
+        Data().clear_list(store_id, clear_all=True)
+        return ['Success']
 
     if match.command('list'):
+        store_id = ' '.join(match.args()[1:])
         items = Data().get_shopping_list(store_id)
+        results = ['Shopping List']
         for item in items:
             results.append(str(item))
         return results
 
+
     if match.command('unsorted'):
         items = Data().list_unsorted()
         for item in items:
             results.append(str(item))
         return results
 
-    if match.command('assign'):
-        store_id = int(match.args()[1])
-        item_id  = int(match.args()[2])
-        result = Data().assign_store(store_id, item_id)
-        return result
+    if match.command('remove-item'):
+        item_id = ' '.join(match.args()[1:])
+        Data().remove_item(item_id)
+        return ['Success']
+
+    if match.command('remove-store'):
+        store_id = ' '.join(match.args()[1:])
+        Data().remove_store(store_id)
+        return ['Success']
 
     return [do_shop.__doc__]