@ f dZddlZddlZddlZddlZddlm Z ddlm Z e j Z dZ dZdZejjejjejjejjejjejjejjejjejjejjef ZGdd eZGd d eZd Z dS) z}Shared utilities used by both downloads and uploads. This utilities are explicitly catered to ``requests``-like transports. N)common)_helpersi =<cTeZdZdZedZedZedZdS) RequestsMixinzMix-in class implementing ``requests``-specific behavior. These are methods that are more general purpose, with implementations specific to the types defined in ``requests``. c|jS)zAccess the status code from an HTTP response. Args: response (~requests.Response): The HTTP response object. Returns: int: The status code. ) status_coderesponses /home/panchajanya/Development/work/volatility3/lib/python3.11/site-packages/google/resumable_media/requests/_request_helpers.py_get_status_codezRequestsMixin._get_status_code<s ##c|jS)a Access the headers from an HTTP response. Args: response (~requests.Response): The HTTP response object. Returns: ~requests.structures.CaseInsensitiveDict: The header mapping (keys are case-insensitive). )headersr s r _get_headerszRequestsMixin._get_headersHs rc|jS)Access the response body from an HTTP response. Args: response (~requests.Response): The HTTP response object. Returns: bytes: The body of the ``response``. )contentr s r _get_bodyzRequestsMixin._get_bodyUs rN)__name__ __module__ __qualname____doc__ staticmethodrrrrr rr5so  $ $\ $   \     \    rrc$eZdZedZdS)RawRequestsMixinc|jdur@d|jtd|_d|_|jS)rFr)decode_contentT)_contentjoinrawstream_SINGLE_GET_CHUNK_SIZE_content_consumedr s r rzRawRequestsMixin._get_bodycsT   % % # ##$:5#QQ!!H *.H &  rN)rrrrrrrr rrbs-!!\!!!rrcd}d}t|j|jz } d} |}|S#t$r }|}Yd}~n?d}~wtj$r*}||jtjvr|}nYd}~nd}~wwxYwtj ||j |j\}} |dz }|| z }| ||s|tj | )a.Attempts to retry a call to ``func`` until success. Expects ``func`` to return an HTTP response and uses ``get_status_code`` to check if the response is retry-able. ``func`` is expected to raise a failure status code as a common.InvalidResponse, at which point this method will check the code against the common.RETRIABLE list of retriable status codes. Will retry until :meth:`~.RetryStrategy.retry_allowed` (on the current ``retry_strategy``) returns :data:`False`. Uses :func:`_helpers.calculate_retry_wait` to double the wait time (with jitter) after each attempt. Args: func (Callable): A callable that takes no arguments and produces an HTTP response which will be checked as retry-able. get_status_code (Callable[Any, int]): Helper to get a status code from a response. retry_strategy (~google.resumable_media.common.RetryStrategy): The strategy to use if the request fails and must be retried. Returns: object: The return value of ``func``. grTN)float initial_delay multiplier_CONNECTION_ERROR_CLASSESrInvalidResponser RETRYABLErcalculate_retry_wait max_sleep retry_allowedtimesleep) funcget_status_coderetry_strategy total_sleep num_retries base_waiterrorr e wait_times r wait_and_retryr=us34KKn233n6OOI  tvvHO)   EEEEEE%   qz**f.>>>   (< ~/1J   9 q y ++KEE K 99s 1 A>AA> A99A>)!r http.clienthttprequests.exceptionsrequestsurllib3.exceptionsurllib3r2google.resumable_mediarr RetryStrategy_DEFAULT_RETRY_STRATEGYr%_DEFAULT_CONNECT_TIMEOUT_DEFAULT_READ_TIMEOUTclient BadStatusLineIncompleteReadResponseNotReady exceptionsConnectionErrorChunkedEncodingErrorTimeout PoolError ProtocolErrorSSLError TimeoutErrorr,objectrrr=rrr rVsY  ))))))++++++.&.00 KKK  ' ,    $  # * * * * * F* * * Z!!!!!}!!!&?????r