From e84f1edfb9d027ac9a9dcd2e3ca74a8469c5a7f6 Mon Sep 17 00:00:00 2001 From: Patryk Gala Date: Thu, 19 Dec 2024 14:13:36 +0100 Subject: [PATCH 1/2] chore: Use retry mechanism while refreshing oauth token. --- src/neptune/internal/oauth.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/neptune/internal/oauth.py b/src/neptune/internal/oauth.py index 6234e03d6..d44e7d832 100644 --- a/src/neptune/internal/oauth.py +++ b/src/neptune/internal/oauth.py @@ -52,7 +52,7 @@ def __call__(self, r): try: return self._add_token(r) except TokenExpiredError: - self._refresh_token() + self.refresh_token_if_needed(force=True) return self._add_token(r) def _add_token(self, r): @@ -62,8 +62,8 @@ def _add_token(self, r): return r @with_api_exceptions_handler - def refresh_token_if_needed(self): - if self.token_expires_at - time.time() < 30: + def refresh_token_if_needed(self, force=False): + if self.token_expires_at - time.time() < 30 or force: self._refresh_token() def _refresh_token(self): From 454a1edc94c61ccd34434b9ef7facfc8f40b1c94 Mon Sep 17 00:00:00 2001 From: Patryk Gala Date: Thu, 19 Dec 2024 14:37:50 +0100 Subject: [PATCH 2/2] chore: Use retry mechanism while refreshing oauth token. --- src/neptune/internal/oauth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/neptune/internal/oauth.py b/src/neptune/internal/oauth.py index d44e7d832..fe94782a9 100644 --- a/src/neptune/internal/oauth.py +++ b/src/neptune/internal/oauth.py @@ -63,7 +63,7 @@ def _add_token(self, r): @with_api_exceptions_handler def refresh_token_if_needed(self, force=False): - if self.token_expires_at - time.time() < 30 or force: + if (self.token_expires_at - time.time()) < 30 or force: self._refresh_token() def _refresh_token(self):