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

added raw back

Browse Source
This commit is contained in:
Jared Vititoe
2025-02-08 00:13:06 -05:00
parent 68beb7b1c4
commit 19224d14df
1 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
app.py
View File

@ -35,19 +35,26 @@ class UnifiAPI:
url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device" url = f"{self.base_url}/proxy/network/v2/api/site/{self.site_id}/device"
response = self.session.get(url, headers=self.headers) response = self.session.get(url, headers=self.headers)
response.raise_for_status() response.raise_for_status()
devices = response.json() devices_data = response.json()
return [{
'name': device['name'], # Add debug logging
'ip': device['ip'], logger.debug(f"Raw device data: {devices_data}")
'type': device['type'],
'connection_type': 'fiber' if device.get('uplink', {}).get('media') == 'sfp' else 'copper', # Handle the response structure correctly
'critical': True if device['type'] in ['udm-pro', 'switch'] else False, devices = []
'device_id': device['mac'] for device in devices_data:
} for device in devices] devices.append({
'name': device.get('name', 'Unknown'),
'ip': device.get('ip', '0.0.0.0'),
'type': device.get('type', 'unknown'),
'connection_type': 'fiber' if device.get('uplink', {}).get('media') == 'sfp' else 'copper',
'critical': True if device.get('type') in ['udm-pro', 'switch'] else False,
'device_id': device.get('mac')
})
return devices
except Exception as e: except Exception as e:
logger.error(f"Error fetching devices: {e}") logger.error(f"Error fetching devices: {e}")
return [] return []
def get_device_details(self, device_id): def get_device_details(self, device_id):
try: try:
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}"