@@ -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