Handle out of process token refreshes

kivy_mainline_and_py3.8
Tyler Goodlet 2018-02-15 16:40:33 -05:00
parent a4fe86719f
commit ff437ce9e1
1 changed files with 12 additions and 5 deletions

View File

@ -60,9 +60,10 @@ class Client:
self._conf = config
self.access_data = {}
self.user_data = {}
self._apply_config(config)
self._reload_config(config)
def _apply_config(self, config):
def _reload_config(self, config=None):
self._conf = config or get_config()
self.access_data = dict(self._conf['questrade'])
async def _new_auth_token(self) -> dict:
@ -130,8 +131,7 @@ class Client:
raise QuestradeError("API is down for maintenance")
elif qterr.args[0].decode() == 'Bad Request':
# likely config ``refresh_token`` is expired
_token_from_user(self._conf)
self._apply_config(self._conf)
self._reload_config()
data = await self._new_auth_token()
else:
raise qterr
@ -313,7 +313,14 @@ async def poll_tickers(
_cache = {}
while True: # use an event here to trigger exit?
try:
quotes_resp = await client.api.quotes(ids=ids)
except QuestradeError as qterr:
if "Access token is invalid" in str(qterr.args[0]):
# out-of-process piker may have renewed already
client._reload_config()
quotes_resp = await client.api.quotes(ids=ids)
start = time.time()
quotes = quotes_resp['quotes']
# log.trace(quotes)