Elgato_dark/trailbase
1
0
Fork 0
mirror of https://github.com/trailbaseio/trailbase synced 2026-04-21 13:37:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Python client: log out on 401 token refresh.

Browse source
This commit is contained in:
Sebastian Jeltsch 2026-04-14 14:07:43 +02:00
parent 65a063d483
commit c522194617
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
client/python/trailbase/__init__.py
View file

@ -470,7 +470,7 @@ class Client:
def refresh_auth_tokens(self):
refreshToken = Client._shouldRefresh(self._tokenState)
if refreshToken is not None:
self._set_token_state(_refreshTokensImpl(self._transport, refreshToken))
self._set_token_state(_refresh_tokens_impl(self._transport, refreshToken))
def _set_token_state(self, tokenState: TokenState) -> TokenState:
self._tokenState = tokenState
@ -503,7 +503,7 @@ class Client:
tokenState = self._tokenState
refreshToken = Client._shouldRefresh(tokenState)
if refreshToken is not None:
tokenState = self._set_token_state(_refreshTokensImpl(self._transport, refreshToken))
tokenState = self._set_token_state(_refresh_tokens_impl(self._transport, refreshToken))
response = self._transport.fetch(
path, method=method, headers=tokenState.headers, queryParams=queryParams, body=data
@ -524,7 +524,7 @@ class Client:
tokenState = self._tokenState
refreshToken = Client._shouldRefresh(tokenState)
if refreshToken is not None:
tokenState = self._set_token_state(_refreshTokensImpl(self._transport, refreshToken))
tokenState = self._set_token_state(_refresh_tokens_impl(self._transport, refreshToken))
return self._transport.stream(
path,
@ -733,7 +733,7 @@ class RecordApi:
return self.subscribe("*")
def _refreshTokensImpl(transport: Transport, refreshToken: str) -> TokenState:
def _refresh_tokens_impl(transport: Transport, refreshToken: str) -> TokenState:
response = transport.fetch(
f"{_AUTH_API}/refresh",
method="POST",
@ -745,10 +745,14 @@ def _refreshTokensImpl(transport: Transport, refreshToken: str) -> TokenState:
},
)
if response.status_code > 200:
raise FetchException(response.status_code, response.text)
return TokenState.build(Tokens.from_json(response.json()))
match response.status_code:
case 200:
return TokenState.build(Tokens.from_json(response.json()))
case 401:
# Refresh token was rejected w/o means to recover. May as well log out.
return TokenState.build(None)
case _:
raise FetchException(response.status_code, response.text)
_logger = logging.getLogger(__name__)