Initial Commit, just checking batteries right now
This commit is contained in:
commit
4da30bdeda
89
check_home_assistant.py
Executable file
89
check_home_assistant.py
Executable file
@ -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()
|
Loading…
Reference in New Issue
Block a user