bumblebee-status/bumblebee/modules/disk.py
Tobias Witek fca3171556 [modules] Allow modules to provide default click actions
Pass the "output" object to the modules' constructor to allow them to
define their own callbacks.
Any user-provided callbacks take precedence and override those of the
module.
2016-11-01 08:09:10 +01:00

40 lines
1.1 KiB
Python

import os
import bumblebee.util
import bumblebee.module
def usage():
return "disk or disk::<path, defaults to '/'>"
def notes():
return "Warning is at 20% free diskspace, Critical at 10%."
def description():
return "Shows free diskspace, total diskspace and the percentage of free disk space."
class Module(bumblebee.module.Module):
def __init__(self, output, args):
super(Module, self).__init__(args)
self._path = args[0] if args else "/"
output.add_callback(module=self.__module__,
button=1, cmd="nautilus {instance}")
def data(self):
st = os.statvfs(self._path)
self._size = st.f_frsize*st.f_blocks
self._free = st.f_frsize*st.f_bavail
self._perc = 100.0*self._free/self._size
return "{} {}/{} ({:05.02f}%)".format(self._path, bumblebee.util.bytefmt(self._free), bumblebee.util.bytefmt(self._size), self._perc)
def instance(self):
return self._path
def warning(self):
return self._perc < 20
def critical(self):
return self._perc < 10
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4