B &dN@sddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZddlTd gZdaedd d d ZGd ddZGdddeZGdddeZGdddZGdd d ZdS)N)CallableAny)_logging)ABNF) parse_url) WebSocketgetdefaulttimeout)* WebSocketApp)reconnectIntervalreturncCs|adS)N) RECONNECT)r r[/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/websocket/_app.py setReconnect)src@sJeZdZdZeepeddddZeeddddZ eedd d d Z dS) DispatcherBasez DispatcherBase N)app ping_timeoutr cCs||_||_dS)N)rr)selfrrrrr__init__2szDispatcherBase.__init__)secondscallbackr cCst||dS)N)timesleep)rrrrrrtimeout6s zDispatcherBase.timeout)r reconnectorr c Csny4tdj|ttdt||ddWn4tk rh}ztdj|dWdd}~XYnXdS)NzQreconnect() - retrying in {seconds_count} seconds [{frame_count} frames in stack])Z seconds_count frame_countT) reconnectingzUser exited {err})err) rinfoformatleninspectstackrrKeyboardInterrupt)rrrerrr reconnect:s zDispatcherBase.reconnect) __name__ __module__ __qualname____doc__rintfloatrrrr'rrrrr.src@s$eZdZdZeeeddddZdS) Dispatcherz Dispatcher N)sock read_callbackcheck_callbackr cCsTxN|jjrNt}||jjjtj||j}|r>|s>P|| qWdS)N) r keep_running selectorsDefaultSelectorregisterr/ EVENT_READselectrclose)rr/r0r1selrrrrreadHs  zDispatcher.read)r(r)r*r+socketrr;rrrrr.Dsr.c@s2eZdZdZeeeddddZedddZdS) SSLDispatcherz SSLDispatcher N)r/r0r1r cCs,x&|jjr&|}|r|sP|qWdS)N)rr2r7)rr/r0r1r:rrrr;Ys  zSSLDispatcher.read)r cCs^|jjj}|r|gSt}||tj||j}| t |dkrZ|ddSdS)Nr) rr/pendingr3r4r5r6r7rr8r")rr/r9r:rrrr7as   zSSLDispatcher.select) r(r)r*r+r<rr;listr7rrrrr=Usr=c@s^eZdZdZepeeddddZee e ddddZ ee dd d d Z ee dd d dZ dS)WrappedDispatcherz WrappedDispatcher N)r dispatcherr cCs$||_||_||_|d|jdS)N)rrrAsignalabort)rrrrArrrrtszWrappedDispatcher.__init__)r/r0r1r cCs&|j|||jo ||j|dS)N)rAr;rr)rr/r0r1rrrr;zszWrappedDispatcher.read)rrr cCs|j||dS)N)rAr)rrrrrrr~szWrappedDispatcher.timeout)rrr cCs|||dS)N)r)rrrrrrr'szWrappedDispatcher.reconnect) r(r)r*r+r,r-r.rr<rr;rr'rrrrr@ps r@c@s eZdZdZd"eepeeeeeeeeeeeeee ddddZ e j fee dddd Zdd d d Zdd d dZdd ddZdd ddZd#eee pee peeee peeee peeeeeeee edddZd$e eeedddZe edddZeeddd d!ZdS)%r z_ Higher level of APIs are provided. The interface is like JavaScript WebSocket object. NT)urlheaderon_open on_messageon_erroron_closeon_pingon_pongon_cont_messager2 get_mask_keycookie subprotocolson_datar<r cCs||_|dk r|ng|_| |_||_||_||_||_||_||_||_ | |_ d|_ | |_ d|_ d|_d|_d|_d|_d|_d|_d|_| |_||_d|_dS)a WebSocketApp initialization Parameters ---------- url: str Websocket url. header: list or dict Custom header for websocket handshake. on_open: function Callback object which is called at opening websocket. on_open has one argument. The 1st argument is this class object. on_message: function Callback object which is called when received data. on_message has 2 arguments. The 1st argument is this class object. The 2nd argument is utf-8 data received from the server. on_error: function Callback object which is called when we get error. on_error has 2 arguments. The 1st argument is this class object. The 2nd argument is exception object. on_close: function Callback object which is called when connection is closed. on_close has 3 arguments. The 1st argument is this class object. The 2nd argument is close_status_code. The 3rd argument is close_msg. on_cont_message: function Callback object which is called when a continuation frame is received. on_cont_message has 3 arguments. The 1st argument is this class object. The 2nd argument is utf-8 string which we get from the server. The 3rd argument is continue flag. if 0, the data continue to next frame data on_data: function Callback object which is called when a message received. This is called before on_message or on_cont_message, and then on_message or on_cont_message is called. on_data has 4 argument. The 1st argument is this class object. The 2nd argument is utf-8 string which we get from the server. The 3rd argument is data type. ABNF.OPCODE_TEXT or ABNF.OPCODE_BINARY will be came. The 4th argument is continue flag. If 0, the data continue keep_running: bool This parameter is obsolete and ignored. get_mask_key: function A callable function to get new mask keys, see the WebSocket.set_mask_key's docstring for more information. cookie: str Cookie value. subprotocols: list List of available sub protocols. Default is None. socket: socket Pre-initialized stream socket. NFr)rErFrOrGrHrQrIrJrKrLrMr2rNr/ last_ping_tm last_pong_tm ping_thread stop_ping ping_intervalr ping_payloadrPprepared_socket has_errored)rrErFrGrHrIrJrKrLrMr2rNrOrPrQr<rrrrs0BzWebSocketApp.__init__)dataopcoder cCs$|jr|j||dkr tddS)a send message Parameters ---------- data: str Message to send. If you set opcode to OPCODE_TEXT, data must be utf-8 string or unicode. opcode: int Operation code of data. Default is OPCODE_TEXT. rzConnection is already closed.N)r/send"WebSocketConnectionClosedException)rr[r\rrrr]s zWebSocketApp.send)r cKs$d|_|jr |jjf|d|_dS)z- Close websocket connection. FN)r2r/r8)rkwargsrrrr8szWebSocketApp.closecCs<d|_|_t|_tj|jd|_d|j_|j dS)Nr)targetT) rSrT threadingEventrVThread _send_pingrUdaemonstart)rrrr_start_ping_threads   zWebSocketApp._start_ping_threadcCs<|jr|j|jr,|jr,|jdd|_|_dS)Nr)rVsetrUis_alivejoinrSrT)rrrr_stop_ping_threads   zWebSocketApp._stop_ping_threadc Cs|j|jrdSxt|j|js|jrt|_ytd|j|j Wqt k r}ztdj |dWdd}~XYqXqWdS)Nz Sending pingzFailed to send ping: {err})r) rVwaitrWr/rrSrdebugZpingrX Exceptionr!)rr&rrrrds  zWebSocketApp._send_pingrrRF)sockoptssloptrWrrXhttp_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authhttp_proxy_timeoutskip_utf8_validationhostoriginrAsuppress_origin proxy_typer'r cs dkr t |dk r$|dkr$td|dk r<|dkr 0zEnsure ping_interval >= 0z#Ensure ping_interval > ping_timeoutzsocket is already openedT) close_framer csNd_jrj|r*|nd\}}d_j||dS)a Tears down the connection. Parameters ---------- close_frame: ABNF frame If close_frame is set, the on_close handler is invoked with the statusCode and reason from the provided frame. FN)rlr2r/r8_get_close_args _callbackrJ)r|close_status_codeZ close_reason)rrrteardownms  z*WebSocketApp.run_forever..teardownF)rr cs|r jr jt j jdk dd _ jtyl jj j j j  j   j dt d jr   j jj Wn6tttttfk r}z||Wdd}~XYnXdS)NT)rprqZfire_cont_framerwZenable_multithread) rFrOrrrsrtrurvrPrxryrzr{r<zWebsocket connected)r/shutdownrrNrM settimeoutr connectrErFrOrPrYrr rWrgr~rGr;r^ConnectionRefusedErrorr% SystemExitro)rr&)checkrAhandleDisconnectrxrtrurrrsrvryr{r;rrwrprqrzrrsetSocks0       z)WebSocketApp.run_forever..setSock)r c s2js Syjd\}}Wn6ttfk rV}zrB|S|Wdd}~XYnX|tjkrj|S|tjkrj |j n|tj krt _ j|j n|tjkrjrj|j |j|jj|j |jnD|j }|tjkr s |d}j||jdj|dS)NTzutf-8)r2r/Zrecv_data_framer^r%rZ OPCODE_CLOSEZ OPCODE_PINGr~rKr[Z OPCODE_PONGrrTrLZ OPCODE_CONTrMrQr\Zfin OPCODE_TEXTdecoderH)Zop_codeframer&r[)custom_dispatcherrrrwrrrr;s4       z&WebSocketApp.run_forever..readcsZjrVtjjk}jjdk}jjjk}jrV|rV|sN|rVtddS)Nrzping/pong timed outT)rrrSrTZWebSocketTimeoutException)Zhas_timeout_expiredZ$has_pong_not_arrived_after_last_pingZhas_pong_arrived_too_late)rrrrsz'WebSocketApp.run_forever..check)r&rr csd_|s j|t|ttfr6rxtdj |drt dj t t dntdj |ddS)NTz{err} - reconnect)rzCCalling custom dispatcher reconnect [{frame_count} frames in stack])rz{err} - goodbye)rZrlr~rI isinstancer%rrr r!rnr"r#r$r'error)r&r)rrAr'rrrrrrsz2WebSocketApp.run_forever..handleDisconnectrhz>6 ! "  zWebSocketApp.run_forever)rrAis_sslr cCs0|rt|||S|pd}|r&t||St||S)N )r@r=r.)rrrArrrrrrs   zWebSocketApp.create_dispatcher)r|r cCsf|jr |sddgS|jrZt|jdkrZd|jd|jd}|jddd}||gSddgSdS)z _get_close_args extracts the close code and reason from the close body if it exists (RFC6455 says WebSocket Connection Close Code is optional) NrBrrzutf-8)rJr[r"r)rr|rreasonrrrr}s zWebSocketApp._get_close_args)rargsr c Gsd|r`y||f|WnHtk r^}z*tdj||d|jrN|||Wdd}~XYnXdS)Nz%error from callback {callback}: {err})rr)rorrr!rI)rrrr&rrrr~ szWebSocketApp._callback)NNNNNNNNTNNNNN)NNrNrRNNNNNFNNNFNN)NF)r(r)r*r+strr?dictrrr<rrrr,r]r8rgrlrdtupler-r.rrrr}r~rrrrr s62U  BN )r#r3sysrar tracebackr<typingrrrRrZ_abnfr_urlrZ_corerr _exceptions__all__rr,rrr.r=r@r rrrrs(