Fewer messages when listing.

[matrix-goal-bot.git] / matrix_goal_bot / tasks.py

diff --git a/matrix_goal_bot/tasks.py b/matrix_goal_bot/tasks.py
index 1bacd918dae0aa41ca6260aed142b02800ed428c..61413ec98ae57928e3afabbc5f5924e0422aae47 100644 (file)
--- 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)
@@ -97,3 +111,14 @@ def setup(c):
     results = Data().setup()
     print(results)
 
+
+@task
+def wins(c):
+    results = Data().wins()
+    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)