31 lines
854 B
Python
31 lines
854 B
Python
|
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
|