From 5d5aea3cf42a47419c3af8b0187cbb36ec46280b Mon Sep 17 00:00:00 2001 From: Jared Vititoe Date: Fri, 7 Feb 2025 23:06:49 -0500 Subject: [PATCH] acquire site id --- app.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/app.py b/app.py index 16f5ba4..2a00b2a 100644 --- a/app.py +++ b/app.py @@ -47,13 +47,20 @@ def send_webhook(device, status, diagnostics): class UnifiAPI: def __init__(self, config): self.base_url = config['unifi']['controller'] - self.site_id = "default" self.session = requests.Session() self.session.verify = False self.headers = { 'X-API-KEY': config['unifi']['api_key'], 'Accept': 'application/json' } + self.site_id = self.get_site_id() + + def get_site_id(self): + url = f"{self.base_url}/proxy/network/api/self/sites" + response = self.session.get(url, headers=self.headers) + response.raise_for_status() + sites = response.json()['data'] + return sites[0]['_id'] def get_all_devices(self): url = f"{self.base_url}/integration/v1/sites/{self.site_id}/devices"