Add zones for shopping

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

diff --git a/matrix_shop_bot/bot.py b/matrix_shop_bot/bot.py
index 0b52d76d5d6f72b7d04d3c54d9aa2234d82dbdbd..9a36c631903c0efe7fd32d10427cbf35c624b180 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -30,21 +30,37 @@ async def do_shop(match):
       extra.help - show extended help
       add  - add an item to shop for
       add.store - add a store
+      at STORE_ID - use this store for future commands
+      buy ID ID ID - mark items as purchased
       clear STORE - clear all purchased items for STORE
       list - re-show list from last used store
+      past STORE - re-show past purchased items from STORE 
       stores - list all stores
-      at STORE_ID - use this store for future commands
-      buy - mark item as purchased
+
+    Examples:
+      .shop past costco
+      .shop at costco
+
+      .shop add sour cream
+      .shop add milk
+    
+      .shop list
+
+      .shop buy 1 2 5 7
+
     '''
     extended_help = '''
       assign ITEM_ID - assign item to last used store id
       assign.item.to.store ITEM_ID STORE_ID - plan to shop for at store for item
-      clear.all STORE - clear **all** items for STORE
+      clear.all STORE - clear **all** items for STORE, including un-purchased
       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
       unsorted - list all items not assigned to any store
+      zone ZONE ITEM - list ITEM under given ZONE 
 
+      Example: 
+        .shop zone dairy sour cream
     '''
 
     if match.command('setup'):
@@ -77,6 +93,12 @@ async def do_shop(match):
         Data().assign_store(item_id, store_id)
         return ['Success']
 
+    if match.command('zone'):
+        zone = match.args()[1]
+        item = ' '.join(match.args()[2:])
+        Data().assign_zone(item, zone)
+        return ['Success']
+
     if match.command('assign'):
         item_id  = int(match.args()[1])
         Data().assign_last_store(item_id)
@@ -125,7 +147,7 @@ async def do_shop(match):
     if match.command('past'):
         store_name = ' '.join(match.args()[1:])
         items = Data().list_purchases(store_name)
-        results = ['Past purchases at {store_name}']
+        results = ['Past purchases...']
         for item in items:
             results.append(str(item))
         return results