Fixup

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

diff --git a/matrix_goal_bot/tasks.py b/matrix_goal_bot/tasks.py
index 5eb9bf7dfac1e9c13d74122ecaeaf8d3048703d0..61413ec98ae57928e3afabbc5f5924e0422aae47 100644 (file)
--- a/matrix_goal_bot/tasks.py
+++ b/matrix_goal_bot/tasks.py
@@ -1,6 +1,6 @@
 from invoke import task
 
 from invoke import task
 

-from .db import Data
+from .db import Data, pretty_print

 
 @task
 def setup(c):
 
 @task
 def setup(c):


@@ -12,6 +12,12 @@ def add_goal(c, name):
     result = Data().add_goal(name)
     print(result)
 
     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)
 @task
 def close_goal(c, goal_id):
     result = Data().complete_goal(goal_id)


@@ -25,13 +31,44 @@ def rename_goal(c, goal_id, name):
 
 
 @task
 
 
 @task

-def add_task(c, goal_id, name):
+def add_task(c, name):
+    result = Data().quick_add_task(name=name)
+    print(result)
+
+
+@task
+def done(c, task_id):
+    result = Data().complete_task(int(task_id))
+    print(result)
+
+@task
+def remove_task(c, goal_id, name):

     result = Data().add_task(goal_id=goal_id, name=name)
     print(result)
 
 
 @task
 def list(c):
     result = Data().add_task(goal_id=goal_id, name=name)
     print(result)
 
 
 @task
 def list(c):

+    goals = Data().get_current_goals()
+    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)
+    print(result)
+
+
+@task
+def list_all(c):

     goals = Data().get_all_goals()
     for goal in goals:
         print(goal)
     goals = Data().get_all_goals()
     for goal in goals:
         print(goal)


@@ -51,15 +88,37 @@ def list_done(c):
         print(goal)
 
 @task
         print(goal)
 
 @task

-def print_goal(c, id):
-    goal,tasks = Data().get_goal(id)
-    print(f' ## {goal}')
-    for task in tasks:
-        if(task.complete_date):
-            print(f' - [x] {task}')
-        else:
-            print(f' - [ ] {task}')
+def print_goal(c, goal_id):
+    goal, tasks = Data().get_goal(goal_id)
+    results = pretty_print(goal, tasks)
+    for line in results:
+        print(line)
+
+@task
+def start(c, goal_id):
+    vision, goals = Data().start_goal(goal_id)
+    results = pretty_print(vision, goals)
+    for line in results:
+        print(line)

 
 

-    
+@task
+def clear(c, goal_id):
+    results = Data().clear(goal_id)
+    print(results)
+
+@task
+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)