backends: Move TW-specific annotation logic to TW backend

[etc/taskwarrior.git] / tasklib / backends.py

diff --git a/tasklib/backends.py b/tasklib/backends.py
index 92102cbb70f17efe471f818c99acb8e459d4336c..f0c73ef4f0012a34c24a7550ef5584f806852f52 100644 (file)
--- a/tasklib/backends.py
+++ b/tasklib/backends.py
@@ -4,6 +4,7 @@ import os
 import re
 import subprocess
 
+from tasklib.task import TaskFilter
 
 VERSION_2_1_0 = six.u('2.1.0')
 VERSION_2_2_0 = six.u('2.2.0')
@@ -18,6 +19,8 @@ VERSION_2_4_5 = six.u('2.4.5')
 
 class Backend(object):
 
+    filter_class = TaskFilter
+
     @abc.abstractmethod
     def filter_tasks(self, filter_obj):
         """Returns a list of Task objects matching the given filter"""
@@ -51,6 +54,14 @@ class Backend(object):
         """
         pass
 
+    @abc.abstractmethod
+    def annotate_task(self, task, annotation):
+        pass
+
+    @abc.abstractmethod
+    def denotate_task(self, task, annotation):
+        pass
+
     @abc.abstractmethod
     def sync(self):
         """Syncs the backend database with the taskd server"""
@@ -230,6 +241,14 @@ class TaskWarrior(object):
 
         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
@@ -252,7 +271,7 @@ class TaskWarrior(object):
             data.pop('id', None)
             data.pop('uuid', None)
 
-            taskfilter = TaskFilter(self)
+            taskfilter = self.filter_class(self)
             for key, value in data.items():
                 taskfilter.add_filter_param(key, value)
 
@@ -269,3 +288,5 @@ class TaskWarrior(object):
 
         return json.loads(output[0])
 
+    def sync(self):
+        self.execute_command(['sync'])