#!/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)