#!/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 = 'All' 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) output.append({'url': u'https://monit.xxx.xxx', 'result': {'All': OrderedDict([(u'staging', 32), (u'monitoring1', 0)])}, 's_rate': {'green': 70.0, 'red': 30.0}, 'name': u'Test Environment'}) 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)