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 7c8d9a26878722360d846a271f0fca7228d446ad..9a36c631903c0efe7fd32d10427cbf35c624b180 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -13,38 +13,54 @@ DATE_FORMAT = '%Y-%m-%d'
 def today():
     return dt.now().strftime(DATE_FORMAT)
 
 def today():
     return dt.now().strftime(DATE_FORMAT)
 

-PREFIX = f'!{MODULE}'
+PREFIX = f'.{MODULE}'

 logger = logging.getLogger(__name__)
 logging.basicConfig(filename=f'{MODULE}.log', encoding='utf-8', level=logging.DEBUG)
 logging.getLogger('nio').setLevel(logging.WARNING)
 
 async def do_shop(match):
     '''
 logger = logging.getLogger(__name__)
 logging.basicConfig(filename=f'{MODULE}.log', encoding='utf-8', level=logging.DEBUG)
 logging.getLogger('nio').setLevel(logging.WARNING)
 
 async def do_shop(match):
     '''

-    Prefix: !shop
+    Prefix: .shop

 
     Synax:
 
     Synax:

-      !shop COMMAND ARGS
+      .shop COMMAND ARGS

 
     Commands:
       help - show this help
       extra.help - show extended help
       add  - add an item to shop for
       add.store - add a store
 
     Commands:
       help - show this help
       extra.help - show extended help
       add  - add an item to shop for
       add.store - add a store

-      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
+      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
       list - re-show list from last used store

+      past STORE - re-show past purchased items from STORE 

       stores - list all stores
       stores - list all stores

-      shop.at STORE_ID - list all items assigned to store and use this store for future commands
+
+    Examples:
+      .shop past costco
+      .shop at costco
+
+      .shop add sour cream
+      .shop add milk
+    
+      .shop list
+
+      .shop buy 1 2 5 7
+

     '''
     extended_help = '''
     '''
     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
       assign.item.to.store ITEM_ID STORE_ID - plan to shop for at store for item

-      clear.all STORE_ID - clear **all** items for STORE_ID from database
-      remove.item ITEM_ID - remove a shopping item from the datatbase
+      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
       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'):
     '''
 
     if match.command('setup'):


@@ -77,6 +93,12 @@ async def do_shop(match):
         Data().assign_store(item_id, store_id)
         return ['Success']
 
         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)
     if match.command('assign'):
         item_id  = int(match.args()[1])
         Data().assign_last_store(item_id)


@@ -89,25 +111,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)
-        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'):
 
     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('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))
         results = ['Shopping List']
         for item in items:
             results.append(str(item))


@@ -120,6 +143,16 @@ async def do_shop(match):
             results.append(str(item))
         return results
 
             results.append(str(item))
         return results
 

+
+    if match.command('past'):
+        store_name = ' '.join(match.args()[1:])
+        items = Data().list_purchases(store_name)
+        results = ['Past purchases...']
+        for item in items:
+            results.append(str(item))
+        return results
+    
+

     if match.command('unsorted'):
         items = Data().get_unsorted_list()
         results = ['Items assigned to no store']
     if match.command('unsorted'):
         items = Data().get_unsorted_list()
         results = ['Items assigned to no store']


@@ -127,7 +160,7 @@ async def do_shop(match):
             results.append(str(item))
         return results
 
             results.append(str(item))
         return results
 

-    if match.command('remove.item'):
+    if match.command('remove'):

         item_id = ' '.join(match.args()[1:])
         Data().remove_item(item_id)
         return ['Success']
         item_id = ' '.join(match.args()[1:])
         Data().remove_item(item_id)
         return ['Success']