o ˷e@sdZddlZddlZddlZddlmZddlmZe Z dZ dZ dZ ejjejjejjejjefZGdd d eZGd d d eZd d ZdS)z}Shared utilities used by both downloads and uploads. This utilities are explicitly catered to ``requests``-like transports. N)common)_helpersi =<c@s4eZdZdZeddZeddZeddZdS) RequestsMixinzMix-in class implementing ``requests``-specific behavior. These are methods that are more general purpose, with implementations specific to the types defined in ``requests``. cC|jS)zAccess the status code from an HTTP response. Args: response (~requests.Response): The HTTP response object. Returns: int: The status code. ) status_coderesponser g/var/www/ideatree/venv/lib/python3.10/site-packages/google/resumable_media/requests/_request_helpers.py_get_status_code5 zRequestsMixin._get_status_codecCr)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 r r r _get_headersAs zRequestsMixin._get_headerscCr)Access the response body from an HTTP response. Args: response (~requests.Response): The HTTP response object. Returns: bytes: The body of the ``response``. )contentr r r r _get_bodyNrzRequestsMixin._get_bodyN)__name__ __module__ __qualname____doc__ staticmethodr rrr r r r r.s   rc@seZdZeddZdS)RawRequestsMixincCs.|jdurd|jjtdd|_d|_|jS)rF)decode_contentT)_contentjoinrawstream_SINGLE_GET_CHUNK_SIZE_content_consumedr r r r r\s zRawRequestsMixin._get_bodyN)rrrrrr r r r r[src Csd}d}t|j|j} d}z|}W|Sty)}z|}WYd}~n#d}~wtjyG}z||jtjvr<|}nWYd}~nd}~wwt ||j |j\}} |d7}|| 7}| ||sc|t | q )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_timer r r wait_and_retryns6     r7)rrequests.exceptionsrequestsurllib3.exceptionsurllib3r,google.resumable_mediarr RetryStrategy_DEFAULT_RETRY_STRATEGYr _DEFAULT_CONNECT_TIMEOUT_DEFAULT_READ_TIMEOUT exceptionsConnectionErrorChunkedEncodingErrorTimeout ProtocolErrorr&objectrrr7r r r r s&   -