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"
def get_devices(self):
url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device"
response = self.session.get(url, headers=self.headers)
response.raise_for_status()
devices = response.json()
return [{
'name': device['name'],
'ip': device['ip'],
'type': device['type'],
'connection_type': 'fiber' if device.get('uplink', {}).get('media') == 'sfp' else 'copper',
'critical': True if device['type'] in ['udm-pro', 'switch'] else False,
'device_id': device['mac']
} for device in devices]
try:
url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device"
response = self.session.get(url, headers=self.headers)
response.raise_for_status()
devices = response.json()
return [{
'name': device['name'],
'ip': device['ip'],
'type': device['type'],
'connection_type': 'fiber' if device.get('uplink', {}).get('media') == 'sfp' else 'copper',
'critical': True if device['type'] in ['udm-pro', 'switch'] else False,
'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):
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.raise_for_status()
return response.json()
def get_device_diagnostics(self, device):
details = self.get_device_details(device['device_id'])
if not details: