Compare commits

..

No commits in common. "3ab0a05b59e3be70a4f1f3f9759139663f435806" and "6cde8347a743a33b00ea33289d7d99c0a0147fbd" have entirely different histories.

3 changed files with 25 additions and 46 deletions

View file

@ -1,12 +0,0 @@
root = true
[*]
indent_style = space
indent_size = 4
end_of_line = lf
charset = utf-8
trim_trailing_whitespace = true
insert_final_newline = true
[*.tolm]
indent_size = 2

View file

@ -1 +0,0 @@
tomlkit

View file

@ -1,41 +1,35 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import json
import requests import requests
import tomlkit import tomlkit
import urllib3 import urllib3
urllib3.disable_warnings() urllib3.disable_warnings()
class StatusPage: class StatusPage:
def get_api_result(self): def do_api_calls(self):
if self.services: #services
return self.services request_url = "{}/v1/objects/services".format(self.config['icinga2_api']['baseurl'])
headers = { headers = {
'Accept': 'application/json', 'Accept': 'application/json',
'X-HTTP-Method-Override': 'GET' 'X-HTTP-Method-Override': 'GET'
} }
requestbody = { requestbody = {
"attrs": [ "name", "state", "last_check_result", "host_name", "display_name" ], "attrs": [ "name", "state", "last_check_result", "host_name", "display_name" ],
"joins": [ "host.name", "host.state", "host.last_check_result", "host.vars" ], "joins": [ "host.name", "host.state", "host.last_check_result", "host.vars" ],
"filter": self.config['filters']['services'], "filter": self.config['filters']['services'],
} }
r = requests.get(request_url,
r = requests.get(
'{}/v1/objects/services'.format(self.config['icinga2_api']['baseurl']),
headers=headers, headers=headers,
json=requestbody, data=json.dumps(requestbody),
auth=(self.config['icinga2_api']['username'], self.config['icinga2_api']['password']), auth=(self.config['icinga2_api']['username'], self.config['icinga2_api']['password']),
verify=False verify=False)
)
if (r.status_code == 200): if (r.status_code == 200):
self.services = r.json()['results'] self.services = r.json()
else: else:
r.raise_for_status() r.raise_for_status()
return self.services
def prettify(self, text): def prettify(self, text):
for search, replace in self.config.get('prettify', {}).items(): for search, replace in self.config.get('prettify', {}).items():
@ -66,31 +60,31 @@ class StatusPage:
<span class="badge badge-{2}">{3}</span> <span class="badge badge-{2}">{3}</span>
</div>""" </div>"""
for service in sorted(self.get_api_result(), 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:
state = int(service['attrs']['state']) state = int(service['attrs']['state'])
if state in (1, 2): if state in (1, 2):
self.ragecounter += state self.ragecounter += state
services.append(services_template.format( services.append(services_template.format(
self.prettify(service['attrs']['display_name']), self.prettify(service['attrs']['display_name']),
state_to_design_mapping[state][0], state_to_design_mapping[state][0],
state_to_design_mapping[state][1], state_to_design_mapping[state][1],
)) ))
if not card_header: if not card_header:
if service['joins']['host']['state'] == 0: if service['joins']['host']['state'] == 0:
card_header = services_hostname_template.format( card_header = services_hostname_template.format(
host, host,
service['joins']['host']['vars']['pretty_name'], service['joins']['host']['vars']['pretty_name'],
'success', 'success',
'UP', 'UP',
) )
else: else:
card_header = services_hostname_template.format( card_header = services_hostname_template.format(
host, host,
service['joins']['host']['vars']['pretty_name'], service['joins']['host']['vars']['pretty_name'],
'danger', 'danger',
'DOWN', 'DOWN',
) )
self.ragecounter += 100 self.ragecounter += 100
@ -107,15 +101,13 @@ class StatusPage:
def render_service_details(self): 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 self.get_api_result(): for service in self.services['results']:
host_names[service['joins']['host']['vars']['pretty_name']] = service['attrs']['host_name'] host_names.add(service['attrs']['host_name'])
# render html for each host_name # render html for each host_name
html_output = [] html_output = []
# Can't use .values() here, since we want to sort by prettyname for host in sorted(host_names):
for prettyname, hostname in sorted(host_names.items()): html_output.append(self.render_services_per_host(host))
html_output.append(self.render_services_per_host(hostname))
return ''.join(html_output) return ''.join(html_output)
@ -140,10 +132,10 @@ class StatusPage:
def __init__(self): def __init__(self):
self.config = tomlkit.loads(open('config.toml').read()) self.config = tomlkit.loads(open('config.toml').read())
self.services = {}
self.ragecounter = 0 self.ragecounter = 0
if __name__ == "__main__": if __name__ == "__main__":
page = StatusPage() page = StatusPage()
service_details = page.render_service_details() page.do_api_calls()
service_details = page.render_service_details()
page.render_index_html(service_details) page.render_index_html(service_details)