JZdZddlmZddlmZddlmZddlmZddlmZGdde Z y ) z1Common download workflow used by GCS API clients.)absolute_import)division)unicode_literals) retry_util)logc0eZdZdZdZdZdZdZddZy) GcsDownloadz(Base class for performing GCS downloads.c<||_||_||_||_y)aInitializes a GcsDownload instance. Args: download_stream (stream): Stream to send the object data to. start_byte (int): Starting point for download (for resumable downloads and range requests). Can be set to negative to request a range of bytes (python equivalent of [:-3]). end_byte (int): Ending byte number, inclusive, for download (for range requests). If None, download the rest of the object. N)_download_stream_initial_start_byte _end_byte _start_byte)selfdownload_stream start_byteend_bytes 2lib/googlecloudsdk/api_lib/storage/gcs_download.py__init__zGcsDownload.__init__s#,D)DDN!Dct)z,Returns True if the error should be retried.NotImplementedError)rexc_type exc_value exc_tracebacks r should_retryzGcsDownload.should_retry-s  rct)a?Performs the download. The child classes should implement this function. Returns: Encoding string for object if requested. Otherwise, None. Raises: CloudApiError: API returned an error. NotImplementedError: This function was not implemented by a class using this interface. r)rs rlaunchzGcsDownload.launch1s   rc|j|||sy|jj}||jkDr||_d|_t j dj|||y)aReturns True if the error should be retried. This method also updates the start_byte to be used for request to be retried. Args: exc_type (type): The type of Exception. exc_value (Exception): The error instance. exc_traceback (traceback): The traceback for the exception. state (core.util.retry.RetryState): The state object maintained by the retryer. Returns: True if the error should be retried. Frz=Retrying download from byte {} after exception: {}. Trace: {}T)rr tellrretrialrdebugformat)rrrrstaters r_should_retry_wrapperz!GcsDownload._should_retry_wrapper@so   Xy- @ &&++-JD$$$#demII!6*h FH rc||s|jStj|j|jS)zPerforms downlaod. Args: retriable_in_flight (bool): Indicates if a download can be retried on network error, resuming the download from the last downloaded byte. Returns: The result returned by launch method. )targetshould_retry_if)rrretryerr%)rretriable_in_flights rrunzGcsDownload.run]s7  [[]   {{D,F,F HHrN)T) __name__ __module__ __qualname____doc__rrrr%r+rrr r s0""   :Hrr N) r/ __future__rrrgooglecloudsdk.api_lib.storagergooglecloudsdk.corerobjectr r0rrr5s)8&'5#RH&RHr