GdZddlZddlZddlmZddlZddlZejZ dZ dZ ejrRddl Z e jZeejj j"ejj j$gZn@ejj j"ejj j$gZdZGddeZGd d ej.Zy) zHThis package facilitates retries for HTTP/REST requests to the registry.N)nestedg?c6tD]}t||syy)NTF)RETRYABLE_EXCEPTION_TYPES isinstance)errexception_types 5lib/third_party/containerregistry/transport/retry_.py ShouldRetryr *s1n#~& 2 c4eZdZdZdZdZdZdZdZdZ y) Factoryz'A factory for creating RetryTransports.c(i|_t|_yN)kwargs!DEFAULT_SOURCE_TRANSPORT_CALLABLEsource_transport_callableselfs r __init__zFactory.__init__5sDK%FD"r c||_|Sr)r)rrs r WithSourceTransportCallablez#Factory.WithSourceTransportCallable9s%>D" Kr c$||jd<|S)N max_retriesr)rrs r WithMaxRetrieszFactory.WithMaxRetries=s!,DKK  Kr c$||jd<|S)Nbackoff_factorr)rrs r WithBackoffFactorzFactory.WithBackoffFactorAs$2DKK ! Kr c$||jd<|S)Nshould_retry_fnr)rr!s r WithShouldRetryFunctionzFactory.WithShouldRetryFunctionEs%4DKK!" Kr cJt|jfi|jS)z;Returns a RetryTransport constructed with the given values.)RetryTransportrrrs r Buildz Factory.BuildIs $88: Jdkk JJr N) __name__ __module__ __qualname____doc__rrrrr"r%r r rr2s%/GKr rc0eZdZdZeeeffd ZdZxZ S)r$zEA wrapper for the given transport which automatically retries errors.cVtt| |||_||_||_yr)superr$r _max_retries_backoff_factor _should_retry)rsource_transportrrr! __class__s r rzRetryTransport.__init__Qs-  .$()9:#D)D(Dr c0d} |jj|i|S#t$rk}||jk\s|j |st j d||dz }tj|jd|zzYd}~d}~wwxYw)aDoes the request, exponentially backing off and retrying as appropriate. Backoff is backoff_factor * (2 ^ (retry #)) seconds. Args: *args: The sequence of positional arguments to forward to the source transport. **kwargs: The keyword arguments to forward to the source transport. Returns: The response of the HTTP request, and its contents. rzRetrying after exception %s.rN) r1request Exceptionr.r0loggingerrortimesleepr/)rargsrretriesrs r r5zRetryTransport.request[sG  ,t$$,,d=f==  d'' 't/A/A#/F  4c:1  4''1g:67s! BA!BB) r&r'r(r)DEFAULT_MAX_RETRIESDEFAULT_BACKOFF_FACTORr rr5 __classcell__)r2s@r r$r$NsM1 6!, )r r$)r)r7r9containerregistry.transportrhttplib2six.moves.http_clientsixHttprr=r>PY3builtinsBrokenPipeErrormoves http_clientIncompleteReadResponseNotReadyrr objectrNestedTransportr$r*r r rNsO .$,MM!77,,/ ii** ii,, ii** ii,, KfK8$V++$r