disk module

i3pystatus/core/util.py

@@ -12,6 +12,10 @@ __all__ = [
     "KeyConstraintDict", "PrefixedKeyDict",
 ]
 
+def round_dict(dic, places):
+    for key, value in dic.items():
+        dic[key] = round(value, places)
+
 class ModuleList(collections.UserList):
     def __init__(self, status_handler, module_base):
         self.status_handler = status_handler

i3pystatus/disk.py

@@ -0,0 +1,41 @@
+import os
+
+from i3pystatus import IntervalModule
+from .core.util import round_dict
+
+class Disk(IntervalModule):
+    """
+    Gets used, free, available and total amount of bytes on the given mounted filesystem.
+
+    These values can also be expressed in percentages with the percentage_used, percentage_free
+    and percentage_avail formats.
+    """
+
+    settings = (
+        "format", "path",
+        ("divisor", "divide all byte values by this value, commonly 1024**3 (gigabyte)"),
+    )
+    required = ("path",)
+    color = "#FFFFFF"
+    format = "{free}/{avail}"
+    divisor = 1024**3
+
+    def run(self):
+        cdict = {}
+
+        stat = os.statvfs(self.path)
+        cdict = {
+            "total": (stat.f_bsize * stat.f_blocks) / self.divisor,
+            "free": (stat.f_bsize * stat.f_bfree) / self.divisor,
+            "avail": (stat.f_bsize * stat.f_bavail) / self.divisor,
+            "used": (stat.f_bsize * (stat.f_blocks - stat.f_bfree)) / self.divisor,
+            "percentage_free": stat.f_bfree / stat.f_blocks * 100,
+            "percentage_avail": stat.f_bavail / stat.f_blocks * 100,
+            "percentage_used": (stat.f_blocks - stat.f_bfree) / stat.f_blocks * 100,
+        }
+        round_dict(cdict, 2)
+
+        self.output = {
+            "full_text": self.format.format(**cdict),
+            "color": self.color
+        }