B fZfdZddlmZmZmZmZddlZddlZddlZddl Z ddl m Z ddl Z ddlZ ddlZ ddlZ ddlZ e jjZedddkr& ddlZejdedZn#e$rd ZYnwxYwd Ze je jfZed Zd Zd ZGd deee j Z!e!eZ"dZ#dZ$ d dZ% d!dee&fdZ'de(de(fdZ)ej*ddZ+ej*ddZ,Gdde-Z.dZ/Gdde j0Z1dS)"zHelpers for :mod:`grpc`.)GenericIteratorOptionalTypeVarN) exceptionsz3.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`.TFPcLt|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_s k/home/panchajanya/Development/work/volatility3/lib/python3.11/site-packages/google/api_core/grpc_helpers.py_patch_callable_namer;s4 9j ) ):&09 ::cdttjfd}|S)z;Map errors for Unary-Unary and Stream-Unary gRPC callables.cn |i|S#tj$r}tj||d}~wwxYwN)grpcRpcErrorrfrom_grpc_error)argskwargsexcrs rerror_remapped_callablez3_wrap_unary_errors..error_remapped_callableIsQ ;9d-f-- -} ; ; ;,S11s : ;s 4/4r functoolswrapsrrs` r_wrap_unary_errorsr!EsG###_Y;;;; ; #"rcjeZdZddZdeefdZdefdZdZdZ dZ d Z d Z d Z d Zd ZdS)_StreamingResponseIteratorTc||_ |rt|j|_dSdS#t$rYdSt$rYdSwxYwr)_wrappednext_stored_first_result TypeError StopIteration)selfwrappedprefetch_first_results r__init__z#_StreamingResponseIterator.__init__Tsx $ @,0,?,?))) @ @    DD    DD s( A AAreturnc|S)z6This iterator is also an iterable that returns itself.r*s r__iter__z#_StreamingResponseIterator.__iter__es rc t|dr |j}|`|St|jS#tj$r}t j||d}~wwxYw)zGet the next response from the stream. Returns: protobuf.Message: A single response from the stream. r'N)r r'r&r%rrrr)r*resultrs r__next__z#_StreamingResponseIterator.__next__isq  ;t344 2-  && &} ; ; ;,S11s : ;s11AAAc6|j|Sr)r% add_callbackr*callbacks rr7z'_StreamingResponseIterator.add_callback{s}))(333rc4|jSr)r%cancelr1s rr;z!_StreamingResponseIterator.cancel~s}##%%%rc4|jSr)r%coder1s rr=z_StreamingResponseIterator.codes}!!###rc4|jSr)r%detailsr1s rr?z"_StreamingResponseIterator.detailss}$$&&&rc4|jSr)r%initial_metadatar1s rrAz+_StreamingResponseIterator.initial_metadatas}--///rc4|jSr)r% is_activer1s rrCz$_StreamingResponseIterator.is_actives}&&(((rc4|jSr)r%time_remainingr1s rrEz)_StreamingResponseIterator.time_remainings}++---rc4|jSr)r%trailing_metadatar1s rrGz,_StreamingResponseIterator.trailing_metadatas}..000rN)T)r __module__ __qualname__r-rr r2r5r7r;r=r?rArCrErGr0rrr#r#Ss"(1+;!;;;;$444&&&$$$'''000)))...11111rr#cdttjfd}|S)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 |i|}tdd}t||S#tj$r}t j||d}~wwxYw)N_prefetch_first_result_T)r,)getattrr#rrrr)rrr4prefetch_firstrrs rrz4_wrap_stream_errors..error_remapped_callables ;Y///F %Y0I4PPN-n } ; ; ;,S11s : ;s)-AAArr s` r_wrap_stream_errorsrOsI###_Y ; ; ; ;  ; #"rcht|trt|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_CLASSESrOr!rs r wrap_errorsrSs3 )122-"9---!),,,rc|r|rtjd|r&tj|||\}}nN|r(tjj|||}n$tj||\}}|r9t|tjjj r| |}tjj j }tjj j|||} tj| } |rtj|| Stj| S)aCreate 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): A file with credentials that can be loaded with :func:`google.auth.load_credentials_from_file`. This argument is mutually exclusive with 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. z<'credentials' and 'credentials_file' are mutually exclusive.)scopesdefault_scopes) default_host)rDuplicateCredentialArgsgoogleauthload_credentials_from_file credentialswith_scopes_if_requireddefaultrQCredentialsWithQuotaProjectwith_quota_project transportrequestsRequestrAuthMetadataPluginmetadata_call_credentialscomposite_channel_credentials"compute_engine_channel_credentials) r\credentials_filerVrUssl_credentialsquota_project_idrW_requestmetadata_plugingoogle_auth_credentialss r_create_composite_credentialsrosF ' 0 J     ?? VN@   QQ  k-EE ~F   ,,.-   QGJV[,HG"445EFF k#,4466Gk+0CC!DO#<_MM P1 4   67NOOOrattempt_direct_pathc n|r| rtdt|||||||} tr`|/|tjjkrt jdt| rt jdttj || fi| S| rt|}tj || fd|i| S)a Create 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. 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)r\rhrVrUrirjrWNzKThe `compression` argument is ignored for grpc_gcp.secure_channel creation.zSThe `attempt_direct_path` argument is ignored for grpc_gcp.secure_channel creation. compression) ValueErrorro HAS_GRPC_GCPr Compression NoCompressionwarningswarnDeprecationWarninggrpc_gcpsecure_channel_modify_target_for_direct_path) targetr\rUrirhrjrVrWrrrprcomposite_credentialss rcreate_channelr#sBT.TRSSS9)%')! P  "{d6F6T'T'T M]"      Mi"   &v/DOOOOO8/77  %  3> BH  rr}r.cd}||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)r} dns_prefixdirect_path_separatortarget_without_ports rr|r|s_"J^^J + +F"F**$ll3//2J3J5HJJ Mr _MethodCall)rltimeoutmetadatar\rr_ChannelRequest)methodrlc"eZdZdZdZ ddZdS) _CallableStubz,Stub for the grpc.*MultiCallable interfaces.c`||_||_d|_ d|_ g|_ g|_dSr)_method_channelresponse responsesrbcalls)r*rchannels rr-z_CallableStub.__init__sJ     C I  E ErNc B|jjt|j||jt ||||||j||j}|j?|t|j}n(td |jt|r ||St|tr|||Std |j)Nz@{method}.response and {method}.responses are mutually exclusive.rz%Method stub for "{}" has no response.)rrbappendrrrrrrr&rsformatcallablerQ Exception)r*rlrrr\rrrs r__call__z_CallableStub.__call__s %%odlG&L&LMMM  (K M M    W%%%= > %// !!'t|!rs 777777777777 &&&&&&!!!!%%%%?.AaCD     "       L5t7UV GCLL::: # # #>1>1>1>1>1TY>1>1>1D( * ###6---. WPWPWPWPx */aa"$aaaaH33<%k$S )+():