Update to complete goals

diff --git a/bot.py b/bot.py
index 731113d2c053a9db0047298683dd752aff315e5c..a6e7f65615c78ab051a406262d385fc60025c77d 100644 (file)
--- 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 fa88dc59c27ee1aa070f179e3b6d54a1899a4721..6c3c332b311859877560ce3c1f4f80b0ff69f404 100644 (file)
--- 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