X-Git-Url: https://apps.delaporte.us/gitweb/matrix-shop-bot.git/blobdiff_plain/ea1529a8cdc661c83ea50c241a47f578de7d68e5..cbc79ff7988c4c70635b2b52dc58e7dea3f5720a:/matrix_shop_bot/bot.py diff --git a/matrix_shop_bot/bot.py b/matrix_shop_bot/bot.py index 9a36c63..ee56105 100644 --- a/matrix_shop_bot/bot.py +++ b/matrix_shop_bot/bot.py @@ -13,17 +13,17 @@ DATE_FORMAT = '%Y-%m-%d' 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): ''' - Prefix: .shop + Prefix: shop Synax: - .shop COMMAND ARGS + shop COMMAND ARGS Commands: help - show this help @@ -34,33 +34,35 @@ async def do_shop(match): 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 + past STORE - show purchased items from STORE in last 90 days stores - list all stores Examples: - .shop past costco - .shop at costco + shop past costco + shop at costco - .shop add sour cream - .shop add milk + shop add sour cream + shop add milk - .shop list + shop list - .shop buy 1 2 5 7 + 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 unsorted - list all items not assigned to any store - zone ZONE ITEM - list ITEM under given ZONE + zone ZONE ITEM - list ITEM under given ZONE - ZONE may not contain spaces Example: - .shop zone dairy sour cream + shop zone dairy sour cream ''' if match.command('setup'): @@ -147,7 +149,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...'] + results = [f'# Previously purchased at {store_name}'] for item in items: results.append(str(item)) return results @@ -160,6 +162,16 @@ async def do_shop(match): results.append(str(item)) return results + 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) @@ -185,9 +197,16 @@ async def shop_bot(room, message, bot): try: if match.prefix(): result_lines = await do_shop(match) + + message_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: for line in ['You have:', ' no tea', str(ex)]: await bot.api.send_text_message(