220 lines
4.9 KiB
Python
Executable File
220 lines
4.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
#
|
|
# For debugging we use the bottom left corner pixel and:
|
|
#
|
|
# show red dot if failed to get data
|
|
# show blue dot if failed to refresh the cookie.
|
|
# show orange when we start refreshing data.
|
|
# show yellow when login fails.
|
|
#
|
|
#
|
|
# ver 1.1 added dispaly of ip address on boot in binary format.
|
|
#
|
|
#
|
|
|
|
|
|
import time
|
|
import json
|
|
import requests
|
|
import time
|
|
import unicornhat as unicorn
|
|
import urllib3
|
|
import socket
|
|
import commands
|
|
|
|
urllib3.disable_warnings()
|
|
|
|
unicorn.set_layout(unicorn.PHAT)
|
|
unicorn.rotation(180)
|
|
unicorn.brightness(0.4)
|
|
height,width=unicorn.get_shape()
|
|
|
|
|
|
# login details for your APIC
|
|
|
|
apic = '10.50.138.215'
|
|
username = 'aci-health'
|
|
password = 'aci-health'
|
|
|
|
|
|
|
|
# show blue pixels in each corner
|
|
def show_icon(timedel):
|
|
unicorn.set_pixel(0,0,0,0,255)
|
|
unicorn.set_pixel(0,3,0,0,255)
|
|
unicorn.set_pixel(7,0,0,0,255)
|
|
unicorn.set_pixel(7,3,0,0,255)
|
|
unicorn.show()
|
|
time.sleep(timedel)
|
|
|
|
|
|
|
|
# show ipv4 address in binary on the 4x8 LEDs
|
|
def show_ip(timedel):
|
|
for address in filter(None, commands.getoutput("hostname -I").split(" ")):
|
|
show_icon(1)
|
|
try:
|
|
print "processing: ",address
|
|
for octetidx,octet in enumerate(address.split(".")):
|
|
bin_octet="{0:08b}".format(int(octet))
|
|
for idx,c in enumerate(bin_octet[::-1]):
|
|
if c == "0":
|
|
unicorn.set_pixel(idx,octetidx,0,255,0)
|
|
elif c == "1":
|
|
unicorn.set_pixel(idx,octetidx,255,0,0)
|
|
print "display: ", address
|
|
unicorn.show()
|
|
time.sleep(timedel)
|
|
except:
|
|
print "skip ipv6"
|
|
pass
|
|
unicorn.clear()
|
|
unicorn.show()
|
|
return()
|
|
|
|
# some colour defs
|
|
red=(255,0,0)
|
|
orange=(255,128,0)
|
|
green=(0,255,0)
|
|
|
|
#
|
|
# get data from APIC function
|
|
#
|
|
def getdata():
|
|
|
|
url='https://%s/api/node/mo/topology/health.json' % apic
|
|
|
|
try:
|
|
print "Getting data from APIC...",apic
|
|
response = requests.get(url,cookies=cook,timeout=2,verify=False)
|
|
json_data = response.json()
|
|
health = json_data["imdata"][0]["fabricHealthTotal"]["attributes"]["cur"]
|
|
print "currnet health score is",health
|
|
show_unicorn(int(health))
|
|
print "\n---------------\n"
|
|
except:
|
|
unicorn.clear()
|
|
unicorn.set_pixel(0,0,255,0,0)
|
|
unicorn.show()
|
|
print "Missing health data, or timeout"
|
|
pass
|
|
return()
|
|
|
|
#
|
|
# refresh session cookie
|
|
#
|
|
|
|
def refreshSession(oldcook):
|
|
url = 'https://%s/api/aaaRefresh.json' % apic
|
|
print "\n---------------\n"
|
|
r = None
|
|
while r is None:
|
|
try:
|
|
r = requests.get(url, cookies=oldcook,timeout=10,verify=False)
|
|
newcook = r.cookies
|
|
print "got a new cookie"
|
|
print "\n---------------\n"
|
|
except:
|
|
unicorn.clear()
|
|
unicorn.set_pixel(0,0,0,0,255)
|
|
unicorn.show()
|
|
print "timeout refreshing cookie"
|
|
print "\n---------------\n"
|
|
newcook = loginAPIC()
|
|
pass
|
|
return(newcook)
|
|
|
|
#
|
|
# display the health score
|
|
#
|
|
|
|
def show_unicorn(health):
|
|
if health == 100:
|
|
colour=green
|
|
level=8
|
|
elif 90 <= health <= 99:
|
|
colour=green
|
|
level=7
|
|
elif 60 <= health <= 89:
|
|
colour=orange
|
|
level=6
|
|
elif 50 <= health <= 59:
|
|
colour=orange
|
|
level=5
|
|
elif 30 <= health <= 49:
|
|
colour=orange
|
|
level=4
|
|
elif 20 <= health <= 29:
|
|
colour=red
|
|
level=2
|
|
else:
|
|
colour=red
|
|
level=1
|
|
|
|
print "level",level
|
|
print "colour",colour
|
|
|
|
for y in range(level):
|
|
for x in range(width):
|
|
r=colour[0]
|
|
g=colour[1]
|
|
b=colour[2]
|
|
unicorn.set_pixel(y,x,r,g,b)
|
|
unicorn.show()
|
|
time.sleep(0.05)
|
|
return()
|
|
|
|
def loginAPIC():
|
|
headers={'content-type':'application/json'}
|
|
auth = {'aaaUser': {'attributes': {'name':username,'pwd':password}}}
|
|
url = 'https://%s/api/aaaLogin.json' % apic
|
|
r = None
|
|
while r is None:
|
|
try:
|
|
r = requests.post( url, data=json.dumps(auth), timeout=5, verify=False )
|
|
print "logging in, status ",r.status_code
|
|
cook = r.cookies
|
|
except:
|
|
unicorn.clear()
|
|
unicorn.set_pixel(0,0,255,255,0)
|
|
unicorn.show()
|
|
print "timeout logging in, but we will keep trying"
|
|
show_ip(5)
|
|
pass
|
|
|
|
return(cook)
|
|
|
|
# --------------------------------------------------------------------------
|
|
# Main program loop starts here
|
|
# --------------------------------------------------------------------------
|
|
|
|
print "Stared monitorig script..."
|
|
|
|
show_icon(1)
|
|
|
|
print "waiting for network"
|
|
|
|
while len(filter(None, commands.getoutput("hostname -I").split(" "))) < 1:
|
|
print "Waiting for at least 1 ip"
|
|
|
|
print "got an ip address"
|
|
|
|
# show ip addresses for 10s (1s delay between if more IPs found)
|
|
|
|
show_ip(10)
|
|
|
|
# Perform initial login, later we will need to refresh the cookie frequently
|
|
|
|
cook = loginAPIC()
|
|
|
|
while True:
|
|
for num in range(0,8):
|
|
unicorn.clear()
|
|
unicorn.set_pixel(0,0,255,128,0)
|
|
unicorn.show()
|
|
getdata()
|
|
time.sleep(5)
|
|
print "Refreshing cookie"
|
|
cook = refreshSession(cook)
|