backends: Move TW-specific annotation logic to TW backend

diff --git a/tasklib/backends.py b/tasklib/backends.py
index a1d858f54e24285aecb0b154700f5e62e9112841..f0c73ef4f0012a34c24a7550ef5584f806852f52 100644 (file)
--- a/tasklib/backends.py
+++ b/tasklib/backends.py
@@ -241,6 +241,14 @@ class TaskWarrior(object):
 
         self.execute_command([task['uuid'], 'done'])
 
 
         self.execute_command([task['uuid'], 'done'])
 

+    def annotate_task(self, task, annotation):
+        args = [task['uuid'], 'annotate', annotation]
+        self.execute_command(args)
+
+    def denotate_task(self, task, annotation):
+        args = [task['uuid'], 'denotate', annotation]
+        self.execute_command(args)
+

     def refresh_task(self, task, after_save=False):
         # We need to use ID as backup for uuid here for the refreshes
         # of newly saved tasks. Any other place in the code is fine
     def refresh_task(self, task, after_save=False):
         # We need to use ID as backup for uuid here for the refreshes
         # of newly saved tasks. Any other place in the code is fine

diff --git a/tasklib/task.py b/tasklib/task.py
index 5a865293302ff15eb84b1ddf52b48fc8ff56b95a..9ae63d48fe53a15ce5cdb551edd34e49b4002fe6 100644 (file)
--- a/tasklib/task.py
+++ b/tasklib/task.py
@@ -653,8 +653,7 @@ class Task(TaskResource):
         if not self.saved:
             raise Task.NotSaved("Task needs to be saved to add annotation")
 
         if not self.saved:
             raise Task.NotSaved("Task needs to be saved to add annotation")
 

-        args = [self['uuid'], 'annotate', annotation]
-        self.warrior.execute_command(args)
+        self.backend.annotate_task(self, annotation)

         self.refresh(only_fields=['annotations'])
 
     def remove_annotation(self, annotation):
         self.refresh(only_fields=['annotations'])
 
     def remove_annotation(self, annotation):


@@ -663,8 +662,8 @@ class Task(TaskResource):
 
         if isinstance(annotation, TaskAnnotation):
             annotation = annotation['description']
 
         if isinstance(annotation, TaskAnnotation):
             annotation = annotation['description']

-        args = [self['uuid'], 'denotate', annotation]
-        self.warrior.execute_command(args)
+
+        self.backend.denotate_task(self, annotation)

         self.refresh(only_fields=['annotations'])
 
     def _get_modified_fields_as_args(self):
         self.refresh(only_fields=['annotations'])
 
     def _get_modified_fields_as_args(self):