o iJ@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZddlT dgZdad d ZGd d d ZGd ddeZGdddeZGdddZGdddZdS)N)_logging)ABNF) parse_url) WebSocketgetdefaulttimeout)* WebSocketAppcCs|adSN) RECONNECT)reconnectIntervalr I/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/websocket/_app.py setReconnect&src@s(eZdZdZddZddZddZdS) DispatcherBasez DispatcherBase cCs||_||_dSr )app ping_timeout)selfrrr r r__init__/s zDispatcherBase.__init__cCst||dSr )timesleeprsecondscallbackr r rtimeout3s  zDispatcherBase.timeoutc Cslztd|ttft||ddWdSty5}ztd|fWYd}~dSd}~ww)Nz9reconnect() - retrying in %s seconds [%s frames in stack]T) reconnectingzUser exited %s)rinfoleninspectstackrrKeyboardInterrupt)rr reconnectorer r r reconnect7s zDispatcherBase.reconnectN)__name__ __module__ __qualname____doc__rrr#r r r rr+s  rc@seZdZdZddZdS) Dispatcherz Dispatcher cCs\|jjr,t}||jjjtj||j}|r|sdS|| |jjsdSdSr ) r keep_running selectorsDefaultSelectorregistersock EVENT_READselectrclose)rr- read_callbackcheck_callbackselrr r rreadDs zDispatcher.readN)r$r%r&r'r5r r r rr(@s r(c@s eZdZdZddZddZdS) SSLDispatcherz SSLDispatcher cCs4|jjr|}|r|sdS||jjsdSdSr )rr)r/)rr-r1r2r4r r rr5UszSSLDispatcher.readcCs^|jjj}|r |gSt}||tj||j}| t |dkr-|ddSdS)Nr) rr-pendingr*r+r,r.r/rr0r)rr-r3r4r r rr/]s    zSSLDispatcher.selectN)r$r%r&r'r5r/r r r rr6Qs r6c@s0eZdZdZddZddZddZdd Zd S) WrappedDispatcherz WrappedDispatcher cCs$||_||_||_|d|jdS)N)rr dispatchersignalabort)rrrr:r r rrpszWrappedDispatcher.__init__cCs,|j|||jo||j|dSdSr )r:r5rr)rr-r1r2r r rr5vszWrappedDispatcher.readcCs|j||dSr )r:rrr r rrzszWrappedDispatcher.timeoutcCs|||dSr )r)rrr!r r rr#}szWrappedDispatcher.reconnectN)r$r%r&r'rr5rr#r r r rr8ls  r8c@seZdZdZ        dddZejfddZdd Zd d Z d d Z ddZ         dddZ dddZ ddZddZdS)r z_ Higher level of APIs are provided. The interface is like JavaScript WebSocket object. NTcCs||_|dur |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)urlheadercookieon_open on_messageon_dataon_erroron_closeon_pingon_pongon_cont_messager) get_mask_keyr- last_ping_tm last_pong_tm ping_thread stop_ping ping_intervalr ping_payload subprotocolsprepared_socket has_errored)rr>r?rArBrDrErFrGrHr)rIr@rPrCsocketr r rrs0B zWebSocketApp.__init__cCs$|jr |j||dkrtddS)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)rdataopcoder r rrTs  zWebSocketApp.sendcKs,d|_|jr|jjdi|d|_dSdS)z- Close websocket connection. FNr )r)r-r0)rkwargsr r rr0s  zWebSocketApp.closecCs<d|_|_t|_tj|jd|_d|j_|j dS)Nr)targetT) rJrK threadingEventrMThread _send_pingrLdaemonstartrr r r_start_ping_threads  zWebSocketApp._start_ping_threadcCs<|jr|j|jr|jr|jdd|_|_dS)Nr)rMsetrLis_alivejoinrJrKr`r r r_stop_ping_threads   zWebSocketApp._stop_ping_threadc Cs|j|jr dS|j|jsH|jr?t|_ztd|j|j Wnt y>}z td|WYd}~nd}~ww|j|jrdSdS)Nz Sending pingzFailed to send ping: %s) rMwaitrNr-rrJrdebugpingrO Exception)rexr r rr] s  zWebSocketApp._send_pingrr=Fcs` durt |dur|dkrtd|dur|dkrtd|r*|r*||kr*tds.gs2ijr9td|_|_|_d_dfdd d fd d fd d fddd fdd t|t j ds rjrt dt t jsjS)a Run event loop for WebSocket framework. This loop is an infinite loop and is alive while websocket is available. Parameters ---------- sockopt: tuple Values for socket.setsockopt. sockopt must be tuple and each element is argument of sock.setsockopt. sslopt: dict Optional dict object for ssl socket option. ping_interval: int or float Automatically send "ping" command every specified period (in seconds). If set to 0, no ping is sent periodically. ping_timeout: int or float Timeout (in seconds) if the pong message is not received. ping_payload: str Payload message to send with each ping. http_proxy_host: str HTTP proxy host name. http_proxy_port: int or str HTTP proxy port. If not set, set to 80. http_no_proxy: list Whitelisted host names that don't use the proxy. http_proxy_timeout: int or float HTTP proxy timeout, default is 60 sec as per python-socks. http_proxy_auth: tuple HTTP proxy auth information. tuple of username and password. Default is None. skip_utf8_validation: bool skip utf8 validation. host: str update host header. origin: str update origin header. dispatcher: Dispatcher object customize reading data from socket. suppress_origin: bool suppress outputting origin header. proxy_type: str type of proxy from: http, socks4, socks4a, socks5, socks5h reconnect: int delay interval when reconnecting Returns ------- teardown: bool False if the `WebSocketApp` is closed or caught KeyboardInterrupt, True if any other exception was raised during a loop. NrzEnsure ping_timeout > 0zEnsure ping_interval >= 0z#Ensure ping_interval > ping_timeoutzsocket is already openedTcsNd_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)rfr)r-r0_get_close_args _callbackrE) close_frameclose_status_code close_reasonr`r rteardownis   z*WebSocketApp.run_forever..teardownFcs|r jr jt j jdu dd _ jtz7 jj j j j  j   j dt d jrF   j jj WdStttttfyt}z ||WYd}~dSd}~ww)NT)sockoptssloptfire_cont_frameskip_utf8_validationenable_multithread) r?r@http_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authhttp_proxy_timeoutrPhostoriginsuppress_origin proxy_typerSzWebsocket connected)r-shutdownrrIrH settimeoutrconnectr>r?r@rPrQrrrNrarmrAr5rUConnectionRefusedErrorr SystemExitrj)rr")checkr:handleDisconnectr|ryrzrwrxr{r}rr5rrurrrsr~r rsetSocks8     z)WebSocketApp.run_forever..setSockc s6jsSz jd\}}Wnttfy,}zr&|WYd}~S|d}~ww|tjkr6|S|tjkrEj |j dS|tj krYt _ j|j dS|tjkryjryj|j |j|jj|j |jdS|j }|tjkrs|d}j||jdj|dS)NTutf-8)r)r-recv_data_framerUr r OPCODE_CLOSE OPCODE_PINGrmrFrV OPCODE_PONGrrKrG OPCODE_CONTrHrCrWfin OPCODE_TEXTdecoderB)op_codeframer"rV)custom_dispatcherrrrurqr rr5sB       z&WebSocketApp.run_forever..readcsZjr+tjjk}jjdk}jjjk}jr+|r+|s'|r+tddS)Nrzping/pong timed outT)rrrJrKWebSocketTimeoutException)has_timeout_expired$has_pong_not_arrived_after_last_pinghas_pong_arrived_too_later`r rrsz'WebSocketApp.run_forever..checkcsd_|sj|t|ttfrr;td|r9t dt t  dSdStd|dS)NTz%s - reconnectz8Calling custom dispatcher reconnect [%s frames in stack]z %s - goodbye)rRrfrmrD isinstancer rrrrhrrrr#error)r"r)rr:r#rrrqr rrs z2WebSocketApp.run_forever..handleDisconnectrbz1Calling dispatcher reconnect [%s frames in stack]r )F)r WebSocketExceptionr-rNrrOr)boolcreate_dispatcherrr>rrhrrrr#rR)rrrrsrNrrOrwrxryrzr{rur|r}r:r~rr#r )rrr:rr|ryrzrwrxr{r}rr5r#rrrurrrsr~rqr run_forevers@>. !  zWebSocketApp.run_forevercCs0|rt|||S|p d}|rt||St||S)N )r8r6r()rrr:is_sslrr r rrs    zWebSocketApp.create_dispatchercCsb|jr|s ddgS|jr-t|jdkr-d|jd|jd}|jddd}||gSddgS)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) Nr9rrr)rErVrr)rrnroreasonr r rrls zWebSocketApp._get_close_argsc Gsx|r:z ||g|RWdSty9}z td|||jr.|||WYd}~dSWYd}~dSd}~wwdS)Nzerror from callback {}: {})rjrrformatrD)rrargsr"r r rrmszWebSocketApp._callback)NNNNNNNNTNNNNN)NNrNr=NNNNNFNNNFNN)NF)r$r%r&r'rrrrTr0rarfr]rrrlrmr r r rr s< \    V )rr*sysrZr tracebackr=r_abnfr_urlr_corerr _exceptions__all__r rrr(r6r8r r r r rs(