[modules] Add module for displaying RAM usage

Shows free RAM, total RAM, free RAM percentage
This commit is contained in:
Tobias Witek 2016-10-31 12:11:39 +01:00
parent 293c8453be
commit 36333c275f

View file

@ -0,0 +1,30 @@
import bumblebee.module
import psutil
def fmt(num, suffix='B'):
for unit in [ "", "Ki", "Mi", "Gi" ]:
if num < 1024.0:
return "{:.2f}{}{}".format(num, unit, suffix)
num /= 1024.0
return "{:05.2f%}{}{}".format(num, "Gi", suffix)
class Module(bumblebee.module.Module):
def __init__(self, args):
super(Module, self).__init__(args)
self._mem = psutil.virtual_memory()
def data(self):
self._mem = psutil.virtual_memory()
free = self._mem.available
total = self._mem.total
return "{}/{} ({:05.02f}%)".format(fmt(self._mem.available), fmt(self._mem.total), 100.0 - self._mem.percent)
def warning(self):
return self._mem.percent < 20
def critical(self):
return self._mem.percent < 10
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4