Skip to content
This repository was archived by the owner on Mar 23, 2026. It is now read-only.

Commit 5f39c8b

Browse files
committed
fix(pr): address pr's comments
1 parent bfea6c3 commit 5f39c8b

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

indexclient/client.py

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,16 @@ def handle_error(resp):
3131
resp.raise_for_status()
3232

3333

34-
def indexd_request_wrapper(func):
34+
def timeout_wrapper(func):
35+
def timeout(*args, **kwargs):
36+
kwargs.setdefault("timeout", 60)
37+
return func(*args, **kwargs)
38+
return timeout
39+
40+
41+
def retry_and_timeout_wrapper(func):
3542
def retry_logic_with_timeout(*args, **kwargs):
36-
kwargs.update({"timeout": 60})
43+
kwargs.setdefault("timeout", 60)
3744
retries = 0
3845
while retries < MAX_RETRIES:
3946
try:
@@ -311,25 +318,25 @@ def list_versions(self, did):
311318
versions.append(Document(self, version["did"], version))
312319
return versions
313320

314-
@indexd_request_wrapper
321+
@retry_and_timeout_wrapper
315322
def _get(self, *path, **kwargs):
316323
resp = requests.get(self.url_for(*path), **kwargs)
317324
handle_error(resp)
318325
return resp
319326

320-
@indexd_request_wrapper
327+
@timeout_wrapper
321328
def _post(self, *path, **kwargs):
322329
resp = requests.post(self.url_for(*path), **kwargs)
323330
handle_error(resp)
324331
return resp
325332

326-
@indexd_request_wrapper
333+
@timeout_wrapper
327334
def _put(self, *path, **kwargs):
328335
resp = requests.put(self.url_for(*path), **kwargs)
329336
handle_error(resp)
330337
return resp
331338

332-
@indexd_request_wrapper
339+
@timeout_wrapper
333340
def _delete(self, *path, **kwargs):
334341
resp = requests.delete(self.url_for(*path), **kwargs)
335342
handle_error(resp)

0 commit comments

Comments
 (0)