d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix attribute error

Browse source
This commit is contained in:
Nicholas St. Germain 2019-04-24 11:35:19 -05:00
parent 737cc11457
commit 7228fa2a39
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
varken/unifi.py
View file

@ -9,6 +9,7 @@ class UniFiAPI(object):
def __init__(self, server, dbmanager): def __init__(self, server, dbmanager):
self.dbmanager = dbmanager self.dbmanager = dbmanager
self.server = server self.server = server
self.site = self.server.site
# Create session to reduce server web thread load, and globally define pageSize for all requests # Create session to reduce server web thread load, and globally define pageSize for all requests
self.session = Session() self.session = Session()
self.logger = getLogger() self.logger = getLogger()
@ -43,13 +44,13 @@ class UniFiAPI(object):
site = [site['name'] for site in get['data'] if site['name'].lower() == self.server.site.lower() site = [site['name'] for site in get['data'] if site['name'].lower() == self.server.site.lower()
or site['desc'].lower() == self.server.site.lower()] or site['desc'].lower() == self.server.site.lower()]
if site: if site:
self.server.site = site[0] self.site = site[0]
else: else:
self.logger.error(f"Could not map site {self.server.site} to a site id/alias") self.logger.error(f"Could not map site {self.server.site} to a site id/alias")
def get_usg_stats(self): def get_usg_stats(self):
now = datetime.now(timezone.utc).astimezone().isoformat() now = datetime.now(timezone.utc).astimezone().isoformat()
endpoint = f'/api/s/{self.server.site}/stat/device' endpoint = f'/api/s/{self.site}/stat/device'
req = self.session.prepare_request(Request('GET', self.server.url + endpoint)) req = self.session.prepare_request(Request('GET', self.server.url + endpoint))
get = connection_handler(self.session, req, self.server.verify_ssl) get = connection_handler(self.session, req, self.server.verify_ssl)