report.py

#!/usr/bin/python

import os
import datetime
import sys
import platform
import socket
import json

from jinja2 import Environment, FileSystemLoader

env = Environment(
    loader=FileSystemLoader('template')
)

template = env.get_template('template.html.j2')


def pinghost(hostname):
    # Check os type to determine which ping command to use
    os_type = platform.platform()
    if "Windows" in os_type:
        response = os.system("ping -n 1 {0}".format(hostname))
    else:
        response = os.system("ping -c 1 {0}".format(hostname))
    return True if response == 0 else False


def checksock(hostname, port):
    if not isinstance(port, int):
        try:
            port = int(port)
        except ValueError:
            print('Port number is not numeric!')
            sys.exit()
    try:
        socket.create_connection((hostname, port), 2)
        return True
    except socket.error:
        print('%s failed on port: %s' % (hostname, str(port)))
        return False


def parsehost(hostfile):
    servers = []
    with open(hostfile, "r") as f:
        data = json.load(f)
        for item in data:
                if not isinstance(item["port"], int):
                    print("%s Port: %s is not a number!" % (item["url"], item["port"]))
                    sys.exit()

                if item["alias"]:
                    servers.append({"hostname": item["url"], "port": item["port"], "name": item["alias"]})
                else:
                    servers.append({"hostname": item["url"], "port": item["port"], "name": item["url"]})
    return servers


def createhtml(output_file_name, host_dict):
    refresh_rate = "60"
    today = (datetime.datetime.now())
    now = today.strftime("%m/%d/%Y %H:%M:%S")
    servers_up = 0.00
    servers_down = 0.00
    servers_percent = 0.00
    for h in host_dict:
        if h.get("status") == "up":
            servers_up += 1
        else:
            servers_down += 1
    server_total = (servers_up + servers_down)
    servers_percent = str(round((((server_total) - servers_down) / (server_total)), 2))

    template.stream(refresh_rate=refresh_rate,
                    today=today, now=now,
                    servers_up=servers_up,
                    servers_down=servers_down,
                    servers_percent=servers_percent,
                    server_total=server_total,
                    host_dict=host_dict).dump('index.html')


def main():
    names_list = "hostnames.json"
    # put the path that you would like the report to be written to
    output_file_name = "index.html"
    hosts = parsehost(names_list)
    for h in hosts:
        if h.get("port"):
            alive = checksock(h.get("hostname"), h.get("port"))
        else:
            alive = pinghost(h.get("hostname"))
        if alive:
            h.update(status="up")
        else:
            h.update(status="down")
    createhtml(output_file_name, hosts)


if __name__ == "__main__":
    print(f"[INFO] {datetime.datetime.now().strftime('%b %d %Y %H:%M:%S')} Checking hosts...")
    main()
    print(f"[INFO] {datetime.datetime.now().strftime('%b %d %Y %H:%M:%S')} Finished checking hosts")