o ;i!) @sBddlmZmZmZmZmZmZddlmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZmZddlmZerXdd lmZmZdd lmZmZdd lmZGd d d Z ede dZ!Gddde"Z#ddZ$dee dee ge ffddZ%ee ge e e dffZ&GdddZ'Gddde"Z(dddee!dee!ge e e dffddfd d!Z)Gd"d#d#e e#d$Z*Gd%d&d&e e#d$Z+Gd'd(d(e'e(d$Z,Gd)d*d*e e#d$Z-Gd+d,d,e e#d$Z.Gd-d.d.e e#d$Z/Gd/d0d0e,Z0Gd1d2d2e e#d$Z1Gd3d4d4e e#d$Z2Gd5d6d6e e#d$Z3Gd7d8d8e,Z4dS)9) TYPE_CHECKINGTypeTypeVarTuple FrozenSetDict)OptionalCallableAny CollectionList Coroutine)chain) defaultdict)Status)Deadline _Metadata)get_annotations) _SendType _RecvType) IEventsTargetIServerMethodFunc)PeercsteZdZUdZdZeeed<eZ e eed<e ed<fddZ ded e d d ffd d Zdfdd ZZS)_Event)__interrupted__ __payload__ __readonly__rc sNt|t|jksJ|jtdd|D] \}}t||qdS)NrF)len __slots__super __setattr__items)selfkwargskeyvalue __class__rB/home/ubuntu/.local/lib/python3.10/site-packages/grpclib/events.py__init__s z_Event.__init__r&r'returnNcs*||jvr td|t||dS)NzRead-only property: {!r})rAttributeErrorformatr!r")r$r&r'r(rr*r"s z_Event.__setattr__cstdddS)NrT)r!r")r$r(rr* interrupt#sz_Event.interruptr,N)__name__ __module__ __qualname__r rr str__annotations__ frozensetrrboolr+r r"r/ __classcell__rrr(r*rs  r _EventType)boundceZdZfddZZS) _EventMetacsXt|}|dp dtdd|D|d<tfdd|D|d<t||||S)Nrrcss|]}|VqdSNr.0namerrr* /sz%_EventMeta.__new__..r c3s|] }|vr|VqdSr=rr>payloadrr*rA0s r)rgettupler6r!__new__)mcsr@basesparams annotationsr(rBr*rF,s z_EventMeta.__new__r1r2r3rFr8rrr(r*r<*r<cs|Sr=r)args_rrr*_ident5srO event_typer,csdtdtffdd }|S)Nfuncr,cs |_|Sr=)__dispatches__)rQrPrr* decorator:sz_dispatches..decorator)r )rPrTrrSr* _dispatches9srUNc@s\eZdZUiZeeeefed<d ddZ deede ddfdd Z d ede fd d Z dS) _Dispatch__dispatch_methods__r,NcCs(tt|_|jD]}t|j|<q dSr=)rlist _listenersrWvaluesrO__dict__)r$r@rrr*r+Fs  z_Dispatch.__init__rPcallbackcCs(|j|j|d|j||dSr=)r[poprWrYappend)r$rPr\rrr* add_listenerKsz_Dispatch.add_listenereventcsD|jjD]}|IdHjrnqtfddjDS)Nc3s|]}t|VqdSr=)getattrr>r`rr*rAXsz)_Dispatch.__dispatch__..)rYr)rrEr)r$r`r\rrbr* __dispatch__Ssz_Dispatch.__dispatch__r0)r1r2r3rWrrrr4r5r+ _Callbackr_r rcrrrr*rVCs   rVcr;) _DispatchMetacsttdd|D}|D](\}}t|dd}|dur8t|tr(t|ts,J|||vs4J||||<q||d<t ||||S)Ncss |] }t|diVqdS)rWN)rar#)r?baserrr*rA^s  z(_DispatchMeta.__new__..rRrW) dictr from_iterabler#ra isinstancetype issubclassrr!rF)rGr@rHrIdispatch_methodsr&r' dispatchesr(rr*rF]s    z_DispatchMeta.__new__rKrrr(r*re[rLretargetrr\cCs|j||dS)zRegisters a listener function for the given target and event type .. code-block:: python3 async def callback(event: SomeEvent): print(event.data) listen(target, SomeEvent, callback) N)rcr_)rnrPr\rrr*listenmsroc@eZdZUdZdZeed<dS) SendMessagezfDispatches before sending message to the other party :param mutable message: message to send messagersNr1r2r3__doc__rr r5rrrr*rq~  rq) metaclassc@rp) RecvMessagezmDispatches after message was received from the other party :param mutable message: received message rrrsNrtrrrr*rxrvrxc@sHeZdZeedddedfddZeedddedfddZd S) _DispatchCommonEventsrsrr,c|t|dIdHSNrr)rcrqr$rsrrr* send_messagez"_DispatchCommonEvents.send_messagercrzr{)rcrxr|rrr* recv_messager~z"_DispatchCommonEvents.recv_messageN) r1r2r3rUrqrr}rxrrrrr*rys ryc@sVeZdZUdZdZeed<ded<eed<ee ed<eed<eeed <d ed <d S) RecvRequesta'Dispatches after request was received from the client :param mutable metadata: invocation metadata :param mutable method_func: coroutine function to process this request, accepts :py:class:`~grpclib.server.Stream` :param read-only method_name: RPC's method name :param read-only deadline: request's :py:class:`~grpclib.metadata.Deadline` :param read-only content_type: request's content type :param read-only user_agent: request's user agent :param read-only peer: request's :py:class:`~grpclib.protocol.Peer` )metadata method_funcrrr method_namedeadline content_type user_agentrpeerN r1r2r3rurrr5r4rrrrrr*rs     rc@rp)SendInitialMetadatazyDispatches before sending headers with initial metadata to the client :param mutable metadata: initial metadata rrNr1r2r3rurrr5rrrr*rrvrc@:eZdZUdZdZeed<eed<ee ed<e ed<dS)SendTrailingMetadataa.Dispatches before sending trailers with trailing metadata to the client :param mutable metadata: trailing metadata :param read-only status: status of the RPC call :param read-only status_message: description of the status :param read-only status_details: additional status details rrstatusstatus_messagestatus_detailsN r1r2r3rurrr5rrr4r rrrr*r   rc@seZdZeededddedeededeedd d e edffd d Z ee ded e efd dZ ee dededeeded e ef ddZdS)_DispatchServerEventsrrrrrrrrrr,c s$|t|||||||dIdHS)N)rrrrrrr)rcr)r$rrrrrrrrrr* recv_requests z"_DispatchServerEvents.recv_requestcrzNr)rcrr$rrrr*send_initial_metadataz+_DispatchServerEvents.send_initial_metadatarrrc|t||||dIdHSN)rrrr)rcrr$rrrrrrr*send_trailing_metadata z,_DispatchServerEvents.send_trailing_metadataN)r1r2r3rUrrr4rrrrrrrrr rrrrr*rsJ   rc@s:eZdZUdZdZeed<eed<ee ed<eed<dS) SendRequesta$Dispatches before sending request to the server :param mutable metadata: invocation metadata :param read-only method_name: RPC's method name :param read-only deadline: request's :py:class:`~grpclib.metadata.Deadline` :param read-only content_type: request's content type rrrrrNrrrrr*rrrc@rp)RecvInitialMetadatazDispatches after headers with initial metadata were received from the server :param mutable metadata: initial metadata rrNrrrrr*rs  rc@r)RecvTrailingMetadataa9Dispatches after trailers with trailing metadata were received from the server :param mutable metadata: trailing metadata :param read-only status: status of the RPC call :param read-only status_message: description of the status :param read-only status_details: additional status details rrrrrNrrrrr*rs   rc @seZdZeedededeedede ef ddZ ee dede efdd Z ee ded ed eed ede ef d dZdS)_DispatchChannelEventsrrrrr,cr)N)rrrr)rcr)r$rrrrrrr* send_request2rz#_DispatchChannelEvents.send_requestcrzr)rcrrrrr*recv_initial_metadataBrz,_DispatchChannelEvents.recv_initial_metadatarrrcrr)rcrrrrr*recv_trailing_metadataKrz-_DispatchChannelEvents.recv_trailing_metadataN)r1r2r3rUrrr4rrrrrrrrr rrrrr*r0s>  r)5typingrrrrrrrr r r r r itertoolsr collectionsrconstrrrr_compatrstreamrr_typingrrprotocolrrr9rjr<rOrUrdrVrerorqrxryrrrrrrrrrrrr*sN            2