diff --git a/app.py b/app.py index a6db34c..ba56c7c 100644 --- a/app.py +++ b/app.py @@ -50,15 +50,10 @@ class UnifiAPI: self.session = requests.Session() self.session.verify = False self.headers = { - 'X-API-KEY': config['unifi']['api_key'], + 'Authorization': f'Bearer {config["unifi"]["api_key"]}', 'Accept': 'application/json', 'Content-Type': 'application/json' } - # First authenticate - auth_url = f"{self.base_url}/api/auth/login" - auth_data = {'username': 'admin', 'apiKey': config['unifi']['api_key']} - self.session.post(auth_url, json=auth_data, headers=self.headers) - self.site_id = self.get_site_id() def get_site_id(self): @@ -72,8 +67,7 @@ class UnifiAPI: url = f"{self.base_url}/proxy/network/api/s/{self.site_id}/stat/device" response = self.session.get(url, headers=self.headers) response.raise_for_status() - return response.json()['data'] - def get_device_details(self, device_id): + return response.json()['data'] def get_device_details(self, device_id): url = f"{self.base_url}/proxy/network/integration/v1/sites/{self.site_id}/devices/{device_id}" response = self.session.get(url, headers=self.headers) response.raise_for_status()