fix: when trying to use an expired refresh token, throw a ManualLoginRequired exception, instead of throwing a random KeyNotFound exception

gboudreau · gboudreau · commit a562ff65e804 · 2025-10-04T17:16:47.000-04:00
diff --git a/ws_api/wealthsimple_api.py b/ws_api/wealthsimple_api.py
@@ -151,6 +151,8 @@ def check_oauth_token(self, persist_session_fct: Optional[Callable[[WSAPISession
                 'x-ws-profile': 'invest'
             }
             response = self.send_post(f"{self.OAUTH_BASE_URL}/token", data, headers)
+            if not 'access_token' in response or not 'refresh_token' in response:
+                raise ManualLoginRequired(f"OAuth token invalid and cannot be refreshed: {response['error'] if 'error' in response else 'Invalid response from API'}")
             self.session.access_token = response['access_token']
             self.session.refresh_token = response['refresh_token']
             if persist_session_fct:

Original file line number	Diff line number	Diff line change
`@@ -151,6 +151,8 @@ def check_oauth_token(self, persist_session_fct: Optional[Callable[[WSAPISession`
`151`	`151`	`'x-ws-profile': 'invest'`
`152`	`152`	`}`
`153`	`153`	`response = self.send_post(f"{self.OAUTH_BASE_URL}/token", data, headers)`
	`154`	`+ if not 'access_token' in response or not 'refresh_token' in response:`
	`155`	`+ raise ManualLoginRequired(f"OAuth token invalid and cannot be refreshed: {response['error'] if 'error' in response else 'Invalid response from API'}")`
`154`	`156`	`self.session.access_token = response['access_token']`
`155`	`157`	`self.session.refresh_token = response['refresh_token']`
`156`	`158`	`if persist_session_fct:`