From c00fe792268b136e39d968c4f7b0faeaa05d585f Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Mon, 13 Apr 2020 14:19:36 +0200
Subject: [PATCH] [modules] re-enable taskwarrior

---
 modules/contrib/taskwarrior.py | 42 ++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 modules/contrib/taskwarrior.py

diff --git a/modules/contrib/taskwarrior.py b/modules/contrib/taskwarrior.py
new file mode 100644
index 0000000..a61ccb2
--- /dev/null
+++ b/modules/contrib/taskwarrior.py
@@ -0,0 +1,42 @@
+"""Displays the number of pending tasks in TaskWarrior.
+
+Requires the following library:
+    * taskw
+
+Parameters:
+    * taskwarrior.taskrc : path to the taskrc file (defaults to ~/.taskrc)
+"""
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+
+try:
+    from taskw import TaskWarrior
+except:
+    pass
+
+
+class Module(bumblebee.engine.Module):
+    """TaskWarrior module."""
+
+    def __init__(self, engine, config):
+        """Initialize taskwarrior module."""
+        super(Module, self).__init__(engine, config,
+                                     bumblebee.output.Widget(
+                                         full_text=self.output))
+        self._pending_tasks_count = "0"
+
+    def update(self, widgets):
+        """Return a string with the number of pending tasks from TaskWarrior."""
+        try:
+            taskrc = self.parameter("taskrc", "~/.taskrc")
+            w = TaskWarrior(config_filename=taskrc)
+            pending_tasks = w.filter_tasks({'status': 'pending'})
+            self._pending_tasks_count = str(len(pending_tasks))
+        except:
+            self._pending_tasks_count = 'Error'
+
+    def output(self, _):
+        """Format the task counter to output in bumblebee."""
+        return "{}".format(self._pending_tasks_count)