jdZddlmZddlmZddlmZddlZddlmZddl m Z ddl m Z ddlm Zddlm ZddlmZddlmZdd lmZdd lmZd j/d j1eej4Dcgc]}|j6c}Zdad edefdZ dZ!dZ"ddZ#ddZ$ycc}w)z5Returns correct API client instance for user command.)absolute_import)division)unicode_literalsN)errorsclient) storage_url)log) propertiesz-Invalid provider. Valid provider prefixes: {}z, bucket_namereturnc>|r|tjjk7rytj} |j |}|j dk(S#tj$r3}t|dd}|dvrtjd||Yd}~yd}~wwxYw)z7Returns true if the given bucket is a GCS zonal bucket.Fzone status_codeN)iiiz.Failed to get storage layout for bucket %s: %s) r ProviderPrefixGCSgcs_json_client JsonClientget_storage_layout locationTypecloud_api_errors CloudApiErrorgetattrr debug)providerr api_clientstorage_layouters 1lib/googlecloudsdk/api_lib/storage/api_factory.py_is_gcs_zonal_bucketr 0s K$>$>$B$BB ))+*  22;?N  & && 00  ' '  !]D1Ko% ii :K    sAB)(BBBc|tjjk(r[|r'tjdddlm}|jStjjjjsWtjjjjtjj j"k(r%tjdt$j&Stjjj(jrBtjjj*jrt,j.St0j2S|tjj4k(rt6j8St;j<t>)a]Returns a CloudApi subclass corresponding to the requested provider. Args: provider (storage_url.ProviderPrefix): Cloud provider prefix. is_zonal_bucket (bool): Whether the bucket is a zonal bucket. Returns: An appropriate CloudApi subclass. Raises: Error: If provider is not a cloud scheme in storage_url.ProviderPrefix. z2Using gRPC Bidi Streaming client for zonal bucket.rrz%Using gRPC client with JSON Fallback.) r rrr r6googlecloudsdk.api_lib.storage.gcs_grpc_bidi_streamingrGcsGrpcBidiStreamingClientr VALUESstorageuse_grpc_if_availableGetBool preferred_apiGetStoragePreferredApiGRPC_WITH_JSON_FALLBACKvaluegcs_grpc_clientGrpcClientWithJsonFallbackgs_xml_access_key_idgs_xml_secret_access_keygcs_xml_client XmlClientrrS3 s3_xml_client S3XmlClientrError _INVALID_PROVIDER_PREFIX_MESSAGE)ris_zonal_bucketgrpc_bidi_streaming_clients r_get_api_classr:Es/++/// iiDEm ' B BB   66>>@!!//335  ) ) A A G G H ii78  7 77!!66::<    % % > > B B D  % %%  % %%;--000  $ $$ ,,7 88c>|r|jdS|jS)zGReturns the thread local storage key for the given provider and bucket._zonal)r,)rr8s r_get_thread_local_storage_keyr>rs$nn V $$ NN r;ct||}tjjjj r9t stjatt t||d}|r|St||}|}tjjjj rtt t||||S)aCReturns thread local API instance based on provider and bucket(if available). Uses thread local storage to make sure only one instance of an API exists per thread per provider + bucket type. Args: provider (storage_url.ProviderPrefix): Cloud provider prefix. bucket_name (str): Name of the bucket. If available, the API client will be chosen based on the bucket type as well. Currently only supported for GCS. Returns: CloudApi client object for the arguments. Raises: Error: If provider is not a cloud scheme in storage_url.ProviderPrefix. N) r r r$r%use_threading_localr'_cloud_api_thread_local_storage threadinglocalrr>r:setattr)rr r8r api_classs rget_apirFys.);?/22::< *(1(9%'%h@ J  X7){*22::< '%h@ r;cFt|t||}|jS)aGets the capabilities of a cloud provider. Args: provider (storage_url.ProviderPrefix): Cloud provider prefix. bucket_name (str): Name of the bucket. If available, the API client will be chosen based on the bucket type as well. Currently only supported for GCS. Returns: The CloudApi.capabilities attribute for the requested provider. Raises: Error: If provider is not a cloud scheme in storage_url.ProviderPrefix. )r:r capabilities)rr rEs rget_capabilitiesrIs)$X{;)   r;)N)%__doc__ __future__rrrrBgooglecloudsdk.api_lib.storagerr'googlecloudsdk.api_lib.storage.gcs_grpcrr-'googlecloudsdk.api_lib.storage.gcs_jsonr&googlecloudsdk.api_lib.storage.gcs_xmlr1%googlecloudsdk.api_lib.storage.s3_xmlr4"googlecloudsdk.command_lib.storager googlecloudsdk.corer r formatjoinsortedVALID_CLOUD_SCHEMESr,r7rAstrboolr r:r>rFrI)schemes0rrZs<''EMMKI5:#*4:: {/N/NO/NVFLL/NO P !#'   **9Z/d KPs5B0