Forgot environment variables in check_hadoop_apps

check_hadoop_apps

@@ -33,11 +33,15 @@ STATE_LABEL = {
 # Default exit code
 EXIT_CODE = STATE_OK
 
+py_env = os.environ.copy()
+py_env['JAVA_HOME'] = '/usr/lib/jvm/java-8-oracle'
+py_env['HADOOP_HOME'] = '/usr/lib/hadoop'
+
 def hadoop_query_running_apps():
 	try:
 		data = {}
 
-		hadoop = subprocess.Popen(["yarn", "application", "-list", "-appStates", "RUNNING"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+		hadoop = subprocess.Popen(["yarn", "application", "-list", "-appStates", "RUNNING"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT, env=py_env)
 
 		# Format the output data from CSV to an array of dicts for ease-of-use
 		hadoopStd = str(hadoop.stdout.read(), 'UTF8')
@@ -65,7 +69,7 @@ def hadoop_query_app(appID):
 	try:
 		data = {}
 
-		yarn = subprocess.Popen(["yarn", "application", "-status", appID], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+		yarn = subprocess.Popen(["yarn", "application", "-status", appID], stdout=subprocess.PIPE, stderr=subprocess.STDOUT, env=py_env)
 
 		yarnStd = str(yarn.stdout.read(), 'UTF8')
 		results = yarnStd.splitlines()
@@ -83,7 +87,7 @@ def hadoop_query_app(appID):
 
 def yarn_kill_app(appID):
 	try:
-		yarn = subprocess.Popen(["yarn", "application", "-kill", appID], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+		yarn = subprocess.Popen(["yarn", "application", "-kill", appID], stdout=subprocess.PIPE, stderr=subprocess.STDOUT, env=py_env)
 	except FileNotFoundError as e:
 		print("UNKNOWN - Could not detect yarn command. |")
 		sys.exit(STATE_OK)
@@ -114,7 +118,7 @@ def main():
 				EXIT_CODE = STATE_WARNING
 
 		if (APP_STATE == STATE_CRITICAL):
-			print("[{0}] {1}: {2}: Has been running for {3} hours.  Killing application... | ".format(id, app['Application-Name'], STATE_LABEL[APP_STATE], str(int(app_info['Running-Time'] / 3600))))
+			print("[{0}] Application-Name: {1}: State {2}: Has been running for {3} hours.  Killing application... | ".format(id, app['Application-Name'], STATE_LABEL[APP_STATE], str(int(app_info['Running-Time'] / 3600))))
 
 	sys.exit(EXIT_CODE)