v/wideq
1
0
Fork 0
mirror of https://github.com/no2chem/wideq.git synced 2025-05-16 15:20:09 -07:00
Code Issues Projects Releases Wiki Activity

Recreate sessions on the fly

Browse Source 
Adapts #88 to avoid a global session object.
This commit is contained in:
Adrian Sampson 2020-04-10 21:43:52 -04:00
parent 6af5eb1ce1
commit fd5a695988
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
wideq/core.py
View File

@ -65,7 +65,7 @@ def get_wideq_logger() -> logging.Logger:
LOGGER = get_wideq_logger() LOGGER = get_wideq_logger()
def get_retry_session(): def retry_session():
"""Get a Requests session that retries HTTP and HTTPS requests. """Get a Requests session that retries HTTP and HTTPS requests.
""" """
# Adapted from: # Adapted from:
@ -84,9 +84,6 @@ def get_retry_session():
return session return session
SESSION = get_retry_session()
def set_log_level(level: int): def set_log_level(level: int):
logger = get_wideq_logger() logger = get_wideq_logger()
logger.setLevel(level) logger.setLevel(level)
@ -201,7 +198,8 @@ def lgedm_post(url, data=None, access_token=None, session_id=None):
if session_id: if session_id:
headers['x-thinq-jsessionId'] = session_id headers['x-thinq-jsessionId'] = session_id
res = SESSION.post(url, json={DATA_ROOT: data}, headers=headers) with retry_session() as session:
res = session.post(url, json={DATA_ROOT: data}, headers=headers)
out = res.json()[DATA_ROOT] out = res.json()[DATA_ROOT]
# Check for API errors. # Check for API errors.
@ -293,7 +291,8 @@ def refresh_auth(oauth_root, refresh_token):
'Accept': 'application/json', 'Accept': 'application/json',
} }
res = SESSION.post(token_url, data=data, headers=headers) with retry_session() as session:
res = session.post(token_url, data=data, headers=headers)
res_data = res.json() res_data = res.json()
if res_data['status'] != 1: if res_data['status'] != 1: