Skip to content

Commit dc76e8b

Browse files
committed
Remove retries
1 parent b4f173f commit dc76e8b

File tree

1 file changed

+4
-32
lines changed

1 file changed

+4
-32
lines changed

singlestoredb/management/manager.py

Lines changed: 4 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
#!/usr/bin/env python
22
"""SingleStoreDB Base Manager."""
3-
import logging
43
import os
54
import sys
65
import time
@@ -58,13 +57,11 @@ def __init__(
5857
base_url: Optional[str] = None, *, organization_id: Optional[str] = None,
5958
):
6059
from .. import __version__ as client_version
61-
refresh_token = not bool(access_token)
6260
access_token = (
6361
access_token or get_token()
6462
)
6563
if not access_token:
6664
raise ManagementError(msg='No management token was configured.')
67-
self._auth_is_jwt = refresh_token and is_jwt(access_token)
6865
self._sess = requests.Session()
6966
self._sess.headers.update({
7067
'Authorization': f'Bearer {access_token}',
@@ -118,35 +115,10 @@ def _doit(
118115
*args: Any,
119116
**kwargs: Any,
120117
) -> requests.Response:
121-
"""Perform HTTP request and retry as needed."""
122-
retries = 5
123-
while retries > 0:
124-
if self._auth_is_jwt:
125-
logging.debug('making request with JWT')
126-
else:
127-
logging.debug('making request without JWT')
128-
res = getattr(self._sess, method.lower())(
129-
urljoin(self._base_url, path), *args, **kwargs,
130-
)
131-
# Retry for authentication errors because the JWT in the
132-
# notebook env might need to be refreshed.
133-
if self._auth_is_jwt and res.status_code == 401:
134-
logging.debug('authorization error with JWT; retrying')
135-
access_token: Optional[str] = str(self._sess.headers['Authorization'])
136-
logging.debug(f'old JWT: {access_token}')
137-
time.sleep(5)
138-
access_token = get_token()
139-
logging.debug(f'new JWT: {access_token}')
140-
if access_token is not None:
141-
self._sess.headers.update({
142-
'Authorization': f'Bearer {access_token}',
143-
})
144-
retries -= 1
145-
continue
146-
elif res.status_code == 401:
147-
logging.debug('authorization error; no retries')
148-
break
149-
return res
118+
"""Perform HTTP request."""
119+
return getattr(self._sess, method.lower())(
120+
urljoin(self._base_url, path), *args, **kwargs,
121+
)
150122

151123
def _get(self, path: str, *args: Any, **kwargs: Any) -> requests.Response:
152124
"""

0 commit comments

Comments
 (0)