o >hD>@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 z/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/autobahn/asyncio/component.py 3s"z _unique_list..)set)seqr rr _unique_list.srcCs.|d}|dddd|ddDS)N_rcss|]}|VqdSN) capitalize)rsr r r 8sz._camel_case_from_snake_case..)splitjoin)snakepartsr 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. websocket)url serializersproxy rawsocketr) serializerFshould not arrive herez"Unknown {} transport option: {}={}Nr )typerrr$r&r r%roptionsitemssetProtocolOptions TypeErrorKeyErrorr" ValueErrorformat)loop transportsession_factoryr%factoryr(kvr r r_create_transport_factory;s8     r8c@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. ) 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)r9dictboolr: SSLContextr0)r<endpointr?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.on_connect_success..lost) is_closingrjrkgetattrrlrconnection_lostr)resultr3protoreasonrp)rc)roron_connect_successs    z=Component._wrap_connection_future..on_connect_successcsjd7_t|dS)Nr)connect_failuresrjrl)errrcr3r ron_connect_failuresz=Component._wrap_connection_future..on_connect_failure)rj add_callbacks)r<r3rcconn_frwr{r rzrr[s (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)r2)logwarnrXget_event_loop_start)r<r2r r rstart$s  zComponent.startr)__name__ __module__ __qualname____doc__rj make_loggerr~rr4r>rDrhr[rr 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)rtrXCancelledErrordebug Exceptionerrorstop)futr=r~r2r rcancel_all_callbackys   z5run..nicely_exit..cancel_all_callback) rrXTask all_tasksAttributeError current_taskcancelgatheradd_done_callback)rtaskstaskrrrrr r nicely_exites&     zrun..nicely_exitctdS)NSIGINTrXrYr rr rzrun..cr)NSIGTERMrr rr rrrcSs |dSr)r)r2argr r r done_callbacks zrun..done_callback)rj start_loggingrXr is_closedset_event_loopnew_event_loopconfigr2radd_signal_handlerrrrNotImplementedErrorr_run run_foreverrclose) components start_loop log_levelrr )r~r2rrr 9s2  !   r )Tr)rXr:r functoolsrrjautobahn.asyncio.websocketrautobahn.asyncio.rawsocketr autobahn.wamprautobahn.wamp.exceptionrautobahn.asyncio.wamprautobahn.wamp.serializerrr __all__rr"r8r r r r r rs$      'X