From 5fdd84b5f7bfa312de2f0303c1d83295a8a1cec7 Mon Sep 17 00:00:00 2001
From: Jared Vititoe <jjvititoe1@gmail.com>
Date: Fri, 7 Feb 2025 22:29:26 -0500
Subject: [PATCH] get site id

---
 app.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/app.py b/app.py
index c79da65..431303d 100644
--- a/app.py
+++ b/app.py
@@ -51,10 +51,17 @@ class UnifiAPI:
             'X-API-KEY': config['unifi']['api_key'],
             'Accept': 'application/json'
         }
-        self.site_id = config['unifi']['site_id']
+        self.site_id = self._get_site_id()
         self.session = requests.Session()
         self.session.verify = False
 
+    def _get_site_id(self):
+        url = f"{self.base_url}/proxy/network/integration/v1/sites"
+        response = self.session.get(url, headers=self.headers)
+        response.raise_for_status()
+        sites = response.json()['data']
+        return sites[0]['id']  # Get first site ID
+
     def get_all_devices(self):
         url = f"{self.base_url}/proxy/network/integration/v1/sites/{self.site_id}/devices"
         response = self.session.get(url, headers=self.headers)