minimal plugin to store job state in a taskwarrior instance

moa/plugin/job/task.py

@@ -0,0 +1,81 @@
+# Copyright 2009-2011 Mark Fiers
+# The New Zealand Institute for Plant & Food Research
+#
+# This file is part of Moa - http://github.com/mfiers/Moa
+#
+# Licensed under the GPL license (see 'COPYING')
+#
+"""
+**task** - integrate taskwarrior to track running tasks
+-------------------------------------------------------
+
+You will need to have taskwarrior installed.
+
+"""
+
+import re
+import socket
+import subprocess as sp
+
+import moa.ui
+import moa.logger
+from moa.sysConf import sysConf
+
+l = moa.logger.getLogger(__name__)
+
+
+def runTask(cl):
+    print cl
+    sp.call(cl.split())
+
+
+def _getTaskId(job):
+    #find the task
+    uid = '%s-%s' % (job.conf.ppud, job.conf.pud)
+    cl = ['task', uid, '_ids']
+    P = sp.Popen(cl, stdout=sp.PIPE)
+    out, err = P.communicate()
+    pidlist = out.strip().split()
+    if len(pidlist) > 1:
+        moa.ui.warn("multiple jobs with the warn")
+        return pidlist[-1]
+
+    if len(pidlist) == 1:
+        return pidlist[0]
+
+    project = job.conf.get('project', 'no_project')
+    title = job.conf.get('title')
+
+    cl = ['task', 'add',
+          'project:"%s"' % project,
+          '+moa', '--', title]
+
+    P = sp.Popen(cl, stdout=sp.PIPE)
+    out, err = P.communicate()
+    taskid = re.search('Created task ([0-9]+)', out).groups()[0]
+    moa.ui.message("new taskwarrior task %s" % taskid)
+
+    runTask("task %s annotate uid:%s-%s" % (taskid, job.conf.ppud,
+                                            job.conf.pud))
+
+    sysConf.plugins.task.id = taskid
+    return taskid
+
+
+def hook_pre_run(job):
+
+    taskid = _getTaskId(job)
+    moa.ui.message("TaskId %s" % taskid)
+    hostname = socket.gethostname()
+
+    runTask("task %s modify +moa +%s" % (taskid, hostname))
+    runTask("task %s start" % (taskid))
+
+
+def hook_finish(job):
+    """
+    cache!!
+    """
+    #print job, job.isMoa()rm
+    taskid = sysConf.plugins.task.id
+    runTask("task %s stop" % (taskid))