add disk space module

2015-12-05 17:34:20 -08:00 · 2015-12-05 17:34:20 -08:00 · 0b1610e1e8
commit 0b1610e1e8
parent a1f4f48ebf
1 changed files with 48 additions and 0 deletions
--- a/pymonitor/monitors/diskspace.py
+++ b/pymonitor/monitors/diskspace.py
@ -0,0 +1,48 @@
+from os import statvfs
+
+def diskspace(filesystems=[]):
+    for fs in filesystems:
+        stats = statvfs(fs)
+        
+        info = {
+            "fs": fs,
+            "diskfree": stats.f_bsize * stats.f_bavail,
+            "diskused": (stats.f_blocks-stats.f_bavail) * stats.f_bsize,
+            "disksize": stats.f_bsize * stats.f_blocks
+        }
+        
+        info["diskpctused"] = round(info["diskfree"]/info["disksize"], 2)
+        info["diskpctfree"] = round(info["diskused"]/info["disksize"], 2)
+        
+        yield info
+
+mapping = {
+    "diskspace": {
+        "properties": {
+            "diskfree": {
+                "type": "long"
+            },
+            "diskused": {
+                "type": "long"
+            },
+            "disksize": {
+                "type": "long"
+            },
+            "diskpctused": {
+                "type": "double"
+            },
+            "diskpctfree": {
+                "type": "double"
+            },
+            "fs": {
+                "type": "string"
+            }
+        }
+    }
+}
+
+if __name__ == '__main__':
+    #avg = load()
+    #print(' '.join([avg["load_1m"], avg["load_5m"], avg["load_15m"]]))
+    for item in diskspace(filesystems=["/", "/tmp/monitor"]):
+        print(item)