From f5b4674001d94e6aadbfbcd074fd730917e6c89d Mon Sep 17 00:00:00 2001 From: Edward Delaporte Date: Mon, 23 Mar 2026 22:32:47 -0500 Subject: [PATCH] Update to complete goals --- bot.py | 8 +++++++- goals/db.py | 7 ++++++- 2 files changed, 13 insertions(+), 2 deletions(-) diff --git a/bot.py b/bot.py index 731113d..a6e7f65 100644 --- a/bot.py +++ b/bot.py @@ -68,8 +68,14 @@ async def do_goal(match): else: results.append(f' - [ ] {task}') return results + + if match.command('done'): + goal_id = int(match.args()[1]) + logger.debug(f'Searching for goal {goal_id}...') + result = Data().complete_goal(goal_id) + return [result] - return 'Ye flask must be load bearing or something.' + return __doc__ @bot.listener.on_message_event diff --git a/goals/db.py b/goals/db.py index fa88dc5..6c3c332 100644 --- a/goals/db.py +++ b/goals/db.py @@ -85,7 +85,12 @@ class Data(): pass def complete_goal(self, id): - pass + with Session(self.engine) as session: + cmd = select(Goal).where(Goal.id == id) + goal = session.exec(cmd).first() + goal.complete_date = dt.today() + session.commit() + return self.get_goal(id) def complete_task(self, id): pass -- 2.47.3