v/wideq
1
0
Fork 0
mirror of https://github.com/no2chem/wideq.git synced 2025-05-19 00:30:10 -07:00
Code Issues Projects Releases Wiki Activity

Wrap devices in lists (fix #1, fix #2)

Browse Source
This commit is contained in:
Adrian Sampson 2018-03-11 09:59:17 -04:00
parent dd9ff96b39
commit 2dc014f17f
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
wideq.py
View File

@ -41,6 +41,19 @@ def oauth2_signature(message, secret):
return base64.b64encode(digest) return base64.b64encode(digest)
def as_list(obj):
"""Wrap non-lists in lists.
If `obj` is a list, return it unchanged. Otherwise, return a
single-element list containing it.
"""
if isinstance(obj, list):
return obj
else:
return [obj]
class APIError(Exception): class APIError(Exception):
"""An error reported by the API.""" """An error reported by the API."""
@ -226,12 +239,12 @@ class Auth(object):
def start_session(self): def start_session(self):
"""Start an API session for the logged-in user. Return the """Start an API session for the logged-in user. Return the
Session object and the user's devices. Session object and a list of the user's devices.
""" """
session_info = login(self.gateway.api_root, self.access_token) session_info = login(self.gateway.api_root, self.access_token)
session_id = session_info['jsessionId'] session_id = session_info['jsessionId']
return Session(self, session_id), session_info['item'] return Session(self, session_id), as_list(session_info['item'])
def refresh(self): def refresh(self):
"""Refresh the authentication, returning a new Auth object. """Refresh the authentication, returning a new Auth object.
@ -263,7 +276,7 @@ class Session(object):
Return a list of dicts with information about the devices. Return a list of dicts with information about the devices.
""" """
return self.post('device/deviceList')['item'] return as_list(self.post('device/deviceList')['item'])
def monitor_start(self, device_id): def monitor_start(self, device_id):
"""Begin monitoring a device's status. """Begin monitoring a device's status.