From: Edward Delaporte Date: Mon, 23 Mar 2026 23:06:12 +0000 (-0500) Subject: Fixup X-Git-Url: https://apps.delaporte.us/gitweb/matrix-goal-bot.git/commitdiff_plain/857ca258c6348a05ac776b4dae7d03a19668941c?hp=d9963fa251c1068be30ed1307fe8d8531e618f20 Fixup --- diff --git a/bot.py b/bot.py index 839bf07..0a40532 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 -from chore.db import Data as ChoreData - -CHORES = ChoreData() SERVER = os.environ.get('MATRIX_SERVER') BOT_NAME = os.environ.get('GOAL_BOT') @@ -22,128 +20,60 @@ 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] + Bot help here? - 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.'] + ''' + # logger.debug(f'match {str(match)}') + if match.command('help'): + return 'help' - if len(results) < 3: - results = [f'No assignments found for {person} on {day}.'] + if match.command('setup'): + result = Data().setup() + return result - for line in results: + if match.command('add'): + new = ' '.join(match.args()[1:]) + result = Data().add_goal(new) + return result - await bot.api.send_text_message( - room.room_id, line) + if match.command('list'): + goals = Data().list_goals() + return goals + + if match.command('add-task'): + result = Data().add_task(goal_id=goal_id, name=name) + return result + + if match.command('print'): + goal_id = new = match.args()[1] + goal,tasks = Data().get_goal(id) + + results = [] + results.append(f' ## {goal}') + for task in tasks: + if(task.complete_date): + results.append(f' - [x] {task}') + else: + results.append(f' - [ ] {task}') + + return 'Ye flask must be load bearing or something.' @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: - 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) + if match.prefix(): + result = await do_goal(match) await bot.api.send_text_message( room.room_id, result)