1
0
mirror of https://github.com/no2chem/wideq.git synced 2025-05-18 08:10:17 -07:00
wideq/example.py
2018-01-08 11:44:08 -08:00

83 lines
1.8 KiB
Python

import wideq
import json
STATE_FILE = 'wideq_state.json'
def load_state():
"""Load the current state data for this example.
"""
try:
with open(STATE_FILE) as f:
return json.load(f)
except IOError:
return {}
def save_state(state):
"""Dump the current state to disk.
"""
with open(STATE_FILE, 'w') as f:
json.dump(state, f)
def print_devices(devices):
for device in devices:
print(device['alias'])
def authenticate(gateway):
login_url = gateway.oauth_url()
print('Log in here:')
print(login_url)
print('Then paste the URL where the browser is redirected:')
callback_url = input()
access_token = wideq.parse_oauth_callback(callback_url)
return wideq.Auth(gateway, access_token)
def example():
state = load_state()
# Get the gateway, which contains the base URLs and hostnames for
# accessing the API.
if 'gateway' in state:
gateway = wideq.Gateway.load(state['gateway'])
else:
gateway = wideq.Gateway.discover()
state['gateway'] = gateway.dump()
save_state(state)
# Authenticate the user.
if 'auth' in state:
auth = wideq.Auth.load(gateway, state['auth'])
else:
auth = authenticate(gateway)
state['auth'] = auth.dump()
save_state(state)
# Start a session.
if 'session' in state:
session = wideq.Session.load(auth, state['session'])
devices = None
else:
session, devices = auth.start_session()
state['session'] = session.dump()
save_state(state)
# Request a list of devices, if we didn't get them "for free"
# already by starting the session.
if not devices:
devices = session.get_devices()
print_devices(devices)
if __name__ == '__main__':
example()