# pylint: disable=C0111,R0903

"""Shows free diskspace, total diskspace and the percentage of free disk space.

Parameters:
    * disk.warning: Warning threshold in % of disk space (defaults to 80%)
    * disk.critical: Critical threshold in % of disk space (defaults ot 90%)
    * disk.path: Path to calculate disk usage from (defaults to /)
"""

import os

import bumblebee.input
import bumblebee.output
import bumblebee.engine

class Module(bumblebee.engine.Module):
    def __init__(self, engine, config):
        super(Module, self).__init__(engine, config,
            bumblebee.output.Widget(full_text=self.diskspace)
        )
        self._path = self.parameter("path", "/")
        self._perc = 0
        self._used = 0
        self._size = 0

        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE,
            cmd="nautilus {}".format(self._path))

    def diskspace(self, widget):
        return "{} {}/{} ({:05.02f}%)".format(self._path,
            bumblebee.util.bytefmt(self._used),
            bumblebee.util.bytefmt(self._size), self._perc
        )

    def update(self, widgets):
        st = os.statvfs(self._path)
        self._size = st.f_frsize*st.f_blocks
        self._used = self._size - st.f_frsize*st.f_bavail
        self._perc = 100.0*self._used/self._size

    def state(self, widget):
        return self.threshold_state(self._perc, 80, 90)

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4

@@ -348,18 +348,22 @@
 %0Aimport 
-o
 s
+hutil
 %0A%0Aimport
@@ -1139,18 +1139,25 @@
 t = 
-os.statvfs
+shutil.disk_usage
 (sel
@@ -1193,28 +1193,13 @@
  st.
-f_frsize*st.f_blocks
+total
 %0A   
@@ -1221,43 +1221,14 @@
  = s
-elf._size - st.f_frsize*st.f_bavail
+t.used
 %0A   

Update update module