From 7aa1bc7b44f6eb2f1f22580bdaa7430e07544bb1 Mon Sep 17 00:00:00 2001
From: Tobias Witek <tobi@tobi-wan-kenobi.at>
Date: Sat, 14 Mar 2020 14:04:45 +0100
Subject: [PATCH] [modules] Reimport disk module

---
 modules/core/disk.py | 88 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 modules/core/disk.py

diff --git a/modules/core/disk.py b/modules/core/disk.py
new file mode 100644
index 0000000..db4d522
--- /dev/null
+++ b/modules/core/disk.py
@@ -0,0 +1,88 @@
+# pylint: disable=C0111,R0903
+
+"""Shows free diskspace, total diskspace and the percentage of free disk space.
+
+Parameters:
+    * disk.warning: Warning threshold in % of disk space (defaults to 80%)
+    * disk.critical: Critical threshold in % of disk space (defaults ot 90%)
+    * disk.path: Path to calculate disk usage from (defaults to /)
+    * disk.open: Which application / file manager to launch (default xdg-open)
+    * disk.format: Format string, tags {path}, {used}, {left}, {size} and {percent} (defaults to "{path} {used}/{size} ({percent:05.02f}%)")
+    * (deprecated) disk.showUsed: Show used space (defaults to yes)
+    * (deprecated) disk.showSize: Show total size (defaults to yes)
+    * (deprecated) disk.showPercent: Show usage percentage (defaults to yes)
+"""
+
+import os
+
+import bumblebee.input
+import bumblebee.output
+import bumblebee.engine
+import bumblebee.util
+
+class Module(bumblebee.engine.Module):
+    def __init__(self, engine, config):
+        super(Module, self).__init__(engine, config,
+            bumblebee.output.Widget(full_text=self.diskspace)
+        )
+        self._path = self.parameter("path", "/")
+        self._format = self.parameter("format", "{used}/{size} ({percent:05.02f}%)")
+        self._app = self.parameter("open", "xdg-open")
+
+        self._used = 0
+        self._left = 0
+        self._size = 0
+        self._percent = 0
+
+        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
+                                       cmd="{} {}".format(self._app,
+                                                          self._path))
+
+
+    def diskspace(self, widget):
+        used_str = bumblebee.util.bytefmt(self._used)
+        size_str = bumblebee.util.bytefmt(self._size)
+        left_str = bumblebee.util.bytefmt(self._left)
+        percent_str = self._percent
+
+        sused = self.has_parameter("showUsed")
+        ssize = self.has_parameter("showSize")
+        spercent = self.has_parameter("showPercent")
+
+        if all(not param for param in (sused, ssize, spercent)):
+            return self._format.format(path=self._path,
+                                       used=used_str,
+                                       left=left_str,
+                                       size=size_str,
+                                       percent=percent_str)
+        else:
+            rv = ""
+            sused = bumblebee.util.asbool(self.parameter("showUsed", True))
+            ssize = bumblebee.util.asbool(self.parameter("showSize", True))
+            spercent = bumblebee.util.asbool(self.parameter("showPercent", True))
+
+            if sused:
+                rv = "{}{}".format(rv, used_str)
+            if sused and ssize:
+                rv = "{}/".format(rv)
+            if ssize:
+                rv = "{}{}".format(rv, size_str)
+            if spercent:
+                if not sused and not ssize:
+                    rv = "{:05.02f}%".format(percent_str)
+                else:
+                    rv = "{} ({:05.02f}%)".format(rv, percent_str)
+            return rv
+
+
+    def update(self, widgets):
+        st = os.statvfs(self._path)
+        self._size = st.f_blocks * st.f_frsize
+        self._used = (st.f_blocks - st.f_bfree) * st.f_frsize
+        self._left = self._size - self._used;
+        self._percent = 100.0 * self._used/self._size
+
+    def state(self, widget):
+        return self.threshold_state(self._percent, 80, 90)
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4