diff --git a/service.py b/service.py
index e5b424c..7ad4af8 100755
--- a/service.py
+++ b/service.py
@@ -8,6 +8,7 @@ import datetime
import sys
urllib3.disable_warnings()
+
class StatusPage:
def do_api_calls(self):
#services
@@ -19,8 +20,14 @@ class StatusPage:
requestbody = {
"attrs": [ "name", "state", "last_check_result", "host_name", "display_name" ],
"joins": [ "host.name", "host.state", "host.last_check_result", "host.vars", "host.display_name" ],
- "filter": self.config['filters']['services'],
}
+ if "filters" in self.config:
+ requestbody["filter"] = self.config['filters']['services']
+ elif "services" in self.config:
+ requestbody["services"] = self.config["services"]["list"]
+ else:
+ raise Exception("One of 'filters' and 'services' required")
+
r = requests.get(request_url,
headers=headers,
data=json.dumps(requestbody),
@@ -102,6 +109,34 @@ class StatusPage:
html_output = html_output + self.render_services_per_host(host)
return html_output
+ def render_service_only_details(self):
+ services_hostname_template = """
+
"""
+
+ with open("services_only_template.html", "r") as f:
+ htmlTemplate = f.read()
+
+ htmlOutput = ""
+
+ for service in sorted(self.services['results'], key=lambda x: self.prettify(x['attrs']['display_name']).lower()):
+ if service['attrs']['state'] == 0:
+ card_header = services_hostname_template.format(service['attrs']['name'], self.prettify(service['attrs']['display_name']), 'success', 'OK')
+ elif service['attrs']['state'] == 1:
+ card_header = services_hostname_template.format(service['attrs']['name'], self.prettify(service['attrs']['display_name']), 'warning', 'WARNING')
+ else:
+ card_header = services_hostname_template.format(service['attrs']['name'], self.prettify(service['attrs']['display_name']), 'danger', 'CRITICAL')
+
+ htmlOutput += htmlTemplate.format(
+ card_header = card_header,
+ services_operational = "",
+ services_warning = "",
+ services_critical = ""
+ )
+ return htmlOutput
+
def render_index_html(self, service_details):
with open("template.html", "r") as f:
@@ -131,5 +166,8 @@ class StatusPage:
if __name__ == "__main__":
page = StatusPage()
page.do_api_calls()
- service_details = page.render_service_details()
+ if page.config['output']['services_only']:
+ service_details = page.render_service_only_details()
+ else:
+ service_details = page.render_service_details()
page.render_index_html(service_details)
diff --git a/services_only_template.html b/services_only_template.html
new file mode 100644
index 0000000..4ced4a1
--- /dev/null
+++ b/services_only_template.html
@@ -0,0 +1,7 @@
+