3 K^Q@sXddlmZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZmZmZmZmZdd lmZmZdd lmZmZmZddlm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-ddZ.Gddde Z/e Gddde/Z0Gddde0Z1Gddde2Z3dS))UnionN)reduce)wamp)public IdGeneratorObservableMixin)uri)message)types)role) exception)ApplicationError ProtocolErrorSessionNotReadySerializationErrorTypeCheckError) IPayloadCodecIAuthenticator)SessionDetails CloseDetailsEncodedPayload)PayloadExceededError) Publication SubscriptionHandler RegistrationEndpointPublishRequestSubscribeRequestUnsubscribeRequest CallRequestInvocationRequestRegisterRequestUnregisterRequestcCstj|ptj|S)N)inspectismethod isfunction)fr(:/tmp/pip-unpacked-wheel-p9ga6z5n/autobahn/wamp/protocol.pyis_method_or_function:sr*c@seZdZdZdZddZeddZeddZed d Z ed d Z ed dZ eddZ dddZ dddZddZdS) BaseSessionzi WAMP session base class. This class implements :class:`autobahn.wamp.interfaces.ISession`. NcCsttj|_|jdddddgdd|_i|_tjttj t i|_ d|_ d|_ d|_d|_d|_d|_d|_t|_dS) z joinleavereadyconnect disconnect)Z valid_eventsFN)txaioZ make_loggerlogZset_valid_events traceback_app_ecls_to_uri_patr INVALID_PAYLOADrPAYLOAD_SIZE_EXCEEDEDr _uri_to_ecls_realm _session_id_authid _authrole _authmethod _authprovider_payload_codecr_request_id_gen)selfr(r(r)__init__Fs&   zBaseSession.__init__cCs|jS)N)r8)r@r(r(r)realmrszBaseSession.realmcCs|jS)N)r9)r@r(r(r) session_idvszBaseSession.session_idcCs|jS)N)r:)r@r(r(r)authidzszBaseSession.authidcCs|jS)N)r;)r@r(r(r)authrole~szBaseSession.authrolecCs|jS)N)r<)r@r(r(r) authmethodszBaseSession.authmethodcCs|jS)N)r=)r@r(r(r) authproviderszBaseSession.authprovidercCsn|dkr8t|dst|j|j|<||j|jdj<n2t|d sHttj|tjjg|j|<||j|<dS)zM Implements :func:`autobahn.wamp.interfaces.ISession.define` N _wampurisr)hasattrAssertionErrorrHr4r7rPatternZURI_TARGET_HANDLER)r@r errorr(r(r)defines zBaseSession.definec Csd}t|drt|j}d}t|dr,|j}|rF|r>||d<nd|i}t|tjrnt|jt krf|jn|j}n$|j |j kr|j |j dj }nd}d} |j r|j jd|||} | rtj|||| j| j| j| jd} ntj|||||} | S) a Create a WAMP error message from an exception. :param request_type: The request type this WAMP error message is for. :type request_type: int :param request: The request ID this WAMP error message is for. :type request: int :param exc: The exception. :type exc: Instance of :class:`Exception` or subclass thereof. :param tb: Optional traceback. If present, it'll be included with the WAMP error message. :type tb: list or None Nargskwargs tracebackrzwamp.error.runtime_errorF)payloadenc_algoenc_keyenc_serializer)rIlistrNrO isinstancer r typerLstr __class__r4_urir>encoder ErrorrQrRrSrT) r@ request_typerequestexctbrRrNrOrLencoded_payloadmsgr(r(r)_message_from_exceptions@       z#BaseSession._message_from_exceptionc Csrd}d}|jr|js8d}|jj|ttj||jd}ny6t|j|j|j|j }|jj d|j |\}|_ |_ WnJtk r}z.|jjd|dttjdj||jd}WYdd}~Xnr2warnr ENC_NO_PAYLOAD_CODECrrQrTrSdecoderLrNrO ExceptionENC_DECRYPT_ERRORformatENC_TRUSTED_URI_MISMATCHr7 onUserErrorr1create_failurer rIrfrgrhri) r@rbr_enc_errlog_msgrareeZeclsr(r(r)_exception_from_messages~                  z#BaseSession._exception_from_message)N)NN)__name__ __module__ __qualname____doc__r2rApropertyrBrCrDrErFrGrMrcrvr(r(r(r)r+>s,       ;r+c@sBeZdZdZd:ddZeddZeddZed d Zed d Z ed;d dZ eddZ eddZ eddZ eddZddZddZddZeddZedd Zed!d"Zed#d$Zd%d&Zed'd(Zedd6d7Zd8d9ZdS)?ApplicationSessionz WAMP endpoint session. NcCs|tj||ptjdd|_tj|_d|_d|_ d|_ d|_ d|_ i|_ i|_i|_i|_i|_i|_i|_i|_i|_dS)zF Implements :func:`autobahn.wamp.interfaces.ISession` realm1)rBNF)r+rAr ComponentConfigconfigr ZDEFAULT_CLIENT_ROLES_session_roles _transportr9r8 _goodbye_sentZ_transport_is_closing _publish_reqs_subscribe_reqs_unsubscribe_reqs _call_reqs_register_reqs_unregister_reqs_subscriptions_registrations _invocations)r@rr(r(r)rA=s" zApplicationSession.__init__cCs |dkst|tst||_dS)zX Implements :func:`autobahn.wamp.interfaces.ISession.set_payload_codec` N)rVrrJr>)r@Z payload_codecr(r(r)set_payload_codec_sz$ApplicationSession.set_payload_codeccCs|jS)zX Implements :func:`autobahn.wamp.interfaces.ISession.get_payload_codec` )r>)r@r(r(r)get_payload_codecgsz$ApplicationSession.get_payload_codeccsL|_jd|}tj|dfddtj|fddfdddS)zV Implements :func:`autobahn.wamp.interfaces.ITransportHandler.onOpen` r/Ncs j|dS)NzWhile notifying 'connect')_swallow_error)fail)r@r(r)wsz+ApplicationSession.onOpen..cs tjjS)N)r1 as_future onConnect)_)r@r(r)r{scs j|dS)NzWhile calling 'onConnect')r)r)r@r(r)r|s)rfirer1 add_callbacks)r@ transportdr()r@r)onOpenns zApplicationSession.onOpencCs|j|jjdS)zP Implements :func:`autobahn.wamp.interfaces.ISession.onConnect` N)r,rrB)r@r(r(r)rszApplicationSession.onConnectc Cs|dkst|tkst|dks0t|tks0tt|tkrZx|D]} t| tksBtqBW|dksrt|tksrt|dkst|tkst|dkst|tkst|jrtd||_d|_t j ||j |||||||d } |j j | dS)zK Implements :func:`autobahn.wamp.interfaces.ISession.join` Nzalready joinedF) rBroles authmethodsrDrE authextra resumableresume_session resume_token)rWrXrJrUdictr9rmr8rr ZHellorrsend) r@rBrrDrErrrrrFrbr(r(r)r,s,   zApplicationSession.joincCs|jr|jjdS)zQ Implements :func:`autobahn.wamp.interfaces.ISession.disconnect` N)rclose)r@r(r(r)r0szApplicationSession.disconnectcCs |jdk S)zS Implements :func:`autobahn.wamp.interfaces.ISession.is_connected` N)r)r@r(r(r) is_connectedszApplicationSession.is_connectedcCs|jdk o|jdk S)zR Implements :func:`autobahn.wamp.interfaces.ISession.is_attached` N)rr9)r@r(r(r) is_attachedszApplicationSession.is_attachedcCsXt|dr6t|jtjr6|jjd|jj|jj dn|jj d|jj|t j |ddS)zR Implements :func:`autobahn.wamp.interfaces.ISession.onUserError` valuez{klass}.onUserError(): "{msg}")klassrbz*{klass}.onUserError(): "{msg}" {traceback})rrbrPN) rIrVrr r r2rjrYrw error_messagerLr1failure_format_traceback)r@rrbr(r(r)rqszApplicationSession.onUserErrorc CsDy|j||Wn.tk r>|jjdtjtjdYnXdS)a This is an internal generic error-handler for errors encountered when calling down to on*() handlers that can reasonably be expected to be overridden in user code. Note that it *cancels* the error, so use with care! Specifically, this should *never* be added to the errback chain for a Deferred/coroutine that will make it out to user code. zInternal error: {tb})r`N)rqrmr2rLr1rrr)r@rrbr(r(r)rs z!ApplicationSession._swallow_errorcsfdd}|S)z Does parameter type checking and validation against type hints and appropriately tells the user code and the caller (through router). cstjf||}g}xjjD]\}}||kr"t|ddtkrtt|||js|jdj ||j t ||j q"t|||s"|jdj ||j t ||j q"W|rt dj |tjf||IdHS)NZ __origin__z'{}' required={} got={}z, )r$ getcallargs__annotations__itemsgetattrrrVZ__args__appendrorwrWrr,r1r)rNrO argumentsresponsenamekind)funcr(r) _type_checks z2ApplicationSession.type_check.._type_checkr()r@rrr()rr) type_checks zApplicationSession.type_checkc"s jdkr ttjrPtjj}fdd}fdd}tj|||nttjrt j j jtjj }fdd}fdd}tj|||nbttj rt j jj}tjj|}fd d}fd d}tj|||ntd jj nttjrjs4tj}jj|d_t j j jtjj }fd d}fd d}tj||| nttjrLjjkr8xjjD]~}|jjp|j} jrxjsjj djddSy4t!j"jj#j$} jj%d| | \} _&_'Wn6t(k r\} zjj d| jddSd} ~ XnX| | krxjj ddSj)rj)fnt*} j&r| t*j&} j'rj'nt+}j,rt j-|j.j/j0j1| j2jj3d |j,<fdd}fdd}tjj4f| |}tj|||qWntdjj nttj5rj6j7krj7j8j6}t9j.|j:d}tj;|j<|ntdjj6 nrttj=r8j6j>kr$j>j8j6}jjkrgjj<t?j|j|j}jjj@|tj;|j<|ntdjj6nttjArj6jBkrjBj8j6}|jCjkrxj|jCD] }d|_Dq~Wj|jC=tj;|jjj6n|ttjo r d}jptjqj]k rj6jFk rjFj8j6j<}njptjrj]k rj6j7k rj7j8j6j<}njptjsj]k rBj6j>k rBj>j8j6j<}njptjtj]k rrj6jBk rrjBj8j6j<}n^jptjuj]k rj6jhk rjhj8j6j<}n.jptjvj]k rЈj6jmk rЈjmj8j6j<}| rtjw| s tjU|jxntd?jjpj6ntd@jjdS)AzY Implements :func:`autobahn.wamp.interfaces.ITransportHandler.onMessage` NcsD|dk r6jjdtjddj|}jj|dSjrDj_j _ j _ j _j_j_j_tjj j jjjjjjjtjdrjjndjjjd jd}tj |dfddtj |fd ddtj |dfd dtj |fd ddtj |dfd ddS) NzSession denied by onWelcomezwamp.error.cannot_authenticatez{0}peer_transport) rBsessionrDrErFrGr serializerrresumedrrr,cs j|dS)NzWhile notifying 'join')r)ru)r@r(r)rBsz?ApplicationSession.onMessage..success..cstjjS)N)r1ronJoin)r)detailsr@r(r)rGscs j|dS)NzWhile firing onJoin)r)ru)r@r(r)rMscs jdS)Nr.)r)r)r@r(r)rRscs j|dS)NzWhile notifying 'ready')r)ru)r@r(r)rXs)!r2infor AbortrorrrBr8rr9rDr:rEr;rFr<rGr=r _router_rolesrrZ _serializerZ SERIALIZER_IDrIrrrrrr1r)resreplyr)rbr@)rr)successsZ     z-ApplicationSession.onMessage..successcs$tjdd}jj|j|dS)Nzwamp.error.cannot_authenticatezError calling onWelcome handlerzWhile firing onWelcome)r rrrr)rur)r@r(r)rL[s z+ApplicationSession.onMessage..errorcs8jd}fdd}fdd}tj||||S)Nr-csS)Nr()r)argr(r) return_arglszAApplicationSession.onMessage..success..return_argcs j|dS)NzWhile firing 'leave' event)r)ru)r@r(r)_errorosz=ApplicationSession.onMessage..success.._error)rr1r)rrrr)rr@)rr)rhs   cs j|dS)NzWhile firing onLeave)r)ru)r@r(r)rtsz,ApplicationSession.onMessage.._errorcs^|dkrtdt|tkr&|jd}t|tkrDtdjt|tj|}jj |dS)Nz4onChallenge user callback did not return a signatureutf8z"signature must be unicode (was {})) rmrWbytesrlrXror Z Authenticaterr) signaturer)r@r(r)r}s    csvj|dtjddj|j}jj|tj|j |jt j j }fdd}fdd}t j ||||S)NzAuthentication failedzwamp.error.cannot_authenticatez{0}csjd|S)Nr-)r)r)rr@r(r)rsz.error..successcs j|dS)NzWhile firing onLeave)r)ru)r@r(r)rsz;ApplicationSession.onMessage..error.._error)rqr rrorrrr rreasonr1ronLeaver)rdrrrr)r@)rr)rLs   z8Received {0} message, and session is not yet establishedcsjd|S)Nr-)r)r)rr@r(r)rscsdjjj}j||S)Nz7While firing onLeave for reason "{0}" and message "{1}")rorr r)ruerrmsg)rbr@r(r)rszjreceived encoded payload with enc_algo={enc_algo}, but no payload codec active - ignoring encoded payload!)rRFzNfailed to decode application payload encoded with enc_algo={enc_algo}: {error})rLrRz-envelope topic URI does not match encoded one) publisherpublisher_authidpublisher_authroletopicretainedrRricsBjr>jdjr>jr2tjj}jj|n jj ddS)NZbrokerz}successfully processed event with acknowledged delivery, but could not send ACK, since the transport was lost in the meantime) Zx_acknowledged_deliveryrZx_acknowledged_event_deliveryrr Z EventReceived publicationrr2rj)rr)rbr@r(r)_successs  z.ApplicationSession.onMessage.._successcsdjjj}j||S)Nz&While firing {0} subscribed under {1}.)rofn subscriptionr)rur)handlerrbr@r(r)rs z5EVENT received for non-subscribed subscription ID {0}) was_encryptedz1PUBLISHED received for non-pending request ID {0}z2SUBSCRIBED received for non-pending request ID {0}rz4UNSUBSCRIBED received for non-pending request ID {0}z5received encoded payload, but no payload codec activeTz.failed to decrypt application payload 1: {err})rdz+failed to decrypt application payload 1: {}zXURI within encrypted payload ('{decrypted_proc}') does not match the envelope ('{proc}'))decrypted_procproczFURI within encrypted payload ('{}') does not match the envelope ('{}')zLcould not deliver progressive call result, because payload decryption failedcsj|ddS)NzWhile firing on_progress)rq)r)r@r(r)rXs)rfrgrhriz.RESULT received for non-pending request ID {0}z6INVOCATION received for request ID {0} already invokedz:INVOCATION received for non-registered registration ID {0}zrmr[r Yieldr^rQrSrTrr)rNrOraZ progress_msg)rbrr@r(r)rs& z.ApplicationSession.onMessage..progress)rcaller caller_authidcaller_authrole procedurerRc!s*jj=d}jrjs,d}jj|nny:t|tjrRjj d|j |j }njj d|g}Wn2t k r}zjjd|dWYdd}~XnX|rt|tjrt jj|j|j|j|j|j|j|j|jd }nt jj|j|j|j|jd}nHt|tjr0t jj|j |j |j|j|j|jd}nt jj|gd}jdkrjjjd jjjdSyjj|Wntk r}z8t jt jjjtj d jj|gd}jj|WYdd}~XnVt!k r$}z8t jt jjjtj"d jj|gd}jj|WYdd}~XnXdS) Nz7trying to send encrypted payload, but no keyring activeFz,failed to encrypt application payload: {err})rd)rQrRrSrTrfrgrhri)rQrRrSrT)rNrOrfrgrhri)rNzCSkipping result of "{}", request {} because transport disconnected.zNsuccess return value from invoked procedure "{0}" could not be serialized: {1}zSsuccess return value from invoked procedure "{0}" exceeds transport size limit: {1})#rr^rRr>r2rjrVr CallResultr[resultsZ kwresultsrmr rrQrSrTrfrgrhrirdebugrorrrr\ Invocation MESSAGE_TYPEr r5rr6)rrartrur)rbr registrationr@r(r)rsj        cs jj=tj|}yj||Wn YnXd}jrFtj|}jtj j j|j |j }yj j|Wntk r}z8tjtj j jtjdjj|gd}j j|WYdd}~XnVtk r}z8tjtj j jtjdjj|gd}j j|WYdd}~XnXdS)NzLerror return value from invoked procedure "{0}" could not be serialized: {1})rNzSsuccess return value from invoked procedure "{0}" exceeds transport size limit: {1})rr^r1Zfailure_messagerqr3rrcr rrrrRrrrr\r r5rorrr6)rdrZ formatted_tbrru)rbrr@r(r)rLCs4   z7INTERRUPT received for non-pending invocation {request})r^zr2rjrrQrTrSrlrNrOrmobjrr details_argZ EventDetailsrrrrrrirZ Publishedr^rpoprrresolveon_replyZ SubscribedrrrZ UnsubscribedrZsubscription_idactiveZResultrrr rkrnrproptionsZ on_progressrqrrrfrgrhrejectlenrrrrendpointrcrZreceive_progressZ CallDetailsrrrr!Z InterruptrcancelZ RegisteredrrZ UnregisteredKeyErrorrrZregistration_idr\r]CallPublish Subscribe UnsubscribeRegister Unregister is_calledrv)"r@rbrrrLr challengerrrraZ decoded_topicruZ invoke_argsZ invoke_kwargsrfutureZpublish_requestrr^Z call_requestrsrtrkwrNZprog_drrOrrrZinvokedregr()rrrbrrr@r) onMessages  C        0                     D'     zApplicationSession.onMessagecsd_jrbtjtjjdjjdtjj}fdd}fdd}tj |||d_tjj }fdd}fd d}tj |||dS) zW Implements :func:`autobahn.wamp.interfaces.ITransportHandler.onClose` Nz=WAMP transport was lost without closing the session {} before)rr csjd|S)Nr-)r)r)rr@r(r)rsz+ApplicationSession.onClose..successcs j|dS)NzWhile firing onLeave)r)ru)r@r(r)rsz*ApplicationSession.onClose.._errorcsjddS)Nr0)Z was_clean)r)r)r@wasCleanr(r)rscs j|dS)NzWhile firing onDisconnect)r)ru)r@r(r)rs) rr9r rZREASON_TRANSPORT_LOSTror1rrr onDisconnect)r@rrrrr()rr@rr)onCloses   zApplicationSession.onClosecCs tddS)zR Implements :func:`autobahn.wamp.interfaces.ISession.onChallenge` zBreceived authentication challenge, but onChallenge not implementedN)rm)r@rr(r(r)rszApplicationSession.onChallengecCsdS)zM Implements :func:`autobahn.wamp.interfaces.ISession.onJoin` Nr()r@rr(r(r)rszApplicationSession.onJoincCsdS)zP Implements :func:`autobahn.wamp.interfaces.ISession.onWelcome` Nr()r@rbr(r(r)rszApplicationSession.onWelcomecstjd}|j|j|j|j|j|jg}g}x"|D]}|j|j |j q0W|rf|j j dt |dx`|D]X|j jdjjjjdtjjstjj|tj|fddfddqlW|S)zB Errback any still outstanding requests with exc. Nz'Cancelling {count} outstanding requests)countzncleaning up outstanding {request_type} request {request_id}, firing errback on user handler {request_on_reply})Zrequest_on_reply request_idr]csjS)N)r)r)r^r(r)r.szBApplicationSession._errback_outstanding_requests..)r1create_future_successrrrrrrextendvaluesclearr2rrrrrrYrwrrr)r@r_rZ all_requestsZ outstandingrequestsr()r^r)_errback_outstanding_requests s2       "z0ApplicationSession._errback_outstanding_requestscsX|jtjkr"jjd|j|jdt|j|j}j|}fdd}tj ||||S)zN Implements :func:`autobahn.wamp.interfaces.ISession.onLeave` z/session closed with reason {reason} [{message}])rr csjrjdS)N)rr0)r)r@r(r)r0=sz.ApplicationSession.onLeave..disconnect) rrZREASON_DEFAULTr2rjr r r r1r)r@rr_rr0r()r@r)r1s   zApplicationSession.onLeavecCsb|jstd|js@|sd}tjj||d}|jj|d|_n |jj d|jdkp\|jj }|S)zL Implements :func:`autobahn.wamp.interfaces.ISession.leave` zsession hasn't joined a realmzwamp.close.normal)rr TzBsession was already requested to leave - not sending GOODBYE againN) r9rrrr rrrr2rj is_closed)r@rr rbrr(r(r)r-Cs  zApplicationSession.leavecCstj}|j|dS)zS Implements :func:`autobahn.wamp.interfaces.ISession.onDisconnect` N)r TransportLostr )r@r_r(r(r)rXszApplicationSession.onDisconnectc Os&t|tkst|dks,t|ttfks,t|dksDt|tksDttj|dj|j j dddd|j dd}|rt |t j rtd|jstj|jj}d}|jr|jjd|||}|r|rtj||f|j|j|j|jd|j}ntj|||j|j|j|jd}n8|r8tj||f||d |j}ntj||||d }|r|jdk rd|j|_|jdk rx|j|_|jdk r|j|_|j dk r|j |_ |r|j!rt"j#}t$|||dk d |j%|<nd}y|jj&|Wn<tk r } z||j%kr |j%|=| WYdd} ~ XnX|S) zP Implements :func:`autobahn.wamp.interfaces.IPublisher.publish` Nz {}.publish()F)r strictallow_empty_components allow_nonerz,options must be of type a.w.t.PublishOptionsT)rQrRrSrT)rNrO)r)'rWrXrJrUrrr check_or_raise_urirorYrwrrVr ZPublishOptionsrmrr rr?nextr>r[rrQrRrSrT message_attrcorrelation_idcorrelation_uricorrelation_is_anchorcorrelation_is_lastZ acknowledger1 create_futurerrr) r@rrNrOrrrarbrrur(r(r)publishcs|          zApplicationSession.publishFc s@t|r|dk s$t|dr | s$t|dksj |}t |||rN|j nd}t ||||j |<|rtj||f|j}n tj||}|r|jdk r|j|_|jdk r|j|_|jdk r|j|_|jdk r|j|_jj||S)Nz{}.subscribe()FT)r rrr)r rrorYrwr?rr1rrrrrrrrrrrrrr) rrrr check_typesrrZ handler_objrb)r@r(r) _subscribes2         z0ApplicationSession.subscribe.._subscriberrHZwildcard)matchexactT)consume_exceptions)callablerIrJrWrXrVr ZSubscribeOptionsrr rr$ getmembersrYr*__dict__Z is_handlerrrZuri_typerKZURI_TYPE_WILDCARDr _check_typesr1gather) r@rrrrr on_replieskrpatrZZsuboptsr()r@r) subscribes,$      zApplicationSession.subscribecCst|tst|jst|j|jks(t||j|jks.canceller)r,)(rWrXrJrUrrr rrorYrwrrVr Z CallOptionsrmrr rr?rr>r[r2ZfailurerrQrRrSrTrrrrrr1rr rr) r@rrNrOrrarbr,rr()rr@r)call%s            zApplicationSession.callc s$t|r|dk s$t|dr | s$t|dksj |}t |||rN|j nd}dk rhdj|}t ||||j |<|rtj||f|j}n tj||}|r|jdk r|j|_|jdk r|j|_|jdk r|j|_|jdk r|j|_jj||S)Nz {}.register()FT)r rrrz{}{})r rrorYrwr?rr1rrrrr"rrrrrrrrr) rrrrrrrZ endpoint_objrb)prefixr@r(r) _registers6          z.ApplicationSession.register.._registerrrHT)r )r!rIrJrWrXrVr ZRegisterOptionsrr rr$r"rYr*r#Z is_endpointrrrr$r1r%) r@rrrr.rr/r&r'rr(rZZregoptsr()r.r@r)registers&$"   zApplicationSession.registercCszt|tst|jst|j|jks(t|js6tj|j j }t j }t |||j|j|<tj||j}|jj||S)zT Called from :meth:`autobahn.wamp.protocol.Registration.unregister` )rVrrJrrrrr rr?rr1rr#rr rr)r@rrrrbr(r(r) _unregisters   zApplicationSession._unregister)N)NNNNNNN)NN)NNF)NNNF) rwrxryrzrArrrrrr,r0rrrqrrrrrrrr rr-rrr)r+r-r0r1r(r(r(r)r|7sV "     #    S $   &   [ D" b Cr|c@steZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZdS) _SessionShima shim that lets us present pep8 API for user-classes to override, but also backwards-compatible for existing code using ApplicationSession "directly". **NOTE:** this is not public or intended for use; you should import either :class:`autobahn.asyncio.wamp.Session` or :class:`autobahn.twisted.wamp.Session` depending on which async framework yo're using. NcCs |j|S)N)on_join)r@rr(r(r)rsz_SessionShim.onJoincCsv|jrjdd|jjDd}dd|jjDd }|j}|j|jjt|jj|pZd|p`d|dn|jdS) NcSsg|]}|jjddqS)rDN)_argsget).0xr(r(r) sz*_SessionShim.onConnect..rcSsg|]}|jjddqS)rEN)r4r5)r6r7r(r(r)r8srdefault)rrDrErr:) _authenticatorsr _merged_authextrar,rrBrUkeys on_connect)r@rDrErr(r(r)rs  z_SessionShim.onConnectc CsBy|j|j}Wn$tk r4tdj|jYnX|j||S)Nz.Received challenge for unknown authmethod '{}')r;rr RuntimeErrorroZ on_challenge)r@r authenticatorr(r(r)rsz_SessionShim.onChallengec Cs\|jdks|jdkrdSy|j|j}Wn$tk rLtdj|jYnX|j||jS)Nz.Received onWelcome for unknown authmethod '{}')rFr;rr?roZ on_welcomer)r@rbr@r(r(r)rsz_SessionShim.onWelcomecCs |j|S)N)on_leave)r@rr(r(r)r!sz_SessionShim.onLeavecCs|jS)N) on_disconnect)r@r(r(r)r$sz_SessionShim.onDisconnectcsttstjdkri_fdd}|d|dj}j}x>|jD]2\}}||krT|||krTtdj||||qTWjj <dS)NcsLtfddtjjgD}t|dkrHtdjdj|dS)Ncs g|]}|jkr|jqSr()r4)r6a)rr(r)r85szG_SessionShim.add_authenticator..at_most_one..rzInconsistent {}s: {} )setrUr;r r ValueErrorror,)runi)r@r@)rr) at_most_one3s  z3_SessionShim.add_authenticator..at_most_onerDrEz4Inconsistent authextra values for '{}': '{}' vs '{}') rVrrJr;rr<rrFror)r@r@rHrmergedr'vr()r@r@r)add_authenticator)s z_SessionShim.add_authenticatorcsHdd|jjDdd}t|t}fddfdd|DS) a: internal helper :returns: a single 'authextra' dict, consisting of all keys from any authenticator's authextra. Note that when the authenticator was added, we already checked that any keys it does contain has the same value as any existing authextra. cSsg|] }|jqSr()r)r6rCr(r(r)r8asz2_SessionShim._merged_authextra..cSs|t|jBS)N)rEr=)r7yr(r(r) extract_keyscsz4_SessionShim._merged_authextra..extract_keyscs0xD]}||kr||SqWtdj|dS)z for anything already in self._authenticators, we checked that it has the same value for any keys in its authextra -- so here we just extract the first one zNo values for '{}'N)rFro)r'r) authextrasr(r)first_value_forhs   z7_SessionShim._merged_authextra..first_value_forcsi|]}||qSr(r()r6r')rOr(r) vsz2_SessionShim._merged_authextra..)r;r rrE)r@rMZ unique_keysr()rNrOr)r<Vs   z_SessionShim._merged_authextracCsdS)Nr()r@rr(r(r)r3~sz_SessionShim.on_joincCs |jdS)N)r0)r@rr(r(r)rAsz_SessionShim.on_leavecCs|j|jjdS)N)r,rrB)r@r(r(r)r>sz_SessionShim.on_connectcCsdS)Nr()r@r(r(r)rBsz_SessionShim.on_disconnect)rwrxryrzr;rrrrrrrKr<r3rAr>rBr(r(r(r)r2s  -(r2c@s&eZdZdZeZdddZddZdS)ApplicationSessionFactoryz( WAMP endpoint session factory. NcCs|ptjdd|_dS)z :param config: The default component configuration. :type config: instance of :class:`autobahn.wamp.types.ComponentConfig` r})rBN)r r~r)r@rr(r(r)rAsz"ApplicationSessionFactory.__init__cCs|j|j}||_|S)z Creates a new WAMP application session. :returns: -- An instance of the WAMP application session class as given by `self.session`. )rrfactory)r@rr(r(r)__call__s z"ApplicationSessionFactory.__call__)N)rwrxryrzr|rrArSr(r(r(r)rQs rQ)4typingrr1r$ functoolsrZautobahnrZ autobahn.utilrrrZ autobahn.wamprr r r r Zautobahn.wamp.exceptionr rrrrZautobahn.wamp.interfacesrrZautobahn.wamp.typesrrrZautobahn.exceptionrZautobahn.wamp.requestrrrrrrrrr r!r"r#r*r+r|r2objectrQr(r(r(r)sD         8z 4/