From 492e092a97aef97d853f95c4e4eefd1579a38190 Mon Sep 17 00:00:00 2001
From: Knut Ahlers <knut@ahlers.me>
Date: Wed, 11 Sep 2019 20:24:21 +0200
Subject: [PATCH] Migrate disk script to block usage

Signed-off-by: Knut Ahlers <knut@ahlers.me>
---
 .config/i3blocks/disk | 93 ++++++++++++++++++++++++-------------------
 1 file changed, 51 insertions(+), 42 deletions(-)

diff --git a/.config/i3blocks/disk b/.config/i3blocks/disk
index 0208759..245bac2 100755
--- a/.config/i3blocks/disk
+++ b/.config/i3blocks/disk
@@ -1,55 +1,64 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 import subprocess
 import sys
 
+from block import *
+
 CRIT = 90.0
 MOUNT_BLACKLIST = ['/dev', '/efi', '/proc', '/run', '/sys', '/tmp']
 FS_BLACKLIST = ['//', 'cgroup', 'tmpfs']
+ICON = '\ufaed'
+
+
+class Disk(Block):
+    def execute(self):
+        output = subprocess.check_output([
+            'sh', '-c', 'df -a -B 1 | awk \'{ printf "%s\\t%d\\t%d\\t%s\\n", $6, $2, $4, $1 }\''
+        ]).decode('utf-8')
+
+        fs = {}
+        for line in output.split('\n'):
+            if len(line) == 0 or line[0] != '/':
+                continue
+            (mountpoint, total, available, filesystem) = line.split('\t')
+            fs[mountpoint] = (total, available, filesystem)
+
+        color = '#50fa7b'
+
+        strings = []
+        for mount, data in fs.items():
+            skip = False
+            for b in MOUNT_BLACKLIST:
+                if mount.startswith(b):
+                    skip = True
+
+            for b in FS_BLACKLIST:
+                if data[2].startswith(b):
+                    skip = True
+
+            if skip:
+                continue
+
+            usage = 0.0
+            if float(data[0]) > 0.0:
+                usage = (1 - float(data[1])/float(data[0])) * 100.0
+
+            if usage > CRIT:
+                color = '#dd0000'
+
+            strings.append('<span color="#8FAAFC">{}</span>: <span color="{}">{:.0f}%</span>'.format(
+                mount,
+                color,
+                usage,
+            ))
+
+        return ' '.join(strings)
 
 
 def main():
-    output = subprocess.check_output([
-        'sh', '-c', 'df -a -B 1 | awk \'{ printf "%s\\t%d\\t%d\\t%s\\n", $6, $2, $4, $1 }\''
-    ]).decode('utf-8')
-
-    fs = {}
-    for line in output.split('\n'):
-        if len(line) == 0 or line[0] != '/':
-            continue
-        (mountpoint, total, available, filesystem) = line.split('\t')
-        fs[mountpoint] = (total, available, filesystem)
-
-    color = '#50fa7b'
-
-    strings = []
-    for mount, data in fs.items():
-        skip = False
-        for b in MOUNT_BLACKLIST:
-            if mount.startswith(b):
-                skip = True
-
-        for b in FS_BLACKLIST:
-            if data[2].startswith(b):
-                skip = True
-
-        if skip:
-            continue
-
-        usage = 0.0
-        if float(data[0]) > 0.0:
-            usage = (1 - float(data[1])/float(data[0])) * 100.0
-
-        if usage > CRIT:
-            color = '#dd0000'
-
-        strings.append('<span color="#8FAAFC">{}</span>: <span color="{}">{:.0f}%</span>'.format(
-            mount,
-            color,
-            usage,
-        ))
-
-    print('\ufaed '+' '.join(strings))
+    block = Disk(ICON)
+    block.render()
 
 
 if __name__ == '__main__':