add service-only mode

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 = """
+                <div id="{0}" class="card-header d-flex justify-content-between align-items-center">
+                    <h4><a href="#{0}">{1}</a></h4>
+                    <span class="badge badge-{2}">{3}</span>
+                </div>"""
+
+        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)