First commit; Can recognize too-full devices

disk-usage-warn

@@ -0,0 +1,173 @@
+#!/usr/env python3
+
+"""
+
+Mike's Disk Usage Warner
+
+A simple script to emit warnings out to stderr if a disk's usage surpasses a threshold
+
+Copyright 2019 Mike Peralta; All rights reserved
+
+Released under the GNU GENERAL PUBLIC LICENSE v3 (See LICENSE file for more)
+
+"""
+
+
+#
+import re
+import subprocess
+import sys
+import yaml
+
+
+#
+class DiskUsageWarn:
+	
+	def __init__(self):
+		
+		self.__config_paths = []
+		self.__configs = []
+		
+		self.consume_arguments()
+	
+	@staticmethod
+	def log(s, o=None):
+		
+		message = "[Disk-Usage-Warn] " + s
+		if o:
+			message += " " + str(o)
+		
+		print(message)
+	
+	def consume_arguments(self):
+		
+		self.__config_paths = []
+		
+		for i in range(1, len(sys.argv)):
+			
+			arg = sys.argv[i]
+			
+			if arg == "--config":
+				i, one_path = self.consume_argument_companion(i)
+				self.__config_paths.append(one_path)
+				self.log("Found config path argument:", one_path)
+	
+	@staticmethod
+	def consume_argument_companion(arg_index):
+		
+		companion_index = arg_index + 1
+		if companion_index >= len(sys.argv):
+			raise Exception("Expected argument after", sys.argv[arg_index])
+		
+		return companion_index, sys.argv[companion_index]
+	
+	#
+	def consume_configs(self):
+		
+		#
+		self.__configs = []
+		
+		#
+		for config_path in self.__config_paths:
+			self.consume_config(config_path)
+		
+		self.log("Consumed " + str(len(self.__configs)) + " configs")
+		
+	#
+	def consume_config(self, path: str):
+		
+		config = self.load_config(path)
+		self.__configs.append(config)
+		self.log("Consumed config: " + path)
+	
+	@staticmethod
+	def load_config(path: str):
+		
+		# Open the file
+		f = open(path)
+		if not f:
+			raise Exception("Unable to open config file: " + path)
+		
+		# Parse
+		config = yaml.load(f)
+		
+		# Add the config file's own path
+		config["path"] = path
+		
+		return config
+	
+	def run(self):
+		
+		self.log("Begin")
+		
+		self.consume_configs()
+		self.do_configs()
+	
+	def do_configs(self):
+	
+		for config in self.__configs:
+			self.do_config(config)
+	
+	def do_config(self, config):
+		
+		# Pull the max usage
+		if "max-usage" not in config.keys():
+			raise Exception("Did not find config key: max-usage")
+		max_percent = config["max-usage"]
+		match = re.match("(?P<integer_percent>[0-9]+)%?", max_percent)
+		if not match:
+			raise Exception("Unable to parse configuration value for max-usage (integer percent)")
+		max_percent = int(match.group("integer_percent"))
+		
+		# Check each device
+		for device in config["devices"]:
+			
+			device_usage = self.get_device_usage(device)
+			self.log("Device Usage: " + device + " ==> " + str(device_usage) + "%")
+			if device_usage > max_percent:
+				self.log("Device is too full: " + str(device))
+	
+	def get_device_usage(self, device):
+		
+		args = ["df", device]
+		
+		returncode, stdout, stderr = self.execute_command(args)
+		if returncode != 0:
+			raise Exception("Failed to poll device usage\n" + stderr)
+		
+		# Grab percent
+		pattern = re.compile(".*?(?P<percent_integer>[0-9]+)%.*?", re.DOTALL)
+		match = pattern.match(str(stdout))
+		if not match:
+			raise Exception("Unable to parse device usage from:\n" + str(stdout))
+		percent_integer = int(match.group("percent_integer"))
+		
+		return percent_integer
+	
+	@staticmethod
+	def execute_command(args):
+		
+		# Start the process
+		#process = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE, env=env)
+		process = subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+		
+		#
+		stdout, stderr = process.communicate()
+		
+		if stdout:
+			stdout = stdout.decode()
+		if stderr:
+			stderr = stderr.decode()
+		
+		return process.returncode, stdout, stderr
+
+
+def main():
+	
+	duw = DiskUsageWarn()
+	duw.run()
+
+if __name__ == "__main__":
+	main()
+
+