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

Obtain refresh token from OAuth

Browse Source
This commit is contained in:
Adrian Sampson 2018-01-20 13:53:19 -05:00
parent 29eb1804c5
commit 8fc2749a9b
2 changed files with 36 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
example.py
View File

@ -31,8 +31,8 @@ def authenticate(gateway):
print(login_url) print(login_url)
print('Then paste the URL where the browser is redirected:') print('Then paste the URL where the browser is redirected:')
callback_url = input() callback_url = input()
access_token = wideq.parse_oauth_callback(callback_url) access_token, refresh_token = wideq.parse_oauth_callback(callback_url)
return wideq.Auth(gateway, access_token) return wideq.Auth(gateway, access_token, refresh_token)
def get_session(state, reauth=False): def get_session(state, reauth=False):

48
wideq.py
View File

@ -82,12 +82,12 @@ def gateway_info():
) )
def oauth_url(oauth_base): def oauth_url(auth_base):
"""Construct the URL for users to log in (in a browser) to start an """Construct the URL for users to log in (in a browser) to start an
authenticated session. authenticated session.
""" """
url = urljoin(oauth_base, 'login/sign_in') url = urljoin(auth_base, 'login/sign_in')
query = urlencode({ query = urlencode({
'country': COUNTRY, 'country': COUNTRY,
'language': LANGUAGE, 'language': LANGUAGE,
@ -101,12 +101,13 @@ def oauth_url(oauth_base):
def parse_oauth_callback(url): def parse_oauth_callback(url):
"""Parse the URL to which an OAuth login redirected to obtain an """Parse the URL to which an OAuth login redirected to obtain two
access token for API credentials. tokens: an access token for API credentials, and a refresh token for
getting updated access tokens.
""" """
params = parse_qs(urlparse(url).query) params = parse_qs(urlparse(url).query)
return params['access_token'][0] return params['access_token'][0], params['refresh_token'][0]
def login(api_root, access_token): def login(api_root, access_token):
@ -125,30 +126,36 @@ def login(api_root, access_token):
class Gateway(object): class Gateway(object):
def __init__(self, oauth_base, api_root): def __init__(self, auth_base, api_root, oauth_root):
self.oauth_base = oauth_base self.auth_base = auth_base
self.api_root = api_root self.api_root = api_root
self.oauth_root = oauth_root
@classmethod @classmethod
def discover(cls): def discover(cls):
gw = gateway_info() gw = gateway_info()
return cls(gw['empUri'], gw['thinqUri']) return cls(gw['empUri'], gw['thinqUri'], gw['oauthUri'])
@classmethod @classmethod
def load(cls, data): def load(cls, data):
return cls(data['oauth_base'], data['api_root']) return cls(data['auth_base'], data['api_root'], data['oauth_root'])
def dump(self): def dump(self):
return {'oauth_base': self.oauth_base, 'api_root': self.api_root} return {
'auth_base': self.auth_base,
'api_root': self.api_root,
'oauth_root': self.oauth_root,
}
def oauth_url(self): def oauth_url(self):
return oauth_url(self.oauth_base) return oauth_url(self.auth_base)
class Auth(object): class Auth(object):
def __init__(self, gateway, access_token): def __init__(self, gateway, access_token, refresh_token):
self.gateway = gateway self.gateway = gateway
self.access_token = access_token self.access_token = access_token
self.refresh_token = refresh_token
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
@ -160,11 +167,24 @@ class Auth(object):
return Session(self, session_id), session_info['item'] return Session(self, session_id), session_info['item']
def dump(self): def dump(self):
return self.access_token return {
'access_token': self.access_token,
'refresh_token': self.refresh_token,
}
@classmethod @classmethod
def load(cls, gateway, data): def load(cls, gateway, data):
return cls(gateway, data) return cls(gateway, data['access_token'], data['refresh_token'])
def refresh(self):
token_url = urljoin(self.gateway.oauth_root, '/oauth2/token')
query = urlencode({
'grant_type': 'refresh_token',
'refresh_token': self.refresh_token,
})
req_url = '{}?{}'.format(token_url, query)
requests.Request('GET', req_url)
class Session(object): class Session(object):