U #i@sddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z ddl Z ddl mZddlmZddlmZmZmZmZddlmZmZmZed d d ZGd d d ZdS)N) defaultdict)wraps)Any DefaultDictDictList)Channel)NotConnectedError)HEARTBEAT_PAYLOADPHOENIX_CHANNEL ChannelEventsMessage)Callback T_ParamSpecT_Retvalfunccs&ttjtjtdfdd }|S)N)argskwargsreturncs|djstj||S)Nr) connectedr __name__)rrr7/tmp/pip-unpacked-wheel-935k1slo/realtime/connection.pywrappers  z"ensure_connection..wrapper)rrrrr)rrrrrensure_connectionsrc@seZdZdidfeeeeefeddddZe dddd Z ddd d Z ddd d Z ddddZ e ddddZddddZddddZe eedddZddddZdS)SocketFN)urlauto_reconnectparams hb_intervalrcCs@||_tt|_d|_||_||_|d|_||_tt|_dS)a `Socket` is the abstraction for an actual socket connection that receives and 'reroutes' `Message` according to its `topic` and `event`. Socket-Channel has a 1-many relationship. Socket-Topic has a 1-many relationship. :param url: Websocket URL of the Realtime server. starts with `ws://` or `wss://` :param params: Optional parameters for connection. :param hb_interval: WS connection is kept alive by sending a heartbeat message. Optional, defaults to 5. FN) rrlistchannelsrr r!Z kept_aliver)selfrrr r!rrr__init__ s zSocket.__init__)rcCs&t}|t||dS)z Wrapper for async def _listen() to expose a non-async interface In most cases, this should be the last method executed as it starts an infinite listening loop. :return: None N)asyncioget_event_looprun_until_completeZgather_listen _keep_aliver$Zlooprrrlisten:sz Socket.listencszr|jIdH}tft|}|jtjkr2Wq|j |j gD],}|j D] }|jd|jfkrL| |j qLqBWqtjjk r|jrtd|IdH|jD] \}}|D]}|IdHqqntdYqYqXqdS)zN An infinite loop that keeps listening. :return: None N*5Connection with server closed, trying to reconnect..."Connection with the server closed.) ws_connectionrecvr jsonloadseventr Zreplyr#gettopicZ listenerscallbackpayload websockets exceptionsConnectionClosedrlogginginfo_connectitemsZ_join exception)r$msgZchannelZclr6r#rrrr)Ds(   zSocket._listencCs t}||d|_dS)zR Wrapper for async def _connect() to expose a non-async interface TN)r&r'r(r>rr+rrrconnectbszSocket.connectcs<t|jIdH}|jr0td||_d|_ntddS)NzConnection was successfulTzConnection Failed) r9rBropenr<r=r0r Exception)r$r0rrrr>js  zSocket._connectcCs t}||d|_dS)zP Wrapper for async def _close() to expose a non-async interface FN)r&r'r(_closerr+rrrclosetsz Socket.closecs|jIdHdS)N)r0rF)r$rrrrE}sz Socket._closecsz@tttjtdd}|jt|IdHt |j IdHWqt j jk r|jrttd|IdHntdYqYqXqdS)zx Sending heartbeat to server every 5 seconds Ping - pong messages to verify connection is alive N)r6r4r8refr.r/)dictr r Z heartbeatr r0sendr2dumpsr&sleepr!r9r:r;rr<r=r>r@)r$datarrrr*s" zSocket._keep_alive)r6rcCs"t|||j}|j|||S)z :param topic: Initializes a channel and creates a two-way association with the socket :return: Channel )rr r#append)r$r6chanrrr set_channelszSocket.set_channelcCsD|jD]4\}}|D]&}td|ddd|jDdqq dS)zd Prints a list of topics and event the socket is listening to :return: None zTopic: z | Events: cSsg|] \}}|qSrr).0e_rrr sz"Socket.summary..]N)r#r?print callbacks)r$r6ZchansrNrrrsummaryszSocket.summary)r __module__ __qualname__strboolrrintr%rr,r)rBr>rFrEr*rrOrWrrrrrs,     r)r&r2r< collectionsr functoolsrtypingrrrrr9Zrealtime.channelrZrealtime.exceptionsr Zrealtime.messager r r r Zrealtime.typesrrrrrrrrrs