LotusGuild/gandalf
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix syntax error

Browse Source
This commit is contained in:
Jared Vititoe
2025-02-08 00:04:42 -05:00
parent b67a5d10c2
commit dc117b276e
1 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
app.py
View File

@ -56,25 +56,28 @@ class UnifiAPI:
self.site_id = "default" self.site_id = "default"
def get_devices(self): def get_devices(self):
url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device" try:
response = self.session.get(url, headers=self.headers) url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device"
response.raise_for_status() response = self.session.get(url, headers=self.headers)
devices = response.json() response.raise_for_status()
return [{ devices = response.json()
'name': device['name'], return [{
'ip': device['ip'], 'name': device['name'],
'type': device['type'], 'ip': device['ip'],
'connection_type': 'fiber' if device.get('uplink', {}).get('media') == 'sfp' else 'copper', 'type': device['type'],
'critical': True if device['type'] in ['udm-pro', 'switch'] else False, 'connection_type': 'fiber' if device.get('uplink', {}).get('media') == 'sfp' else 'copper',
'device_id': device['mac'] 'critical': True if device['type'] in ['udm-pro', 'switch'] else False,
} for device in devices] 'device_id': device['mac']
} for device in devices]
except Exception as e:
logger.error(f"Error fetching devices: {e}")
return []
def get_device_details(self, device_id): def get_device_details(self, device_id):
url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device/{device_id}" url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device/{device_id}"
response = self.session.get(url, headers=self.headers) response = self.session.get(url, headers=self.headers)
response.raise_for_status() response.raise_for_status()
return response.json() return response.json()
def get_device_diagnostics(self, device): def get_device_diagnostics(self, device):
details = self.get_device_details(device['device_id']) details = self.get_device_details(device['device_id'])
if not details: if not details: