This commit is contained in:
2025-02-07 22:27:26 -05:00
parent d8ede8f264
commit de5efc15cb

25
app.py
View File

@ -47,37 +47,36 @@ def send_webhook(device, status, diagnostics):
class UnifiAPI:
def __init__(self, config):
self.base_url = config['unifi']['controller']
self.api_key = config['unifi']['api_key']
self.headers = {
'X-API-KEY': config['unifi']['api_key'],
'Accept': 'application/json'
}
self.site_id = config['unifi']['site_id']
self.session = requests.Session()
self.session.verify = False
self.headers = {
'Authorization': f'Bearer {self.api_key}',
'Accept': 'application/json'
}
def get_all_devices(self):
url = f"{self.base_url}/proxy/network/api/s/{self.site_id}/stat/device"
url = f"{self.base_url}/proxy/network/integration/v1/sites/{self.site_id}/devices"
response = self.session.get(url, headers=self.headers)
response.raise_for_status()
return response.json()['data']
def get_device_details(self, device_id):
url = f"{self.base_url}/proxy/network/api/s/{self.site_id}/stat/device/{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()
data = response.json()['data'][0]
data = response.json()
return {
'state': data['state'],
'firmware': {
'version': data['version'],
'updatable': data.get('upgradable', False)
'version': data['firmwareVersion'],
'updatable': data.get('firmwareUpdatable', False)
},
'network': {
'ip': data['ip'],
'mac': data['mac']
'ip': data['ipAddress'],
'mac': data['macAddress']
},
'interfaces': self._parse_interfaces(data.get('port_table', {}))
'interfaces': self._parse_interfaces(data.get('interfaces', {}))
}
def get_device_diagnostics(self, device):