From 797230da94f97c0fae34867deb4272560d6c6e84 Mon Sep 17 00:00:00 2001
From: abdoulaye <yatera.a@hotmail.fr>
Date: Tue, 21 Aug 2018 11:56:09 +0200
Subject: [PATCH] feat: add notmuch count module

Displays the result of a notmuch count query
default : unread emails wich path do not contained "Trash" (notmuch count "tag:unread AND NOT path:/.*Trash.*/")

Parameters:
    notmuch_count.query: notmuch count query to show result

Errors:
    if the notmuch query failed, the shown value is  -1

Dependencies:
    notmuch (https://notmuchmail.org/)
---
 bumblebee/modules/notmuch_count.py | 51 ++++++++++++++++++++++++++++++
 themes/icons/awesome-fonts.json    |  3 ++
 2 files changed, 54 insertions(+)
 create mode 100644 bumblebee/modules/notmuch_count.py

diff --git a/bumblebee/modules/notmuch_count.py b/bumblebee/modules/notmuch_count.py
new file mode 100644
index 0000000..88e6869
--- /dev/null
+++ b/bumblebee/modules/notmuch_count.py
@@ -0,0 +1,51 @@
+# pylint: disable=C0111,R0903
+
+"""Displays the result of a notmuch count query
+   default : unread emails wich path do not contained "Trash" (notmuch count "tag:unread AND NOT path:/.*Trash.*/")
+
+Parameters:
+    * notmuch_count.query: notmuch count query to show result 
+
+Errors:
+    if the notmuch query failed, the shown value is  -1
+
+Dependencies:
+    notmuch (https://notmuchmail.org/)
+"""
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import os
+
+class Module(bumblebee.engine.Module):
+
+
+    def __init__(self, engine, config):
+        super(Module, self).__init__(engine, config,
+            bumblebee.output.Widget(full_text=self.output)
+        )
+        self._notmuch_count_query = self.parameter("query", "tag:unread AND NOT path:/.*Trash.*/")
+        self._notmuch_count = self.count_notmuch()
+
+
+    def output(self, widget):
+       self._notmuch_count = self.count_notmuch()
+       return  str(self._notmuch_count)
+
+
+    def state(self, widgets):
+        if self._notmuch_count == 0:
+            return "empty"
+        return "items"
+
+
+    def count_notmuch(self):
+        try:
+            notmuch_count_cmd = "notmuch count " + self._notmuch_count_query
+            notmuch_count = int(bumblebee.util.execute(notmuch_count_cmd))
+            return notmuch_count
+        except Exception:
+            return -1
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
diff --git a/themes/icons/awesome-fonts.json b/themes/icons/awesome-fonts.json
index b9491da..4d43115 100644
--- a/themes/icons/awesome-fonts.json
+++ b/themes/icons/awesome-fonts.json
@@ -15,6 +15,9 @@
 	"load": { "prefix": "" },
 	"layout": { "prefix": "" },
 	"layout-xkb": { "prefix": "" },
+	"notmuch_count": { "empty": {"prefix": "\uf0e0" },
+	"items": {"prefix": "\uf0e0" }
+	},
 	"todo": { "empty": {"prefix": "" },
 	"items": {"prefix": "" },
 	"uptime": {"prefix": "" }