From 5c6b91dd065f361c4db8e8b6eecb668eaf1ff062 Mon Sep 17 00:00:00 2001
From: Tomas Babej <tomasbabej@gmail.com>
Date: Thu, 26 Mar 2015 23:15:04 +0100
Subject: [PATCH] taskwiki: Add TaskWikiAnnotate method

---
 ftplugin/vimwiki.vim | 1 +
 taskwiki/taskwiki.py | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/ftplugin/vimwiki.vim b/ftplugin/vimwiki.vim
index d801d15..3a83796 100644
--- a/ftplugin/vimwiki.vim
+++ b/ftplugin/vimwiki.vim
@@ -30,6 +30,7 @@ command! -range TaskWikiDelete :<line1>,<line2>py SelectedTasks().delete()
 command! -range TaskWikiStart :<line1>,<line2>py SelectedTasks().start()
 command! -range TaskWikiStop :<line1>,<line2>py SelectedTasks().stop()
 command! -range -nargs=* TaskWikiMod :<line1>,<line2>py SelectedTasks().modify(<q-args>)
+command! -range -nargs=* TaskWikiAnnotate :<line1>,<line2>py SelectedTasks().annotate(<q-args>)
 
 " Disable <CR> as VimwikIFollowLink
 if !hasmapto('<Plug>VimwikiFollowLink')
diff --git a/taskwiki/taskwiki.py b/taskwiki/taskwiki.py
index 7deb03c..65a6e9d 100644
--- a/taskwiki/taskwiki.py
+++ b/taskwiki/taskwiki.py
@@ -73,6 +73,14 @@ class SelectedTasks(object):
         if not self.tasks:
             print("No tasks selected.")
 
+    def annotate(self, annotation):
+        if not annotation:
+            annotation = util.get_input("Enter annotation: ")
+
+        for vimwikitask in self.tasks:
+            vimwikitask.task.add_annotation(annotation)
+            print("Task \"{0}\" annotated.".format(vimwikitask['description']))
+
     def info(self):
         for vimwikitask in self.tasks:
             out = util.tw_execute_safely(self.tw, [vimwikitask['uuid'], 'info'])