o /ia@sdZddlZddlZddlmZmZmZmZddlZddl Z ddl Z ddl Z ddl Z ddlZ ddlZddlmZmZe jjZedddkr`zddlZededZWn ey_d ZYnwd ZejejfZed Zd d Zd dZ Gdddeeej!Z"e"eZ#ddZ$ddZ%       d(ddZ&         d)dee'fddZ(de)de)fddZ*e+ddZ,e+d d!Z-Gd"d#d#e.Z/d$d%Z0Gd&d'd'ej1Z2dS)*zHelpers for :mod:`grpc`.N)GenericIteratorOptionalTypeVar) exceptionsgeneral_helpersz3.zSupport for grpcio-gcp is deprecated. This feature will be removed from `google-api-core` after January 1, 2024. If you need to continue to use this feature, please pin to a specific version of `google-api-core`.TFPcCst|ds |jj|_dSdS)zFix-up gRPC callable attributes. gRPC callable lack the ``__name__`` attribute which causes :func:`functools.wraps` to error. This adds the attribute if needed. __name__N)hasattr __class__r  callable_rP/var/www/passon-env/lib/python3.10/site-packages/google/api_core/grpc_helpers.py_patch_callable_name:s rc"ttfdd}|S)z;Map errors for Unary-Unary and Stream-Unary gRPC callables.c s8z|i|WStjy}zt||d}~wwN)grpcRpcErrorrfrom_grpc_error)argskwargsexcr rrerror_remapped_callableHs  z3_wrap_unary_errors..error_remapped_callabler functoolswrapsrrrr r_wrap_unary_errorsDsrc@sveZdZdddZdeefddZdefddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZdS)_StreamingResponseIteratorTcCsJ||_z|rt|j|_WdSWdStyYdSty$YdSwr)_wrappednext_stored_first_result TypeError StopIteration)selfwrappedprefetch_first_resultrrr__init__Ss  z#_StreamingResponseIterator.__init__returncCs|S)z6This iterator is also an iterable that returns itself.rr&rrr__iter__dz#_StreamingResponseIterator.__iter__c CsNzt|dr|j}|`|WSt|jWStjy&}zt||d}~ww)zGet the next response from the stream. Returns: protobuf.Message: A single response from the stream. r#N)r r#r"r!rrrr)r&resultrrrr__next__hs   z#_StreamingResponseIterator.__next__cCs |j|Sr)r! add_callbackr&callbackrrrr0zs z'_StreamingResponseIterator.add_callbackcC |jSr)r!cancelr+rrrr4} z!_StreamingResponseIterator.cancelcCr3r)r!coder+rrrr6r5z_StreamingResponseIterator.codecCr3r)r!detailsr+rrrr7r5z"_StreamingResponseIterator.detailscCr3r)r!initial_metadatar+rrrr8r5z+_StreamingResponseIterator.initial_metadatacCr3r)r! is_activer+rrrr9r5z$_StreamingResponseIterator.is_activecCr3r)r!time_remainingr+rrrr:r5z)_StreamingResponseIterator.time_remainingcCr3r)r!trailing_metadatar+rrrr;r5z,_StreamingResponseIterator.trailing_metadataN)T)r __module__ __qualname__r)rr r,r/r0r4r6r7r8r9r:r;rrrrr Rs  r cr)a Wrap errors for Unary-Stream and Stream-Stream gRPC callables. The callables that return iterators require a bit more logic to re-map errors when iterating. This wraps both the initial invocation and the iterator of the return value to re-map errors. c sPz|i|}tdd}t||dWStjy'}zt||d}~ww)N_prefetch_first_result_T)r()getattrr rrrr)rrr.prefetch_firstrr rrrs  z4_wrap_stream_errors..error_remapped_callablerrrr r_wrap_stream_errorssrAcCst|tr t|St|S)a.Wrap a gRPC callable and map :class:`grpc.RpcErrors` to friendly error classes. Errors raised by the gRPC callable are mapped to the appropriate :class:`google.api_core.exceptions.GoogleAPICallError` subclasses. The original `grpc.RpcError` (which is usually also a `grpc.Call`) is available from the ``response`` property on the mapped exception. This is useful for extracting metadata from the original error. Args: callable_ (Callable): A gRPC callable. Returns: Callable: The wrapped gRPC callable. ) isinstance_STREAM_WRAP_CLASSESrArr rrr wrap_errorss rDc Cs|dur ttjt|r|rtd|r"tjj |||d\}}n|r/tjj j |||d}n tjj ||d\}}|rHt |tjj jrH||}tjjj}tjjjj|||d} t| } |rgt|| St| S)a Create the composite credentials for secure channels. Args: credentials (google.auth.credentials.Credentials): The credentials. If not specified, then this function will attempt to ascertain the credentials from the environment using :func:`google.auth.default`. credentials_file (str): Deprecated. A file with credentials that can be loaded with :func:`google.auth.load_credentials_from_file`. This argument is mutually exclusive with credentials. This argument will be removed in the next major version of `google-api-core`. .. warning:: Important: If you accept a credential configuration (credential JSON/File/Stream) from an external source for authentication to Google Cloud Platform, you must validate it before providing it to any Google API or client library. Providing an unvalidated credential configuration to Google APIs or libraries can compromise the security of your systems and data. For more information, refer to `Validate credential configurations from external sources`_. .. _Validate credential configurations from external sources: https://cloud.google.com/docs/authentication/external/externally-sourced-credentials default_scopes (Sequence[str]): A optional list of scopes needed for this service. These are only used when credentials are not specified and are passed to :func:`google.auth.default`. scopes (Sequence[str]): A optional list of scopes needed for this service. These are only used when credentials are not specified and are passed to :func:`google.auth.default`. ssl_credentials (grpc.ChannelCredentials): Optional SSL channel credentials. This can be used to specify different certificates. quota_project_id (str): An optional project to use for billing and quota. default_host (str): The default endpoint. e.g., "pubsub.googleapis.com". Returns: grpc.ChannelCredentials: The composed channel credentials object. Raises: google.api_core.DuplicateCredentialArgs: If both a credentials object and credentials_file are passed. Nz<'credentials' and 'credentials_file' are mutually exclusive.)scopesdefault_scopes) default_host)warningswarnr_CREDENTIALS_FILE_WARNINGDeprecationWarningrDuplicateCredentialArgsgoogleauthload_credentials_from_file credentialswith_scopes_if_requireddefaultrBCredentialsWithQuotaProjectwith_quota_project transportrequestsRequestrAuthMetadataPluginmetadata_call_credentialscomposite_channel_credentials"compute_engine_channel_credentials) rPcredentials_filerFrEssl_credentialsquota_project_idrG_requestmetadata_plugingoogle_auth_credentialsrrr_create_composite_credentialssB0       rcattempt_direct_pathc Ks|r| rtdt|||||||d} tr7|dur%|tjjkr%tdt| r-tdtt j || fi| S| r=t |}tj || fd|i| S)apCreate a secure channel with credentials. Args: target (str): The target service address in the format 'hostname:port'. credentials (google.auth.credentials.Credentials): The credentials. If not specified, then this function will attempt to ascertain the credentials from the environment using :func:`google.auth.default`. scopes (Sequence[str]): A optional list of scopes needed for this service. These are only used when credentials are not specified and are passed to :func:`google.auth.default`. ssl_credentials (grpc.ChannelCredentials): Optional SSL channel credentials. This can be used to specify different certificates. credentials_file (str): A file with credentials that can be loaded with :func:`google.auth.load_credentials_from_file`. This argument is mutually exclusive with credentials. .. warning:: Important: If you accept a credential configuration (credential JSON/File/Stream) from an external source for authentication to Google Cloud Platform, you must validate it before providing it to any Google API or client library. Providing an unvalidated credential configuration to Google APIs or libraries can compromise the security of your systems and data. For more information, refer to `Validate credential configurations from external sources`_. .. _Validate credential configurations from external sources: https://cloud.google.com/docs/authentication/external/externally-sourced-credentials quota_project_id (str): An optional project to use for billing and quota. default_scopes (Sequence[str]): Default scopes passed by a Google client library. Use 'scopes' for user-defined scopes. default_host (str): The default endpoint. e.g., "pubsub.googleapis.com". compression (grpc.Compression): An optional value indicating the compression method to be used over the lifetime of the channel. attempt_direct_path (Optional[bool]): If set, Direct Path will be attempted when the request is made. Direct Path is only available within a Google Compute Engine (GCE) environment and provides a proxyless connection which increases the available throughput, reduces latency, and increases reliability. Note: - This argument should only be set in a GCE environment and for Services that are known to support Direct Path. - If this argument is set outside of GCE, then this request will fail unless the back-end service happens to have configured fall-back to DNS. - If the request causes a `ServiceUnavailable` response, it is recommended that the client repeat the request with `attempt_direct_path` set to `False` as the Service may not support Direct Path. - Using `ssl_credentials` with `attempt_direct_path` set to `True` will result in `ValueError` as this combination is not yet supported. kwargs: Additional key-word args passed to :func:`grpc_gcp.secure_channel` or :func:`grpc.secure_channel`. Note: `grpc_gcp` is only supported in environments with protobuf < 4.0.0. Returns: grpc.Channel: The created channel. Raises: google.api_core.DuplicateCredentialArgs: If both a credentials object and credentials_file are passed. ValueError: If `ssl_credentials` is set and `attempt_direct_path` is set to `True`. z7Using ssl_credentials with Direct Path is not supported)rPr\rFrEr]r^rGNzKThe `compression` argument is ignored for grpc_gcp.secure_channel creation.zSThe `attempt_direct_path` argument is ignored for grpc_gcp.secure_channel creation. compression) ValueErrorrc HAS_GRPC_GCPr Compression NoCompressionrHrIrKgrpc_gcpsecure_channel_modify_target_for_direct_path) targetrPrEr]r\r^rFrGrerdrcomposite_credentialsrrrcreate_channel2s@M rormr*cCs<d}||d}d}||vr|dd}d||}|S)ag Given a target, return a modified version which is compatible with Direct Path. Args: target (str): The target service address in the format 'hostname[:port]' or 'dns://hostname[:port]'. Returns: target (str): The target service address which is converted into a format compatible with Direct Path. If the target contains `dns:///` or does not contain `:///`, the target will be converted in a format compatible with Direct Path; otherwise the original target will be returned as the original target may already denote Direct Path. zdns:///z:///:rz google-c2p)replacesplit)rm dns_prefixdirect_path_separatortarget_without_portrrrrls rl _MethodCall)r`timeoutmetadatarPre_ChannelRequest)methodr`c@s$eZdZdZddZ dddZdS) _CallableStubz,Stub for the grpc.*MultiCallable interfaces.cCs.||_||_d|_ d|_ g|_ g|_dSr)_method_channelresponse responsesrVcalls)r&r{channelrrrr)sz_CallableStub.__init__NcCs|jjt|j||jt||||||j||j}|jdur8|dur/t |j}n t dj |jdt |r@||St |trG||durM|St d |j)Nz@{method}.response and {method}.responses are mutually exclusive.r{z%Method stub for "{}" has no response.)r~rVappendrzr}rrwrrr"rfformatcallablerB Exception)r&r`rxryrPrerrrr__call__s*    z_CallableStub.__call__)NNNN)r r<r=__doc__r)rrrrrr|s r|cCs|ddS)amSimplifies a gRPC method name. When gRPC invokes the channel to create a callable, it gives a full method name like "/google.pubsub.v1.Publisher/CreateTopic". This returns just the name of the method, in this case "CreateTopic". Args: method (str): The name of the method. Returns: str: The simplified name of the method. /)rsplitpoprrrr_simplify_method_names rc@seZdZdZgfddZddZddZ   dd d Z   dd d Z   dddZ   dddZ dddZ ddZ ddZ dS) ChannelStubaA testing stub for the grpc.Channel interface. This can be used to test any client that eventually uses a gRPC channel to communicate. By passing in a channel stub, you can configure which responses are returned and track which requests are made. For example: .. code-block:: python channel_stub = grpc_helpers.ChannelStub() client = FooClient(channel=channel_stub) channel_stub.GetFoo.response = foo_pb2.Foo(name='bar') foo = client.get_foo(labels=['baz']) assert foo.name == 'bar' assert channel_stub.GetFoo.requests[0].labels = ['baz'] Each method on the stub can be accessed and configured on the channel. Here's some examples of various configurations: .. code-block:: python # Return a basic response: channel_stub.GetFoo.response = foo_pb2.Foo(name='bar') assert client.get_foo().name == 'bar' # Raise an exception: channel_stub.GetFoo.response = NotFound('...') with pytest.raises(NotFound): client.get_foo() # Use a sequence of responses: channel_stub.GetFoo.responses = iter([ foo_pb2.Foo(name='bar'), foo_pb2.Foo(name='baz'), ]) assert client.get_foo().name == 'bar' assert client.get_foo().name == 'baz' # Use a callable def on_get_foo(request): return foo_pb2.Foo(name='bar' + request.id) channel_stub.GetFoo.response = on_get_foo assert client.get_foo(id='123').name == 'bar123' cCsg|_ i|_dSr)rV _method_stubs)r&rrrrr)Es zChannelStub.__init__cCs"t|}t|||j|<|j|Sr)rr|r)r&r{rrr_stub_for_methodLs zChannelStub._stub_for_methodcCs z|j|WStytwr)rKeyErrorAttributeError)r&keyrrr __getattr__Qs   zChannelStub.__getattr__NFcC ||S)z(grpc.Channel.unary_unary implementation.rr&r{request_serializerresponse_deserializer_registered_methodrrr unary_unaryW zChannelStub.unary_unarycCr)z)grpc.Channel.unary_stream implementation.rrrrr unary_streamarzChannelStub.unary_streamcCr)z)grpc.Channel.stream_unary implementation.rrrrr stream_unarykrzChannelStub.stream_unarycCr)z*grpc.Channel.stream_stream implementation.rrrrr stream_streamurzChannelStub.stream_streamcCdS)z&grpc.Channel.subscribe implementation.Nr)r&r2try_to_connectrrr subscriber-zChannelStub.subscribecCr)z(grpc.Channel.unsubscribe implementation.Nrr1rrr unsubscriber-zChannelStub.unsubscribecCr)z"grpc.Channel.close implementation.Nrr+rrrcloser-zChannelStub.close)NNF)F)r r<r=rr)rrrrrrrrrrrrrr s0 7         r)NNNNNNN) NNNNNNNNF)3r collectionsrtypingrrrrrH google.authrMgoogle.auth.credentialsgoogle.auth.transport.grpcgoogle.auth.transport.requestsgoogle.protobufrgoogle.api_corerrprotobuf __version__PROTOBUF_VERSIONrjrIrKrg ImportErrorUnaryStreamMultiCallableStreamStreamMultiCallablerCr rrCallr GrpcStreamrArDrcboolrostrrl namedtuplerwrzobjectr|rChannelrrrrrsz   B l p 6