66 lines
1.6 KiB
Python
Executable File
66 lines
1.6 KiB
Python
Executable File
#!/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', 'dev', 'proc', 'run', 'sys', '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():
|
|
block = Disk(ICON)
|
|
block.render()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|