Fix name

[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..ee56105c77384feea18483b715ad5e267641ea79 100644 (file)
--- a/matrix_shop_bot/bot.py
+++ b/matrix_shop_bot/bot.py
@@ -4,7 +4,7 @@ import os
 
 import simplematrixbotlib as botlib
 
 
 import simplematrixbotlib as botlib
 

-from .db import Data, pretty_print
+from .db import Data

 
 MODULE='shop'
 
 
 MODULE='shop'
 


@@ -13,60 +13,178 @@ 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
 
     Commands:
       help - show this help

+      extra.help - show extended help

       add  - add an item to shop for
       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
-      list STORE_ID - list all items assigned to store
-      list-unsorted - list all items not assigned to any store
+      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 - show purchased items from STORE in last 90 days
+      stores - list all stores
+
+    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, including un-purchased
+      rename OLD,NEW - rename an item
+      rename.store OLD,NEW - rename a 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
       setup - run one time database setup

+      unsorted - list all items not assigned to any store
+      zone ZONE ITEM - list ITEM under given ZONE - ZONE may not contain spaces
+
+      Example: 
+        shop zone dairy sour cream

     '''
 
     if match.command('setup'):
         result = Data().setup()
         return [result]
 
     '''
 
     if match.command('setup'):
         result = Data().setup()
         return [result]
 

+    if match.command('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)
     if match.command('add'):
         new = ' '.join(match.args()[1:])
         result = Data().add_item(new)

-        return [f'Item added with id {result}']
+        results = [f'Item added with id {result}']
+        for item in Data().get_last_shopping_list():
+            results.append(str(item))
+        return results

 
 

-    if match.command('add-store'):
+    if match.command('add.store'):

         new = ' '.join(match.args()[1:])
         result = Data().add_store(new)
         return [f'Store added with id {result}']
 
         new = ' '.join(match.args()[1:])
         result = Data().add_store(new)
         return [f'Store added with id {result}']
 

+    if match.command('assign.item.to.store'):
+        item_id  = int(match.args()[1])
+        store_id = int(match.args()[2])
+        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)
+
+        # Now print the list for the assigned store
+        items = Data().get_last_shopping_list()
+        results = ['Shopping List']
+        for item in items:
+            results.append(str(item))
+        return results
+
+    if match.command('buy'):
+        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'):
+        store_name = ' '.join(match.args()[1:])
+        results = Data().clear_list(store_name)
+        return results
+
+    if match.command('clear.all'):
+        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:])
+        items = Data().get_shopping_list(store_name)
+        results = ['Shopping List']
+        for item in items:
+            results.append(str(item))
+        return results

 
     if match.command('list'):
 
     if match.command('list'):

-        items = Data().get_shopping_list(store_id)
+        items = Data().get_last_shopping_list()
+        results = ['Shopping List']
+        for item in items:
+            results.append(str(item))
+        return results
+
+
+    if match.command('past'):
+        store_name = ' '.join(match.args()[1:])
+        items = Data().list_purchases(store_name)
+        results = [f'# Previously purchased at {store_name}']

         for item in items:
             results.append(str(item))
         return results
         for item in items:
             results.append(str(item))
         return results

+    

 
     if match.command('unsorted'):
 
     if match.command('unsorted'):

-        items = Data().list_unsorted()
+        items = Data().get_unsorted_list()
+        results = ['Items assigned to no store']

         for item in items:
             results.append(str(item))
         return results
 
         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('rename'):
+        old,new = ' '.join(match.args()[1:]).split(',')
+        Data().rename_item(old,new)
+        return ['Success']
+
+    if match.command('rename.store'):
+        old,new = ' '.join(match.args()[1:]).split(',')
+        Data().rename_store(old,new)
+        return ['Success']
+
+    if match.command('remove'):
+        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']
+
+
+    if match.command('extra.help'):
+        return [extended_help]

 
     return [do_shop.__doc__]
 
 
     return [do_shop.__doc__]
 


@@ -79,10 +197,20 @@ async def shop_bot(room, message, bot):
     try:
         if match.prefix():
             result_lines = await do_shop(match)
     try:
         if match.prefix():
             result_lines = await do_shop(match)

+
+            message_lines = []

             for line in result_lines:
             for line in result_lines:

-                await bot.api.send_text_message(
-                    room.room_id, str(line))
+                message_lines.append(str(line))
+
+            message = '\n'.join(message_lines)
+
+            await bot.api.send_text_message(
+                room.room_id, message)
+

     except Exception as ex:
     except Exception as ex:

-        return ['You have:', '    no tea']
+        for line in ['You have:', '    no tea', str(ex)]:
+            await bot.api.send_text_message(
+                room.room_id, str(line))
+