a dP@sdZddlZddlZddlmZmZmZmZmZddl Z ddl m Z ddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl#m%Z%ddl#m&Z&ddl#m'Z'ddl#m(Z(ddl#m)Z)ddl*m+Z+d ,e j-Z.ej/dd!kreej0d"d#d$Z1neej0d"d%d$Z1e$ee j2d&d'd(Z3Gd)d*d*Z4Gd+d,d,e4ej5Z5Gd-d.d.e4ej6Z6Gd/d0d0e4ej7Z7Gd1d2d2e4ej8Z8Gd3d4d4ej9Z9d;e:ee$ee j2eeed5d6d7Z;dr?r@r*r*r+__init__bs z_BaseMultiCallable.__init__)metadatar.r%cCs"|pt}|rtt||}|S)zBased on the provided values for or initialise the final metadata, as it should be used for the current call. )rrZaugment_metadata)rCr.r*r*r+_init_metadatats   z!_BaseMultiCallable._init_metadata)NN)__name__ __module__ __qualname____doc__r'AbstractEventLoop__annotations__r AioChannelbytesr!rrrrrrrB staticmethodrgrpc CompressionrDr*r*r*r+r2Rs4       r2c @sVeZdZddddddeeeeeeejee eej e j ee fdddZdS)UnaryUnaryMultiCallableNtimeoutrC credentialswait_for_readyr.requestrRrCrSrTr.r%c Csh|||}|jscCsg|_g|_g|_g|_|dur|D]}t|tr@|j|q$t|trX|j|q$t|trp|j|q$t|t r|j|q$t d |d tj d tj d tj d t j q$t |_t t|t||||j|_dS)aConstructor. Args: target: The target to which to connect. options: Configuration options for the channel. credentials: A cygrpc.ChannelCredentials or None. compression: An optional value indicating the compression method to be used over the lifetime of the channel. interceptors: An optional list of interceptors that would be used for intercepting any RPC executed with that channel. NzInterceptor {} must be z{} or z{}. )rdrerfrg isinstancerappendrrr ValueErrorformatrEr Zget_working_loopr3rKrencoder1r4)rArhrirSr.r>Z interceptorr*r*r+rBAs@         zChannel.__init__cs|Sr&r*rAr*r*r+ __aenter__tszChannel.__aenter__cs|dIdHdSr&_close)rAexc_typeexc_valexc_tbr*r*r+ __aexit__wszChannel.__aexit__c sH|jrdS|jt}g}g}|D]}z|jdd}Wn@ty~}z(dt|vrhWYd}~q*nWYd}~n d}~00|sq*|d}|jd} | r*t | t j r*t | dr| j|jurq*n,t | dr| j j|jurq*ntd| || ||q*|r&|r&tj||d IdH|D]} | q*|jdS) Nr )limitframerrAr4 _cython_callzUnrecognized call object: )rR)r4closedclosingr,Z get_stackAttributeErrorstrf_localsgetrjr Callhasattrryr Z InternalErrorrkr'waitcancelclose) rAgraceZtasksZcallsZ call_tasksZtaskstackZattribute_errorrx candidaterXr*r*r+rrzsF            zChannel._closeN)rcs||IdHdSr&rq)rArr*r*r+rsz Channel.closecCs"t|dr|js|jdS)Nr4)rr4rzrror*r*r+__del__s  zChannel.__del__F)try_to_connectr%cCs|j|}tj|Sr&)r4Zcheck_connectivity_staterZ1CYGRPC_CONNECTIVITY_STATE_TO_CHANNEL_CONNECTIVITY)rArresultr*r*r+ get_states zChannel.get_state)last_observed_stater%cs"|j|jddIdHsJdS)Nr)r4Zwatch_connectivity_statevalue)rArr*r*r+wait_for_state_changes zChannel.wait_for_state_changer$cs:|jdd}|tjjkr6||IdH|jdd}q dS)NT)r)rrNChannelConnectivityZREADYr)rAstater*r*r+ channel_readys  zChannel.channel_ready)r;r<r=r%cCs"t|jt||||j|g|jSr&)rPr4rrnrdr3rAr;r<r=r*r*r+ unary_unaryszChannel.unary_unarycCs"t|jt||||j|g|jSr&)r]r4rrnrer3rr*r*r+ unary_streamszChannel.unary_streamcCs"t|jt||||j|g|jSr&)r^r4rrnrfr3rr*r*r+ stream_unaryszChannel.stream_unarycCs"t|jt||||j|g|jSr&)rbr4rrnrgr3rr*r*r+ stream_streamszChannel.stream_stream)N)F)NN)NN)NN)NN)+rErFrGr'rIrJr rKrrrrrr}rrrNChannelCredentialsrOrrrBrprvrrrZrrr\rrrrr!rrPrr]rr^rrbrr*r*r*r+rc9st        3H      rcrhrir.r>cCst||durdn|d||S)aCreates an insecure asynchronous Channel to a server. Args: target: The server address options: An optional list of key-value pairs (:term:`channel_arguments` in gRPC Core runtime) to configure the channel. compression: An optional value indicating the compression method to be used over the lifetime of the channel. interceptors: An optional sequence of interceptors that will be executed for any call executed with this channel. Returns: A Channel. Nr*)rcrr*r*r+insecure_channelsrrhrSrir.r>cCst||durdn||j||S)a<Creates a secure asynchronous Channel to a server. Args: target: The server address. credentials: A ChannelCredentials instance. options: An optional list of key-value pairs (:term:`channel_arguments` in gRPC Core runtime) to configure the channel. compression: An optional value indicating the compression method to be used over the lifetime of the channel. interceptors: An optional sequence of interceptors that will be executed for any call executed with this channel. Returns: An aio.Channel. Nr*)rcZ _credentialsrr*r*r+secure_channel<sr)NNN)NNN)>rHr'systypingrrrrrrNrrr Z grpc._cythonr r r Z_callrrrrZ _interceptorrrrrrrrrr _metadatar_typingrrrrr r!_utilsr"rm __version__r/ version_infor(r,rOr1r2rPr]r^rbrcr}rrrr*r*r*r+s                             2 - . - -i