From 626ed97fcdba2a49d37b7e411982f89e0bd1788d Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Sat, 18 Apr 2020 16:24:26 +0200
Subject: [PATCH] [modules] re-enable indicator

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

diff --git a/modules/contrib/indicator.py b/modules/contrib/indicator.py
new file mode 100644
index 0000000..3b57e33
--- /dev/null
+++ b/modules/contrib/indicator.py
@@ -0,0 +1,51 @@
+#pylint: disable=C0111,R0903
+
+"""Displays the indicator status, for numlock, scrolllock and capslock 
+
+Parameters:
+    * indicator.include: Comma-separated list of interface prefixes to include (defaults to "numlock,capslock")
+    * indicator.signalstype: If you want the signali type color to be "critical" or "warning" (defaults to "warning")
+"""
+
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        widgets = []
+        self.status = False
+        super(Module,self).__init__(engine, config, widgets)
+        self._include = tuple(filter(len, self.parameter("include", "NumLock,CapsLock").split(",")))
+        self._signalType = self.parameter("signaltype") if not self.parameter("signaltype") is None else "warning"
+
+    def update(self, widgets):
+        self._update_widgets(widgets)
+
+    def state(self, widget):
+        states = []
+        if widget.status:
+            states.append(self._signalType)
+        elif not widget.status:
+            states.append("normal")
+        return states
+
+    def _update_widgets(self, widgets):
+        status_line = "" 
+        for line in bumblebee.util.execute("xset q").replace(" ", "").split("\n"):
+            if "capslock" in line.lower():
+                status_line = line  
+                break
+            
+        for indicator in self._include:
+            widget = self.widget(indicator)
+            if not widget:
+                widget = bumblebee.output.Widget(name=indicator)
+                widgets.append(widget)
+
+            widget.status = True if indicator.lower()+":on" in status_line.lower() else False
+            widget.full_text(indicator)
+
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4