move stuff to class, add prettifier, use pretty names

This commit is contained in:
Sophie Schiller 2021-04-10 15:34:41 +02:00
parent e03a10abb6
commit f899e6e13b

View file

@ -6,124 +6,118 @@ import configparser
import urllib3 import urllib3
urllib3.disable_warnings() urllib3.disable_warnings()
class StatusPage:
def do_api_calls(self):
#services
request_url = "{}/v1/objects/services".format(self.config['icinga2_api']['baseurl'])
headers = {
'Accept': 'application/json',
'X-HTTP-Method-Override': 'GET'
}
requestbody = {
"attrs": [ "name", "state", "last_check_result", "host_name", "display_name" ],
"joins": [ "host.name", "host.state", "host.last_check_result", "host.vars" ],
"filter": self.config['filters']['services'],
}
r = requests.get(request_url,
headers=headers,
data=json.dumps(requestbody),
auth=(self.config['icinga2_api']['username'], self.config['icinga2_api']['password']),
verify=False)
def do_api_calls(config): if (r.status_code == 200):
data = {} self.services = r.json()
else:
#services r.raise_for_status()
request_url = "{}/v1/objects/services".format(config['icinga2_api']['baseurl'])
headers = {
'Accept': 'application/json',
'X-HTTP-Method-Override': 'GET'
}
requestbody = {
"attrs": [ "name", "state", "last_check_result", "host_name", "display_name" ],
"joins": [ "host.name", "host.state", "host.last_check_result" ],
"filter": config['filters']['services'],
}
r = requests.get(request_url,
headers=headers,
data=json.dumps(requestbody),
auth=(config['icinga2_api']['username'], config['icinga2_api']['password']),
verify=False)
if (r.status_code == 200):
data['services'] = r.json()
else:
r.raise_for_status()
return data
def render_text_output(data): def prettify(self, text):
print("{:50s} {:10s}".format("host", "status")) for search, replace in self.config['prettify'].items():
for host in data['hosts']['results']: text = text.replace(search.upper(), replace)
print("{:50s} {}".format(host['name'], host['attrs']['state']))
for service in data['services']['results']: return text
print("{:50s} {}".format(service['name'], service['attrs']['state']))
def render_services_per_host(host, data): def render_services_per_host(self, host):
services_operational = '' services_operational = ''
services_warning = '' services_warning = ''
services_critical = '' services_critical = ''
card_header = '' card_header = ''
services_template = """ services_template = """
<li class="list-group-item d-flex justify-content-between align-items-center"> <li class="list-group-item d-flex justify-content-between align-items-center">
{} {}
<span class="badge badge-{}">{}</span> <span class="badge badge-{}">{}</span>
</li> </li>
""" """
services_hostname_template = """ services_hostname_template = """
<div id="{0}" class="card-header d-flex justify-content-between align-items-center"> <div id="{0}" class="card-header d-flex justify-content-between align-items-center">
<h4><a href="#{0}">{0}</a></h4> <h4><a href="#{0}">{1}</a></h4>
<span class="badge badge-{1}">{2}</span> <span class="badge badge-{2}">{3}</span>
</div>""" </div>"""
for service in sorted(data['services']['results'], key=lambda x: x['attrs']['display_name']): for service in sorted(self.services['results'], key=lambda x: x['attrs']['display_name']):
if service['attrs']['host_name'] == host: if service['attrs']['host_name'] == host:
if service['attrs']['state'] == 0: if service['attrs']['state'] == 0:
services_operational = services_operational + services_template.format(service['attrs']['display_name'], 'success', 'OK') services_operational = services_operational + services_template.format(self.prettify(service['attrs']['display_name']), 'success', 'OK')
elif service['attrs']['state'] == 1: elif service['attrs']['state'] == 1:
services_warning = services_warning + services_template.format(service['attrs']['display_name'], 'warning', 'WARNING') services_warning = services_warning + services_template.format(self.prettify(service['attrs']['display_name']), 'warning', 'WARNING')
else: else:
services_critical = services_critical + services_template.format(service['attrs']['display_name'], 'danger', 'CRITICAL') services_critical = services_critical + services_template.format(self.prettify(service['attrs']['display_name']), 'danger', 'CRITICAL')
if service['joins']['host']['state'] == 0: if service['joins']['host']['state'] == 0:
card_header = services_hostname_template.format(host, 'success', 'UP') card_header = services_hostname_template.format(host, service['joins']['host']['vars']['pretty_name'], 'success', 'UP')
else: else:
card_header = services_hostname_template.format(host, 'danger', 'DOWN') card_header = services_hostname_template.format(host, service['joins']['host']['vars']['pretty_name'], 'danger', 'DOWN')
with open("services_template.html", "r") as f: with open("services_template.html", "r") as f:
htmlTemplate = f.read() htmlTemplate = f.read()
htmlOutput = htmlTemplate.format( htmlOutput = htmlTemplate.format(
card_header = card_header, card_header = card_header,
services_operational = services_operational, services_operational = services_operational,
services_warning = services_warning, services_warning = services_warning,
services_critical = services_critical services_critical = services_critical
) )
return htmlOutput return htmlOutput
def render_service_details(data): def render_service_details(self):
# generate list of hosts by scanning services for unique host_name # generate list of hosts by scanning services for unique host_name
host_names = [] host_names = set()
for service in data['services']['results']: for service in self.services['results']:
if service['attrs']['host_name'] not in host_names: host_names.add(service['attrs']['host_name'])
host_names.append(service['attrs']['host_name']) # render html for each host_name
# render html for each host_name html_output = ""
html_output = "" for host in sorted(host_names):
for host in sorted(host_names): html_output = html_output + self.render_services_per_host(host)
html_output = html_output + render_services_per_host(host, data) return html_output
return html_output
def render_index_html(filename, service_details): def render_index_html(self, service_details):
with open("template.html", "r") as f: with open("template.html", "r") as f:
htmlTemplate = f.read() htmlTemplate = f.read()
htmlOutput = htmlTemplate.format( htmlOutput = htmlTemplate.format(
services = service_details services = service_details
) )
with open(filename, "w") as f: with open(self.config['output']['filename'], "w") as f:
f.write(htmlOutput) f.write(htmlOutput)
def main(): def __init__(self):
config = configparser.ConfigParser() self.config = configparser.ConfigParser()
config['icinga2_api'] = { self.config['icinga2_api'] = {
'baseurl': 'https://localhost:5665', 'baseurl': 'https://localhost:5665',
'username': 'root', 'username': 'root',
'password': 'foobar' 'password': 'foobar'
} }
with open('config.conf', 'r') as configfile: with open('config.conf', 'r') as configfile:
config.read('config.conf') self.config.read('config.conf')
data = do_api_calls(config)
service_details = render_service_details(data)
render_index_html(config['output']['filename'], service_details)
if __name__ == "__main__": if __name__ == "__main__":
main() page = StatusPage()
page.do_api_calls()
service_details = page.render_service_details()
page.render_index_html(service_details)