o iD>@sddlZddlZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZd Zd d Zd d ZddZGddde jZdddZdS)N)wraps)WampWebSocketClientFactory)WampRawSocketClientFactory) component) TransportLost)Session)create_transport_serializerscreate_transport_serializer) Componentruncstfdd|DS)zM Return a list with unique elements from sequence, preserving order. cs"g|] }|vr|s|qS)add).0xseenr `C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\autobahn/asyncio/component.py 3s"z _unique_list..N)set)seqr rr _unique_list.srcCs.|d}|dddd|ddDS)N_rcss|]}|VqdSN) capitalize)rsr r r 8sz._camel_case_from_snake_case..)splitjoin)Zsnakepartsr r r_camel_case_from_snake_case6s $r!c Cs|jdkrt|}t||j||jd}n|jdkr't|jd}t||d}nJd|j D]=\}}z |j d i||iWq0t t fymz|j d it ||iWnt t fyjtd|j||wYq0w|S) z3 Create a WAMP-over-XXX transport factory. Z websocket)url serializersproxyZ rawsocketr) serializerFshould not arrive herez"Unknown {} transport option: {}={}Nr )typerrr"r$r r#roptionsitemsZsetProtocolOptions TypeErrorKeyErrorr! ValueErrorformat)loop transportsession_factoryr#factoryr%kvr r r_create_transport_factory;s8     r4c@sHeZdZdZeZeZ ddZ ddZ ddZ dd Z d d d Z d S)r z A component establishes a transport and attached a session to a realm using the transport for communication. The transports a component tries to use can be configured, as well as the auto-reconnect strategy. cCs t|tjS)z" Internal helper. N) isinstancesslSSLError)selfer r r _is_ssl_errorrs zComponent._is_ssl_errorcCsPt|tr$d|vr"|d}t|ttfrdSt|tjrdStddStd)Ntlsz?'tls' configuration must be a dict, bool or SSLContext instancezI'endpoint' configuration must be a dict or IStreamClientEndpoint provider)r5dictboolr6 SSLContextr,)r8endpointr;r r r_check_native_endpointxs   z Component._check_native_endpointcCst|||}|jr>|jdd}t|tkrtdt||j||jd|jdd}t t j ||d}| |||S|jdd kr|jd d } | d vrXtd | |jd} t| t krltdt| |jd} t| tkrtdt| |jdd}t|tkrtdt||jdd} d} | rt| tr| D] }|dvrtd|q| d| }t|t krtd|| dd}|} d} |durtjtjj|d} nt| tjr| } n| dvr| r| } n tdt| |j|| | | | d}t t j ||d}| |||S|jddkrG|jd}t|jdd}|j||d}t t j ||d}| |||SJd )!z? Create and connect a WAMP-over-XXX transport. timeout z RuntimeErrorcreate_unix_connection)r8r.r/r0doner1rAfZtime_frFrCrDr;Z tls_hostnamer2rIZ cert_fnamerOr r r_connect_transports              zComponent._connect_transportcs:fdd}fdd}t||dt|d||S)Ncs`|\}}|rtst|dd}tt|dS|jtfdd}||_dS)NZ_onclose_reasonzConnection already closedcs2|}ts|durtd}t||S)Nzfailed to complete connection)txaio is_calledrreject)ZfailZrtn)r^origr rlost s   zKComponent._wrap_connection_future..on_connect_success..lost) is_closingrarbgetattrrcrconnection_lostr)resultr/protoreasonre)r^)rdron_connect_successs    z=Component._wrap_connection_future..on_connect_successcsjd7_t|dS)Nr)Zconnect_failuresrarc)errr^r/r ron_connect_failuresz=Component._wrap_connection_future..on_connect_failure)raZ add_callbacks)r8r/r^Zconn_frlror rnrrVs (z!Component._wrap_connection_futureNcCs(|dur|jdt}|j|dS)a~ This starts the Component, which means it will start connecting (and re-connecting) to its configured transports. A Component runs until it is "done", which means one of: - There was a "main" function defined, and it completed successfully; - Something called ``.leave()`` on our session, and we left successfully; - ``.stop()`` was called, and completed successfully; - none of our transports were able to connect successfully (failure); :returns: a Future which will resolve (to ``None``) when we are "done" or with an error if something went wrong. NzUsing default loop)r.)logwarnrSget_event_loop_start)r8r.r r rstart$s  zComponent.startr)__name__ __module__ __qualname____doc__ra make_loggerrprr0r:r@r`rVrtr r r rr bsb6r Tinfocs|dur tj|dtr!ttttj_t fddz t j fdd t j fddWn tyMYnwdd }t|||rrzWn tjykYnwdSdS) a. High-level API to run a series of components. This will only return once all the components have stopped (including, possibly, after all re-connections have failed if you have re-connections enabled). Under the hood, this calls XXX fixme for asyncio -- if you wish to manage the loop yourself, use the :meth:`autobahn.asyncio.component.Component.start` method to start each component yourself. :param components: the Component(s) you wish to run :type components: instance or list of :class:`autobahn.asyncio.component.Component` :param start_loop: When ``True`` (the default) this method start a new asyncio loop. :type start_loop: bool :param log_level: a valid log-level (or None to avoid calling start_logging) :type log_level: string N)levelc sjd|dztj}Wn tyt}Ynw|D]}ztj}Wn ty5t}Ynw||ur>|qfdd}tj|}||dS)NzShutting down due to {signal})signalc szCz|Wn'tjydYn ty.}z jd|dWYd}~nd}~wwWdSWdSWdSw)NzAll task cancelledz&Error while shutting down: {exception}) exception)rirSCancelledErrordebug Exceptionerrorstop)futr9rpr.r rcancel_all_callbackys   z5run..nicely_exit..cancel_all_callback) rzrSTask all_tasksAttributeError current_taskcancelgatheradd_done_callback)r|taskstaskrrrrr r nicely_exites&     zrun..nicely_exitctdS)NSIGINTrSrTr rr rzrun..cr)NSIGTERMrr rr rrrcSs |dSr)r)r.argr r r done_callbacks zrun..done_callback)raZ start_loggingrSrr is_closedset_event_loopnew_event_loopconfigr.ryadd_signal_handlerr|rrNotImplementedErrorr_run run_foreverr~close) componentsZ start_loop log_levelrr )rpr.rrr 9s2  !   r )Trz)rSr6r| functoolsrraZautobahn.asyncio.websocketrZautobahn.asyncio.rawsocketrZ autobahn.wamprZautobahn.wamp.exceptionrZautobahn.asyncio.wamprZautobahn.wamp.serializerrr __all__rr!r4r r r r r rs$      'X