rdZddlmZddlmZddlmZddlZddlmZddl m Z ddl m Z dd l mZd Zd d Zy) z,Utilities for retrying requests on failures.)absolute_import)division)unicode_literalsN) http_wrapper)errors) properties)retrycd}||_y)aSets the retry function for the apitools transfer object. Replaces the Apitools' default retry function HandleExceptionsAndRebuildHttpConnections with a custom one which calls HandleExceptionsAndRebuildHttpConnections and then raise a custom exception. This is useful when we don't want MakeRequest method in Apitools to retry the http request directly and instead let the caller decide the next action. Args: apitools_transfer_object (apitools.base.py.transfer.Transfer): The Apitools' transfer object. ctj|t|jtr3|jj t j k(r |jtj)N) apitools_http_wrapper)HandleExceptionsAndRebuildHttpConnections isinstanceexcOSErrorerrnoENOSPCrRetryableApiError) retry_argss 0lib/googlecloudsdk/api_lib/storage/retry_util.py_handle_error_and_raisez/set_retry_func.._handle_error_and_raise*sSCCJO :>>7+ NN ELL 0 NN  " " $$N) retry_func)apitools_transfer_objectrs rset_retry_funcrs% )@%rctjtjjj j tjjjj dztjjjj j|||tjjjj dz|S)afRetries the target with specific default value. This function is intended to be used for all gcloud storage's API requests that require custom retry handling (e.g downloads and uploads). Args: target (Callable): The function to call and retry. should_retry_if (Callable): func(exc_type, exc_value, exc_traceback, state) that returns True or False. target_args (Sequence|None): A sequence of positional arguments to be passed to the target. target_kwargs (Dict|None): A dict of keyword arguments to be passed to target. Returns: Whatever the target returns. i) max_retrialswait_ceiling_msexponential_sleep_multiplier)argskwargssleep_msshould_retry_if) r RetryerrVALUESstorage max_retriesGetIntmax_retry_delayrRetryOnExceptionbase_retry_delay)targetr" target_args target_kwargss rretryerr.=s& $$,,88??A ''//??FFH4O    # # @ @ G G I    $$,,==DDFM)  + +r)NN)__doc__ __future__rrrrapitools.base.pyrr googlecloudsdk.api_lib.storagergooglecloudsdk.corergooglecloudsdk.core.utilr rr.rrr6s.3&' B1**@@+r