1 files changed, 48 insertions, 0 deletions
diff --git a/overseerlib/plugins/diskfree_plugin.py b/overseerlib/plugins/diskfree_plugin.py
new file mode 100644
index 0000000..d0de752
--- /dev/null
+++ b/overseerlib/plugins/diskfree_plugin.py
@@ -0,0 +1,48 @@
+#!/usr/bin/python
+#
+# Copyright (C) 2014  Codethink Limited
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+
+import os
+
+import cliapp
+
+import overseerlib
+
+
+class DiskFreePlugin(cliapp.Plugin):
+
+    def enable(self):
+        self.app.add_subcommand('diskfree', self.diskfree,
+                                arg_synopsis='PARTITION THRESHOLD')
+
+    def disable(self):
+        pass
+
+    def diskfree(self, args):
+        if len(args) < 2:
+            raise cliapp.AppException('diskfree needs a partition and a '
+                                      'threshold to be specified')
+        partition = args[0]
+        threshold = args[1]
+        threshold_type = args[2] if len(args) == 3 else 'high'
+
+        if threshold.isdigit() == False:
+            threshold = self.app.human_to_bytes(threshold)
+
+        st = os.statvfs(partition)
+        free = st.f_bavail * st.f_frsize
+        print free