Create NOC

NOC

@@ -0,0 +1,224 @@
+#!/usr/bin/env python
+import sys
+import os
+import datetime
+import time
+import urllib
+import urllib2
+import base64
+import json
+import traceback
+import RPi.GPIO as GPIO
+
+#General Options
+Green = 8
+Yellow = 9
+Red = 10
+Strobe = 7
+
+# Can't figure out why something isn't working? Turn this on.
+DEBUG = False
+
+LoopSleep = 5
+# Reverse logic here because a ground for the relay activates it
+On = False
+Off = True
+
+#Internet monitor options
+TestURLList = ["http://www.google.com","http://www.espn.com","http://www.edgenuity.com"]
+TestURLTimeout = 5
+MonitorSiteURLs = ["[UrlToTest]","[AnotherURLToTest]"]
+MonitorSiteTimeout = 6
+
+#Team city monitor options
+TeamCityURL = "[TeamcityUrl]/httpAuth/app/rest/builds?locator=running:any"
+TeamCityRunningURL = "[TeamcityUrl]/httpAuth/app/rest/builds?locator=running:true"
+TeamCityUsername = ""
+TeamCityPassword = ""
+BuildIdExclusions = ["bt86"] # Ignore builds that always fail for that team that can't get it together
+BuildIdStartsWithExclusion = "InDevelopment" # Way to prevent new builds from triggering alarm while being created
+
+
+def main():
+	GPIO.setwarnings(False)
+	GPIO.setmode(GPIO.BCM)
+	GPIO.setup(Green,GPIO.OUT)
+	GPIO.setup(Yellow,GPIO.OUT)
+	GPIO.setup(Red,GPIO.OUT)
+	GPIO.setup(Strobe,GPIO.OUT)
+
+	Light(Green,On)
+	time.sleep(1)
+	Light(Green,Off)
+	Light(Yellow,On)
+	time.sleep(1)
+	Light(Yellow,Off)
+	Light(Red,On)
+	time.sleep(1)
+	Light(Red,Off)
+	Light(Strobe,On)
+	time.sleep(1)
+	Light(Strobe,Off)
+
+	greenState = Off
+	yellowState = Off
+	redState = Off
+	strobeState = Off
+
+	while True:
+		if hasFailingBuilds():
+			console("Build Failure")
+			if redState == Off: #wasn't already off, checking to see if this is the loop that turns it on
+				os.system('mpg321 SadTrombone.mp3 &')
+			redState = On
+			greenState = Off
+			Light(Red,On)
+			Light(Green,Off)
+		else:
+			debug("Build OK")
+			redState = Off
+			greenState = On
+			Light(Green,On)
+			Light(Red,Off)
+
+		if hasRunningBuilds():
+			debug("Build Running")
+			yellowState = On
+			Light(Yellow,On)
+		else:
+			debug("No Build Running")
+			yellowState = Off
+			Light(Yellow,Off)
+
+		if connectionWorks() and applicationWorks():
+			debug("Connection OK")
+			strobeState = Off
+			Light(Strobe,Off)
+		else:
+			debug("Connection Failure")
+			if strobeState == Off: #wasn't already off, checking to see if this is the loop that turns it on
+				os.system('mpg321 SystemIsDown.mp3 &')
+			strobeState = On
+			Light(Strobe,On)
+
+		time.sleep(LoopSleep)
+
+def debug(message):
+	if(DEBUG):
+		print("%s : %s" % (datetime.datetime.now(), message))
+
+def console(message):
+	print("%s : %s" % (datetime.datetime.now(), message))
+
+def applicationWorks():
+	for i in MonitorSiteURLs:
+		try:
+			debug(i)
+			urllib2.urlopen(i,timeout=MonitorSiteTimeout).close()
+		except:
+			console(i)
+			console(sys.exc_info()[1])
+			return False
+			#pass
+
+	return True
+
+
+
+def connectionWorks():
+	#rotate the array so we aren't always trying the same outside site
+	global TestURLList
+	TestURLList = rotate(TestURLList)
+
+	for i in TestURLList:
+		try:
+			debug(i)
+			urllib2.urlopen(i,timeout=TestURLTimeout).close()
+			return True
+		except:
+			console(i)
+			console(sys.exc_info()[1])
+			pass
+
+	return False
+
+
+def hasFailingBuilds():
+	buildData = get_builds(TeamCityURL)
+	latestBuilds = get_latest_builds(buildData)
+	failures = searchForStatus(latestBuilds, "FAILURE")
+
+	if len(failures) > 0:
+		console(failures)
+		return True
+	else:
+		return False
+
+def hasRunningBuilds():
+	buildData = get_builds(TeamCityRunningURL)
+	latestBuilds = get_latest_builds(buildData)
+
+	if len(latestBuilds) > 0:
+		return True
+	else:
+		return False
+
+def searchForStatus(lastBuildStatus, status):
+	matchingBuildTypes = []
+	for i in lastBuildStatus:
+		if lastBuildStatus[i] == status:
+			matchingBuildTypes.append(i)
+
+	return matchingBuildTypes
+
+def get_latest_builds(jsonBuildData):
+	lastBuilds = {}
+	lastBuildStatus = {}
+
+	if int(jsonBuildData["count"]) > 0:
+		for i in jsonBuildData["build"]:
+			if i["buildTypeId"].startswith(BuildIdStartsWithExclusion):
+				pass
+			elif i["buildTypeId"] not in lastBuilds:
+				lastBuilds[i["buildTypeId"]] = i.get("number", -1)
+				lastBuildStatus[i["buildTypeId"]] = i.get("status","SUCCESS")
+			elif int(lastBuilds[i["buildTypeId"]]) < int(i.get("number", -1)):
+				lastBuilds[i["buildTypeId"]] = i.get("number", -1)
+				lastBuildStatus[i["buildTypeId"]] = i.get("status","SUCCESS")
+
+		for i in BuildIdExclusions:
+			if i in lastBuilds:
+				del lastBuilds[i]
+				del lastBuildStatus[i]
+
+	return lastBuildStatus
+
+def get_builds(url):
+	username = TeamCityUsername
+	password = TeamCityPassword
+
+	userpass = '%s:%s' % (username, password)	
+
+	request = urllib2.Request(url)
+
+	authInfo = base64.encodestring("%s:%s" % (username,password)).replace('\n', '')
+	request.add_header("Authorization", "Basic %s" % authInfo)
+	request.add_header('Accept', 'application/json')
+	response = urllib2.urlopen(request)
+
+	data = json.loads(response.read())
+	return data
+
+def Light(pin, status):
+	GPIO.output(pin,status)
+
+def rotate(l, y=1):
+   if len(l) == 0:
+      return l
+   y = y % len(l)    # Why? this works for negative y
+
+   return l[y:] + l[:y]
+
+
+if __name__=="__main__":
+	main()