Plugin for monitoring directory size

nixstatsagent/nixstatsagent.py

@@ -26,7 +26,7 @@
 import urllib
 import urllib2
 
-__version__ = '1.1.54'
+__version__ = '1.1.55'
 
 __FILEABSDIRNAME__ = os.path.dirname(os.path.abspath(__file__))
 

nixstatsagent/plugins/dirsize.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+import os
+import subprocess
+import plugins
+import json
+
+class Plugin(plugins.BasePlugin):
+    __name__ = 'dirsize'
+
+    def run(self, config):
+        '''
+        Monitor total directory sizes, specify the directories you want to monitor in /etc/nixstats.ini
+        '''
+
+        data = {}
+        my_dirs = config.get('dirsize', 'dirs').split(',')
+
+        for dir in my_dirs:
+            data[dir] = {'bytes': os.popen('du -c {} | grep total'.format(dir)).read().replace('total', '').rstrip()}
+
+        return data
+
+
+if __name__ == '__main__':
+    Plugin().execute()