From 4da30bdedaf9145f0e286edb6c95b2955a50c60e Mon Sep 17 00:00:00 2001
From: Adam Goldsmith <adam@adamgoldsmith.name>
Date: Thu, 29 Sep 2022 17:38:59 -0400
Subject: [PATCH] Initial Commit, just checking batteries right now

---
 check_home_assistant.py | 89 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100755 check_home_assistant.py

diff --git a/check_home_assistant.py b/check_home_assistant.py
new file mode 100755
index 0000000..a240fd4
--- /dev/null
+++ b/check_home_assistant.py
@@ -0,0 +1,89 @@
+#!/usr/bin/env python3
+
+import argparse
+import logging
+import sys
+from urllib.parse import urljoin
+
+import requests
+import nagiosplugin
+
+_log = logging.getLogger("nagiosplugin")
+
+
+class Batteries(nagiosplugin.Resource):
+    def __init__(self, url: str, token: str):
+        self.url = url
+        self.token = token
+
+    def hass_get(self, endpoint: str) -> requests.Response:
+        headers = {
+            "Authorization": "Bearer " + self.token,
+            "Content-Type": "application/json",
+        }
+
+        r = requests.get(urljoin(self.url, endpoint), headers=headers)
+        if not r.ok:
+            raise Exception("Failed to query Home Assistant API: " + r.text)
+
+        return r.json()
+
+    def check_api(self):
+        message = self.hass_get("/api/").get("message")
+        if message != "API running.":
+            print("ERROR: " + message)
+            sys.exit(1)
+        else:
+            print("OK: " + message)
+
+    def probe(self):
+        response = self.hass_get("/api/states")
+        for state in response:
+            if state["attributes"].get("device_class") == "battery":
+                if state["state"] == "unavailable":
+                    _log.info(f"{state['entity_id']} unavailable")
+                yield nagiosplugin.Metric(
+                    state["entity_id"],
+                    float(state["state"]) if state["state"].isnumeric() else -1,
+                    uom=state["attributes"].get("unit_of_measurement"),
+                    min=0,
+                    max=100,
+                    context="battery",
+                )
+
+
+@nagiosplugin.guarded
+def main():
+    argp = argparse.ArgumentParser(description=__doc__)
+    argp.add_argument(
+        "-w",
+        "--warning",
+        metavar="RANGE",
+        default="",
+        help="return warning if battery percentage is outside RANGE",
+    )
+    argp.add_argument(
+        "-c",
+        "--critical",
+        metavar="RANGE",
+        default="",
+        help="return critical if battery percentage is outside RANGE",
+    )
+    argp.add_argument(
+        "-t", "--token", required=True, type=str, help="API token for Home Assistant"
+    )
+    argp.add_argument(
+        "-u", "--url", required=True, type=str, help="URL for Home Assistant"
+    )
+    argp.add_argument("-v", "--verbose", action="count", default=0)
+    args = argp.parse_args()
+
+    check = nagiosplugin.Check(
+        Batteries(args.url, args.token),
+        nagiosplugin.ScalarContext("battery", args.warning, args.critical),
+    )
+    check.main(args.verbose)
+
+
+if __name__ == "__main__":
+    main()