=dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l mZGd d e jZ dd Z d ddZ ddZ ddZ ddZy)z1Helper Classes for using gapic clients in gcloud.)absolute_import) annotations)division)unicode_literals)external_account) exceptions)requests)creds)storeceZdZdZy)MissingStoredCredentialsErrorz?Indicates stored credentials do not exist or are not available.N)__name__ __module__ __qualname____doc__%lib/googlecloudsdk/core/gapic_util.pyr r sGrr ctj|dtjs t d|rtj _jfd}|_S)aReturns a credential object for use by gapic client libraries. Currently, we set _quota_project on the credentials, unlike for http requests, which add quota project through request wrapping to implement go/gcloud-quota-model-v2. Additionally, we wrap the refresh method and plug in our own google.auth.transport.Request object that uses our transport. Args: enable_resource_quota: bool, By default, we are going to tell APIs to use the quota of the project being operated on. For some APIs we want to use gcloud's quota, so you can explicitly disable that behavior by passing False here. allow_account_impersonation: bool, True to allow use of impersonated service account credentials for calls made with this client. If False, the active user credentials will always be used. Returns: A google auth credentials.Credentials object. Raises: MissingStoredCredentialsError: If a google-auth credential cannot be loaded. T)allow_account_impersonationuse_google_authzUnable to load credentialsc~ttjr jryt j S)N) isinstancegoogle_auth_external_account Credentialsvalidr GoogleAuthRequest)request credentialsoriginal_refreshs rWrappedRefreshz+GetGapicCredentials..WrappedRefreshJs<$0026A6G6G  H668 99r)r LoadIfEnabledr IsGoogleAuthCredentialsr GetQuotaProject_quota_project_idrefresh)enable_resource_quotarr!rr s @@rGetGapicCredentialsr(!sp6##"=+  & &{ 3 '(D EE$)$9$9+$FK! !((:'+ rNc8ddlm}|j||||S)aInitializes a BidiRpc instances. Args: client: GAPIC Wrapper client to use. start_rpc (grpc.StreamStreamMultiCallable): The gRPC method used to start the RPC. initial_request: The initial request to yield. This is useful if an initial request is needed to start the stream. metadata: The metadata headers to use for the RPC. It is a list of tuples. The first string in the tuple is the header name and the second is the header value. Returns: A bidiRPC instance. rgapic_util_internal)initial_requestmetadata)googlecloudsdk.corer+BidiRpc)client start_rpcr,r-r+s r MakeBidiRpcr2Xs(,6  $ $ i8 % rc|jd}|r |jn |j}|r||}||||S)aInstantiates a gapic REST client with gcloud defaults and configuration. Args: client_class: a gapic client class. credentials: google.auth.credentials.Credentials, the credentials to use. address_override_func: function, function to call to override the client host. It takes a single argument which is the original host. mtls_enabled: bool, True if mTLS is enabled for this client. _ Returns: A gapic API client. rest)hostr transport)get_transport_classDEFAULT_MTLS_ENDPOINTDEFAULT_ENDPOINT) client_classraddress_override_func mtls_enabledtransport_classaddresss rMakeRestClientr@usO !44VL . .LDaDa'#G,G W+F HHrc Hddlm}||j||||||S)aInstantiates a gapic API client with gcloud defaults and configuration. grpc cannot be packaged like our other Python dependencies, due to platform differences and must be installed by the user. googlecloudsdk.core.gapic depends on grpc and must be imported lazily here so that this module can be imported safely anywhere. Args: client_class: a gapic client class. credentials: google.auth.credentials.Credentials, the credentials to use. address_override_func: function, function to call to override the client host. It takes a single argument which is the original host. mtls_enabled: bool, True if mTLS is enabled for this client. attempt_direct_path: bool, True if we want to attempt direct path gRPC where possible redact_request_body_reason: str, the reason why the request body must be redacted if --log-http is used. If None, the body is not redacted. Returns: A gapic API client. rr*r6)r.r+ MakeTransport)r;rr<r=attempt_direct_pathredact_request_body_reasonr+s r MakeClientrEs4<6 #11      $   rc Fddlm}||j|||||S)a"Instantiates a gapic API client with gcloud defaults and configuration. grpc cannot be packaged like our other Python dependencies, due to platform differences and must be installed by the user. googlecloudsdk.core.gapic depends on grpc and must be imported lazily here so that this module can be imported safely anywhere. Args: client_class: a gapic client class. credentials: google.auth.credentials.Credentials, the credentials to use. address_override_func: function, function to call to override the client host. It takes a single argument which is the original host. mtls_enabled: bool, True if mTLS is enabled for this client. attempt_direct_path: bool, True if we want to attempt direct path gRPC where possible Returns: A gapic API client. rr*r6)r.r+MakeAsyncTransport)r;rr<r=rCr+s rMakeAsyncClientrHs166 #66       r)TT)NN)r-zlist[tuple[str, str]] | None)NF)NFFN)NFF)r __future__rrrr google.authrrr.rr googlecloudsdk.core.credentialsr r Errorr r(r2r@rErHrrrrMs8&"'H*(11HJ$4$4H/3484t-1 + >*. %H6# )^ %r