check_home_assistant_state.py

@@ -1,11 +1,9 @@
 #!/usr/bin/env python3
 
 import argparse
-import dataclasses
 import logging
 import sys
 from urllib.parse import urljoin
-from typing import Optional
 
 import requests
 import nagiosplugin
@@ -13,15 +11,10 @@ import nagiosplugin
 _log = logging.getLogger("nagiosplugin")
 
 
-@dataclasses.dataclass
+class Batteries(nagiosplugin.Resource):
-class Entities(nagiosplugin.Resource):
+    def __init__(self, url: str, token: str):
-    url: str
+        self.url = url
-    token: str
+        self.token = token
-    device_class: str
-    min: float
-    max: float
-    filters: list[str]
-    attribute: Optional[str]
 
     def hass_get(self, endpoint: str) -> requests.Response:
         headers = {
@@ -46,36 +39,36 @@ class Entities(nagiosplugin.Resource):
     def probe(self):
         response = self.hass_get("/api/states")
         for state in response:
-            if state["attributes"].get("device_class") == self.device_class and all(
+            if state["attributes"].get("device_class") == "battery":
-                state["attributes"].get(k) == v for k, v in self.filters
-            ):
-                if self.attribute is not None:
-                    value = state["attributes"].get(self.attribute, -1)
-                    uom = None
-                else:
-                    value = float(state["state"]) if state["state"].isnumeric() else -1
-                    uom = state["attributes"].get("unit_of_measurement")
-
                 if state["state"] == "unavailable":
                     _log.info(f"{state['entity_id']} unavailable")
                 yield nagiosplugin.Metric(
                     state["entity_id"],
-                    value,
+                    float(state["state"]) if state["state"].isnumeric() else -1,
-                    uom,
+                    uom=state["attributes"].get("unit_of_measurement"),
-                    context=self.device_class,
+                    min=0,
-                    min=self.min,
+                    max=100,
-                    max=self.max,
+                    context="battery",
                 )
 
 
-def key_value(arg: str):
-    k, _, v = arg.partition("=")
-    return k, v
-
-
 @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"
     )
@@ -83,63 +76,11 @@ def main():
         "-u", "--url", required=True, type=str, help="URL for Home Assistant"
     )
     argp.add_argument("-v", "--verbose", action="count", default=0)
-
-    argp.add_argument(
-        "-d",
-        "--device-class",
-        type=str,
-        required=True,
-        help="A device class",
-    )
-    argp.add_argument(
-        "-w",
-        "--warning",
-        metavar="RANGE",
-        required=True,
-        help="return warning if battery percentage is outside RANGE",
-    )
-    argp.add_argument(
-        "-c",
-        "--critical",
-        metavar="RANGE",
-        required=True,
-        help="return critical if battery percentage is outside RANGE",
-    )
-    argp.add_argument(
-        "--min",
-        type=float,
-        help="Min for performance data",
-    )
-    argp.add_argument(
-        "--max",
-        type=float,
-        help="Max for performance data",
-    )
-    argp.add_argument(
-        "-a", "--attribute", type=str, help="Check attribute instead of value"
-    )
-    argp.add_argument(
-        "-f",
-        "--filter",
-        type=key_value,
-        default=[],
-        nargs="*",
-        help="Filter by 'attribute=value'",
-    )
-
     args = argp.parse_args()
 
     check = nagiosplugin.Check(
-        Entities(
+        Batteries(args.url, args.token),
-            args.url,
+        nagiosplugin.ScalarContext("battery", args.warning, args.critical),
-            args.token,
-            args.device_class,
-            args.min,
-            args.max,
-            args.filter,
-            args.attribute,
-        ),
-        nagiosplugin.ScalarContext(args.device_class, args.warning, args.critical),
     )
     check.main(args.verbose)