126 lines
3.1 KiB
Python
Executable file
126 lines
3.1 KiB
Python
Executable file
#!/usr/bin/python
|
|
|
|
import web
|
|
import requests
|
|
import xmltodict
|
|
import json
|
|
import os
|
|
import sys
|
|
import datetime
|
|
from collections import OrderedDict
|
|
from operator import itemgetter
|
|
import utils
|
|
|
|
urls = ('/', 'index',
|
|
'/help', 'help',
|
|
'/download', 'download'
|
|
)
|
|
|
|
app = web.application(urls, globals())
|
|
render = web.template.render('templates/', base="layout")
|
|
|
|
# Uncomment to turn debug off
|
|
web.config.debug = False
|
|
|
|
# Variables
|
|
output = []
|
|
|
|
# Functions
|
|
def calculate_count(data):
|
|
count = {}
|
|
z, nz = 0, 0
|
|
for _,v in data.items():
|
|
ls = v.values()
|
|
for v in ls:
|
|
if v == 0:
|
|
z += 1
|
|
else:
|
|
nz += 1
|
|
count['green'] = z
|
|
count['red'] = nz
|
|
return count
|
|
|
|
def arrange_services(services):
|
|
checks = {}
|
|
for service in services:
|
|
name = service['name']
|
|
name1, _, name2 = name.partition('-')
|
|
if name2 == '':
|
|
name2 = name1
|
|
name1 = 'others'
|
|
if name1 not in list(checks.keys()):
|
|
checks[name1] = OrderedDict()
|
|
checks[name1][name2] = int(service['status'])
|
|
return checks
|
|
|
|
def sorted_checks(checks):
|
|
s_checks = {}
|
|
for k,v in checks.iteritems():
|
|
s_checks[k] = OrderedDict(
|
|
sorted(v.iteritems(),key=itemgetter(1), reverse=True)
|
|
)
|
|
return s_checks
|
|
|
|
def getMonit():
|
|
output = []
|
|
xmlQuery = "/_status?format=xml"
|
|
|
|
with open('{0}/conf/servers.json'.format(os.path.expanduser('.'))) as f:
|
|
cf = json.loads(f.read())
|
|
|
|
for site in cf:
|
|
s = cf[site]
|
|
r = requests.get(s['url'] + xmlQuery,
|
|
auth=(s['user'], s['passwd']))
|
|
|
|
allstat = json.loads(json.dumps(xmltodict.parse(r.text)['monit']))
|
|
|
|
services = allstat['service']
|
|
server = {}
|
|
|
|
checks = arrange_services(services)
|
|
s_checks = sorted_checks(checks)
|
|
count = calculate_count(s_checks)
|
|
server = dict(name=site, url=s['url'],
|
|
result=s_checks, s_rate=count)
|
|
output.append(server)
|
|
print(datetime.datetime.now())
|
|
return(output)
|
|
|
|
# Classes
|
|
|
|
|
|
class monitDashboard(web.application):
|
|
def run(self, port=8080, *middleware):
|
|
func = self.wsgifunc(*middleware)
|
|
return web.httpserver.runsimple(func, ('0.0.0.0', port))
|
|
|
|
|
|
class index(object):
|
|
def GET(self):
|
|
return render.index(output=getMonit(),
|
|
now=datetime.datetime.now())
|
|
|
|
|
|
class help(object):
|
|
def GET(self):
|
|
return render.help()
|
|
|
|
|
|
class download(object):
|
|
def GET(self):
|
|
filename = 'health_report.xlsx'
|
|
output = getMonit()
|
|
utils.generate_report_excel(output, filename)
|
|
web.header('Content-Disposition',
|
|
'attachment; filename="health_report.xlsx"')
|
|
web.header('Content-type', 'application/octet-stream')
|
|
web.header('Cache-Control', 'no-cache')
|
|
return open(filename, 'rb').read()
|
|
|
|
|
|
# Main
|
|
if __name__ == "__main__":
|
|
app = monitDashboard(urls, globals())
|
|
app.run(port=8080)
|