X-Git-Url: https://us-delaporte.pagekite.me/gitweb/matrix-shop-bot.git/blobdiff_plain/17242e775a86365e2e2e1a7bfea3b9a6b7847b53..19bec36c1b5f7bcf005ca9b842dd902a32675ce0:/matrix_shop_bot/bot.py diff --git a/matrix_shop_bot/bot.py b/matrix_shop_bot/bot.py index 9c0bec0..734cdb2 100644 --- a/matrix_shop_bot/bot.py +++ b/matrix_shop_bot/bot.py @@ -4,157 +4,128 @@ import os import simplematrixbotlib as botlib -from .db import Data, pretty_print +from .db import Data -SERVER = os.environ.get('MATRIX_SERVER') -BOT_NAME = os.environ.get('GOAL_BOT') -BOT_PWD = os.environ.get('GOAL_BOT_PWD') - -def get_bot(): - creds = botlib.Creds(SERVER, BOT_NAME, BOT_PWD) - return botlib.Bot(creds) +MODULE='shop' DATE_FORMAT = '%Y-%m-%d' def today(): return dt.now().strftime(DATE_FORMAT) -PREFIX = '!goal' - +PREFIX = f'!{MODULE}' logger = logging.getLogger(__name__) -logging.basicConfig(filename='goals.log', encoding='utf-8', level=logging.DEBUG) +logging.basicConfig(filename=f'{MODULE}.log', encoding='utf-8', level=logging.DEBUG) logging.getLogger('nio').setLevel(logging.WARNING) -async def do_goal(match): +async def do_shop(match): ''' - Prefix: !goal + Prefix: !shop Synax: - !goal COMMAND ARGS + !shop COMMAND ARGS Commands: help - show this help - add - add a goal - add-task ID - add a task under the goal specified by ID - add-vision - add a new vision (collection of goals) - close-goal ID - mark goal done by ID - close-vision ID - mark vision as complete by ID - done ID - mark task done by ID - list - list incomplete goals - list-all - list everything - list-done - list completed goals - print GOAL_ID - print a goal and all tasks - print-vision ID - print a vision and all goals - set-vision GOAL_ID VISION_ID- associate GOAL_ID with VISION_ID - visions - list incomplete visions + add - add an item to shop for + add-store - add a store + assign ITEM_ID STORE_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 + unsorted - list all items not assigned to any store + remove-item 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 ''' if match.command('setup'): 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_goal(new) - return [f'Goal added with id {result}'] + result = Data().add_item(new) + return [f'Item added with id {result}'] - if match.command('list'): - goals = Data().get_all_goals() - results = ['# Goals list...'] - for goal in goals: - results.append(str(goal)) - return results + if match.command('add-store'): + new = ' '.join(match.args()[1:]) + result = Data().add_store(new) + return [f'Store added with id {result}'] + if match.command('assign'): + item_id = int(match.args()[1]) + store_id = int(match.args()[2]) - if match.command('list-done'): - goals = Data().get_done_goals() - results = ['# Completed Goals...'] - for goal in goals: - results.append(str(goal)) - return results + if not isinstance(store_id, int): + store_id = None + Data().assign_store(item_id, store_id) - if match.command('add-task'): - goal_id = int(match.args()[1]) - new = ' '.join(match.args()[2:]) - result = Data().add_task(goal_id=goal_id, name=new) - goal,tasks = Data().get_goal(goal_id) - results = pretty_print(goal, tasks) - return results - # return [f'Added task with id {result}'] - - if match.command('add-vision'): - new = ' '.join(match.args()[1:]) - result = Data().add_vision(new) - return [f'Vision added with id {result}'] - - if match.command('print'): - goal_id = int(match.args()[1]) - logger.debug(f'Searching for goal {goal_id}...') - goal,tasks = Data().get_goal(goal_id) - results = pretty_print(goal, tasks) + # 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('print-vision'): - get_id = int(match.args()[1]) - vision,goals = Data().get_vision(get_id) - results = pretty_print(vision, goals) - return results - - if match.command('close-goal'): - goal_id = int(match.args()[1]) - vision_id = Data().complete_goal(goal_id) - vision,goals = Data().get_vision(vision_id) - results = pretty_print(vision, goals) - return results + if match.command('clear'): + store_id = ' '.join(match.args()[1:]) + Data().clear_list(store_id) + self.last_store_id = None + return ['Success'] - if match.command('close-vision'): - vision_id = int(match.args()[1]) - Data().complete_vision(vision_id) - vision,goals = Data().get_vision(vision_id) - results = pretty_print(vision, goals) - return results + 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_str = ' '.join(match.args()[1]) - if match.command('done'): - task_id = int(match.args()[1]) - goal_id = Data().complete_task(task_id) - goal,tasks = Data().get_goal(goal_id) - results = pretty_print(goal, tasks) - return results + store_id = int(store_id_str) if store_id_str else None + items = Data().get_shopping_list(store_id) - if match.command('list-all'): - goals = Data().get_all_goals() - results = ['# Everything is...'] - for goal in goals: - _,tasks = Data().get_goal(goal.id) - details = pretty_print(goal, tasks) - results += details + results = ['Shopping List'] + for item in items: + results.append(str(item)) return results - if match.command('visions'): - items = Data().get_all_visions() - results = ['# Visions'] - if not items: - results.append('-- No current visions --') + if match.command('unsorted'): + items = Data().get_unsorted_list() + results = ['Items assigned to no store'] for item in items: - results.append(item) + results.append(str(item)) return results + if match.command('remove-item'): + item_id = ' '.join(match.args()[1:]) + Data().remove_item(item_id) + return ['Success'] - if match.command('set-vision'): - goal_id = int(match.args()[1]) - vision_id = int(match.args()[2]) - Data().assign_vision(goal_id=goal_id, vision_id=vision_id) - vision,goals = Data().get_vision(vision_id) - results = pretty_print(vision, goals) - return results + if match.command('remove-store'): + store_id = ' '.join(match.args()[1:]) + Data().remove_store(store_id) + return ['Success'] - return [do_goal.__doc__] + return [do_shop.__doc__] -async def goal_bot(room, message, bot): +async def shop_bot(room, message, bot): match = botlib.MessageMatch(room, message, bot, PREFIX) if not match.is_not_from_this_bot() and match.prefix(): @@ -162,11 +133,14 @@ async def goal_bot(room, message, bot): try: if match.prefix(): - result_lines = await do_goal(match) + result_lines = await do_shop(match) for line in result_lines: await bot.api.send_text_message( room.room_id, str(line)) except Exception as ex: - return ['Ye cannot get ye flask.'] + for line in ['You have:', ' no tea', str(ex)]: + await bot.api.send_text_message( + room.room_id, str(line)) +