diff --git a/error.html b/error.html deleted file mode 100644 index df90698..0000000 --- a/error.html +++ /dev/null @@ -1,41 +0,0 @@ - - - - - ${title} - - - - - -
- -
-
-
-
-

Something went wrong

-
-
-

- There was an error rendering the status page. - Admins have been notified. -

-
-
-
-
-
- - - diff --git a/requirements.txt b/requirements.txt index ea09288..618cd8d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,2 @@ Mako tomlkit -requests diff --git a/service.py b/service.py index d81b7d3..b0fc09e 100755 --- a/service.py +++ b/service.py @@ -1,39 +1,29 @@ #!/usr/bin/env python3 -import logging -import shutil -import sys -from datetime import datetime +import requests +import urllib3 from os import environ -import requests import tomlkit -import urllib3 from mako.template import Template urllib3.disable_warnings() CONFIGFILE = environ.get('STATUSPAGE_CONFIG', 'config.toml') - class StatusPage: def get_api_result(self): if self.services: - log.debug('services already exist, returning early') - return self.services - headers = {'Accept': 'application/json', 'X-HTTP-Method-Override': 'GET'} + headers = { + 'Accept': 'application/json', + 'X-HTTP-Method-Override': 'GET' + } requestbody = { - "attrs": [ - "name", - "state", - "last_check_result", - "host_name", - "display_name", - ], - "joins": ["host", "host.state", "host.last_check_result", "host.vars"], + "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'], } @@ -41,31 +31,25 @@ class StatusPage: '{}/v1/objects/services'.format(self.config['icinga2_api']['baseurl']), headers=headers, json=requestbody, - auth=( - self.config['icinga2_api']['username'], - self.config['icinga2_api']['password'], - ), - verify=False, + auth=(self.config['icinga2_api']['username'], self.config['icinga2_api']['password']), + verify=False ) - self.logger.info(f'got http status code {r.status_code}') - self.logger.debug(r.text) - - if r.status_code == 200: + if (r.status_code == 200): self.services = r.json()['results'] else: r.raise_for_status() - self.logger.info(f'got {len(self.services)} services from api') - return self.services + def prettify(self, text): for search, replace in self.config.get('prettify', {}).items(): text = text.replace(search, replace) return text + def get_services_per_host(self): state_to_design_mapping = [ ('success', 'OK'), @@ -76,11 +60,6 @@ class StatusPage: result = {} for service in self.get_api_result(): - self.logger.info( - f'now processing {service["attrs"]["host_name"]} "{service["attrs"]["display_name"]}"' - ) - self.logger.debug(service) - host = service['joins']['host']['vars']['pretty_name'] if host not in result: @@ -101,16 +80,14 @@ class StatusPage: if state in (1, 2): self.ragecounter += state - result[host]['services'][ - self.prettify(service['attrs']['display_name']) - ] = { + result[host]['services'][self.prettify(service['attrs']['display_name'])] = { 'badge': state_to_design_mapping[state][0], 'state': state_to_design_mapping[state][1], } - self.logger.info(f'ragecounter is now {self.ragecounter}') return result + def render_html(self, service_details): if self.ragecounter == 0: mood = '🆗' @@ -119,45 +96,27 @@ class StatusPage: else: mood = '🔥' - self.logger.info('rendering output html') - - start = datetime.now() - template = Template( - filename=self.config['output'].get('template', 'template.html') - ) + template = Template(filename=self.config['output'].get('template', 'template.html')) output = template.render( title=self.config['output'].get('page_title', 'Status Page'), mood=mood, hosts=service_details, ) - end = datetime.now() - - self.logger.info(f'rendered in {(end-start).total_seconds():.09f}s') with open(self.config['output']['filename'], 'w') as f: f.write(output) + def __init__(self): self.config = tomlkit.loads(open(CONFIGFILE).read()) self.services = {} self.ragecounter = 0 - self.logger = logging.getLogger('StatusPage') - handler = logging.StreamHandler(sys.stdout) - formatter = logging.Formatter( - '%(levelname)s {%(filename)s:%(lineno)d} %(message)s' - ) - handler.setFormatter(formatter) - self.logger.addHandler(handler) - self.logger.setLevel(self.config.get('loglevel', 'INFO')) - - if __name__ == "__main__": page = StatusPage() + service_details = page.get_services_per_host() - try: - service_details = page.get_services_per_host() - page.render_html(service_details) - except Exception as e: - shutil.copyfile('error.html', page.config['output']['filename']) - raise e + from pprint import pprint + pprint(service_details) + + page.render_html(service_details) diff --git a/template.html b/template.html index 2734803..9916ce3 100644 --- a/template.html +++ b/template.html @@ -39,10 +39,5 @@ % endfor -