1 files changed, 41 insertions, 0 deletions
diff --git a/modules/scratchpad-counter.py b/modules/scratchpad-counter.py
new file mode 100644
index 0000000..5fc3c71
--- /dev/null
+++ b/modules/scratchpad-counter.py
@@ -0,0 +1,41 @@
+import i3
+import time
+import random
+from pprint import pprint
+
+CACHED_TIME = 1
+POSITION = 1
+HIDE_WHEN_NONE = True
+STRFORMAT = "{} ⌫"
+
+def find_scratch(tree):
+    if tree["name"] == "__i3_scratch":
+        return tree
+    else:
+        for x in tree["nodes"]:
+            result = find_scratch(x)
+            if result:
+                return result
+
+        return None
+
+class Py3status:
+    def __init__(self, *args, **kwargs):
+        self.count = -1
+        super(Py3status).__init__(*args, **kwargs)
+
+    def currentTitle(self, json, i3status_config):
+        count = len(find_scratch(i3.get_tree()).get("floating_nodes", []))
+
+        if self.count != count:
+            transformed = True
+            self.count = count
+        else:
+            transformed = False
+
+        return (POSITION, {
+                    'transformed': transformed,
+                    'full_text': '' if HIDE_WHEN_NONE and count == 0 else STRFORMAT.format(count),
+                    'name': 'scratchpad-count', 
+                    'cached_until': time.time() + CACHED_TIME,
+                    })