X-Git-Url: https://apps.delaporte.us/gitweb/matrix-goal-bot.git/blobdiff_plain/303394a7dcea600b7f4e9db20a7056c8433eef75..refs/heads/main:/matrix_goal_bot/tasks.py

diff --git a/matrix_goal_bot/tasks.py b/matrix_goal_bot/tasks.py
index 9784b42..61413ec 100644
--- a/matrix_goal_bot/tasks.py
+++ b/matrix_goal_bot/tasks.py
@@ -12,6 +12,12 @@ def add_goal(c, name):
     result = Data().add_goal(name)
     print(result)
 
+@task
+def add_vision(c, name):
+    result = Data().add_vision(name)
+    print(result)
+
+
 @task
 def close_goal(c, goal_id):
     result = Data().complete_goal(goal_id)
@@ -47,6 +53,14 @@ def list(c):
     for goal in goals:
         print(goal)
 
+
+@task
+def list_visions(c):
+    items = Data().get_all_visions()
+    for item in items:
+        print(item)
+
+
 @task
 def later(c, goal_id):
     result = Data().put_off_goal(goal_id)
@@ -104,3 +118,7 @@ def wins(c):
     for line in results:
         print(line)
 
+@task
+def assign_to_vision(c, goal_id, vision_id):
+    result = Data().assign_vision(goal_id, vision_id)
+    print(result)