1dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z Gd d ejZ y ) z*Download workflow used by GCS gRPC client.)absolute_import)division)unicode_literals) cloud_api) gcs_download) grpc_util) retry_utilc\eZdZdZfdZdZdZejdZ fdZ xZ S) GrpcDownloadz$Perform GCS Download using gRPC API.c |t ||||||_||_||_||_||_| |_y)a!Initializes a GrpcDownload instance. Args: gapic_client (StorageClient): The GAPIC API client to interact with GCS using gRPC. cloud_resource (resource_reference.ObjectResource): See cloud_api.CloudApi.download_object. download_stream (stream): Stream to send the object data to. start_byte (int): See super class. end_byte (int): See super class. digesters (dict): See cloud_api.CloudApi.download_object. progress_callback (function): See cloud_api.CloudApi.download_object. download_strategy (cloud_api.DownloadStrategy): Download strategy used to perform the download. decryption_key (encryption_util.EncryptionKey|None): The decryption key to be used to download the object if the object is encrypted. N)super__init__ _gapic_client_cloud_resource _digesters_progress_callback_download_strategy_decryption_key) self gapic_clientcloud_resourcedownload_stream start_byteend_byte digestersprogress_callbackdownload_strategydecryption_key __class__s 7lib/googlecloudsdk/api_lib/storage/gcs_grpc/download.pyrzGrpcDownload.__init__sG6 )T#OZJ%D)DDO/D/D)Dc0tj|||S)See super class.)r is_retriable)rexc_type exc_value exc_tracebacks r should_retryzGrpcDownload.should_retryAs  " "8Y FFr!c tj|j|j|j|j |j |j|j|j|j S)r#) rrrrrrrrr) rdownload_objectrr_download_streamrr _start_byte _end_byterrrs r launchzGrpcDownload.launchEsb  $ $''++--//11##11++  r!c"|jS)aDownloads the object. On retriable errors, the entire download will be re-performed instead of resuming from a particular byte. This is useful for streaming download cases. Unlike Apitools, the GAPIC client doesn't retry the request by default, hence we are using the decorator. Returns: Encoding string for object if requested. Otherwise, None. )r/r.s r simple_downloadzGrpcDownload.simple_downloadSs ;;=r!c|jtjjk(r|j St |dS)r#T)retriable_in_flight)rrDownloadStrategyONE_SHOTr1r run)rrs r r6zGrpcDownload.runcsA )"<"<"E"EE  ! ! ## 9d 'D ' AAr!) __name__ __module__ __qualname____doc__rr(r/r grpc_default_retryerr1r6 __classcell__)rs@r r r s=,!*FG "" # BBr!r N) r: __future__rrrgooglecloudsdk.api_lib.storagerr'googlecloudsdk.api_lib.storage.gcs_grpcrr GcsDownloadr r!r rBs51&'47=>MB<++MBr!