X-Git-Url: https://us-delaporte.pagekite.me/gitweb/matrix-goal-bot.git/blobdiff_plain/e268526289d0a4957706be0b40e752b6123ca0b2..a82ef10eea38d5cec30ae6e4b2d6a7257735ac53:/bot.py diff --git a/bot.py b/bot.py index 839bf07..efc56a5 100644 --- a/bot.py +++ b/bot.py @@ -1,13 +1,11 @@ -import os from datetime import datetime as dt +import logging +import os import simplematrixbotlib as botlib -from school import (get_assignments, STUDENTS) +from goals.db import Data, pretty_print -from chore.db import Data as ChoreData - -CHORES = ChoreData() SERVER = os.environ.get('MATRIX_SERVER') BOT_NAME = os.environ.get('GOAL_BOT') @@ -22,130 +20,135 @@ creds = botlib.Creds(SERVER, BOT_NAME, BOT_PWD) bot = botlib.Bot(creds) PREFIX = '!goal' -@bot.listener.on_message_event -async def goals(room, message): - match = botlib.MessageMatch(room, message, bot, PREFIX) - - if not match.is_not_from_this_bot() and match.prefix(): - return +logger = logging.getLogger(__name__) +logging.basicConfig(filename='goals.log', encoding='utf-8', level=logging.DEBUG) +logging.getLogger('nio').setLevel(logging.WARNING) - if match.command("echo"): - await bot.api.send_text_message( - room.room_id, " ".join(arg for arg in match.args()) - ) - - help_text = ''' - # Print this help text - !school help - # List available students - !school list - # Get today's assignments - !school get [NAME] - # Get an assignment for a date - !school get [Name] 2026-03-09 +async def do_goal(match): ''' - if match.command("help"): - await bot.api.send_text_message( - room.room_id, help_text) - - if match.command("list"): - student_list = f"Known students are {STUDENTS}." - await bot.api.send_text_message( - room.room_id, student_list) - - person = '' - if match.command("get"): - print(match.args()) - if(len(match.args()) < 2): - result = ['Please specify person to get assignments for.'] - else: - person = match.args()[1] - - if(len(match.args()) > 2): - day = match.args()[2] - else: - day = today() - - if(person and day): - await bot.api.send_text_message(room.room_id, f'Fetching assignments for {person} on {day}') - results = get_assignments(person, day) - else: - results = ['Oops.'] - - if len(results) < 3: - results = [f'No assignments found for {person} on {day}.'] - - for line in results: + Prefix: !goal + + Synax: + !goal COMMAND ARGS + + Commands: + help - show this help + add - add a goal + add-task ID - add a task under the goal specified by ID + list - list incomplete goals + list-all - list everything + print ID - print a goal and all tasks + print-vision ID - print a vision and all goals + done ID - mark task done by ID + goal-done ID - mark goal done by ID + set-vision GOAL_ID VISION_ID - associate GOAL_ID with VISION_ID + visions - list incomplete visions - await bot.api.send_text_message( - room.room_id, line) + ''' + try: + + if match.command('setup'): + result = Data().setup() + return [result] + + if match.command('add'): + new = ' '.join(match.args()[1:]) + result = Data().add_goal(new) + return [f'Goal 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-task'): + goal_id = int(match.args()[1]) + new = ' '.join(match.args()[2:]) + result = Data().add_task(goal_id=goal_id, name=new) + 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) + return results + + + 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('goal-done'): + goal_id = int(match.args()[1]) + logger.debug(f'Searching for goal {goal_id}...') + Data().complete_goal(goal_id) + goal,tasks = Data().get_goal(goal_id) + results = pretty_print(goal, tasks) + return results + + + 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 + + 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 + return results + + if match.command('visions'): + items = Data().get_all_visions() + results = ['Visions are'] + if not items: + results.append('-- No current visions --') + + for item in items: + results.append(item) + return results + + + if match.command('set-vision'): + goal_id = int(match.args()[1]) + vision_id = int(match.args()[1]) + Data().assign_vision(goal_id, vision_id) + vision,goals = Data().get_vision(vision_id) + results = pretty_print(vision, goals) + return results + + return [do_goal.__doc__] + except Exception as ex: + return ['Ye cannot get ye flask.'] @bot.listener.on_message_event -async def chores(room, message): - match = botlib.MessageMatch(room, message, bot, '!chore') +async def goals(room, message): + match = botlib.MessageMatch(room, message, bot, PREFIX) if not match.is_not_from_this_bot() and match.prefix(): return - if match.command('list'): - - await bot.api.send_text_message( - room.room_id, 'Listing chores...') - - result = CHORES.list() - for item in result: + if match.prefix(): + result_lines = await do_goal(match) + for line in result_lines: await bot.api.send_text_message( - room.room_id, item) - - await bot.api.send_text_message( - room.room_id, '-- End of List --') - - if match.command('setup'): - await bot.api.send_text_message( - room.room_id, f'Doing chore setup...') - result = CHORES.setup() - result = await bot.api.send_text_message( - room.room_id, f'Result: {result}') - - - if match.command('add'): - new_chore = ' '.join(match.args()[1:]) - - await bot.api.send_text_message( - room.room_id, f'Adding chore...{new_chore}...') - - result = CHORES.add(new_chore) - await bot.api.send_text_message( - room.room_id, result) - - - - if match.command('check'): - chore_id = match.args()[1] - - result = CHORES.check(chore_id) - await bot.api.send_text_message( - room.room_id, result) - - if match.command('done'): - chore_id = match.args()[1] - result = CHORES.done(chore_id) - await bot.api.send_text_message( - room.room_id, result) - - if match.command('undo'): - chore_id = match.args()[1] - result = CHORES.undo(chore_id) - await bot.api.send_text_message( - room.room_id, result) - - - if match.command('remove'): - chore_id = match.args()[1] - result = CHORES.remove(chore_id) - await bot.api.send_text_message( - room.room_id, result) + room.room_id, str(line)) bot.run()