=\j8@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl TddlmZdgZGd d d ZGd d d ZGd ddeZdS)z N)ABNF) WebSocketgetdefaulttimeout)*)_logging WebSocketAppc@s.eZdZdZddZddZdS) Dispatcherz Dispatcher cCs||_||_dS)N)app ping_timeout)selfr r r :/tmp/pip-build-jxowagp5/websocket-client/websocket/_app.py__init__/s zDispatcher.__init__cCs^xW|jjrYtj|jjjfff|j\}}}|rO|sOP|qWdS)N)r keep_runningselectsockr )r r read_callbackcheck_callbackrwer r rread3s* zDispatcher.readN)__name__ __module__ __qualname____doc__rrr r r rr +s  r c@s:eZdZdZddZddZddZdS) SSLDispatcherz SSLDispatcher cCs||_||_dS)N)r r )r r r r r rrAs zSSLDispatcher.__init__cCs:x3|jjr5|j}|r+|s+P|qWdS)N)r rr)r rrrrr r rrEs   zSSLDispatcher.readcCsM|jjj}|jr"|gStj|fff|j\}}}|S)N)r rpendingrr )r rrrrr r rrMs  'zSSLDispatcher.selectN)rrrrrrrr r r rr=s   rc@seZdZdZddddddddddddddd ZejddZdd Zd d Z ddd dd ddddddddddddZ ddZ ddZ ddZ dS)rz_ Higher level of APIs are provided. The interface is like JavaScript WebSocket object. NTcCs||_|dk r|ng|_| |_||_||_||_||_||_||_||_ | |_ d|_ | |_ d|_ d|_d|_| |_dS)a4 WebSocketApp initialization Parameters ---------- url: websocket url. header: list or dict custom header for websocket handshake. on_open: callable object which is called at opening websocket. this function has one argument. The argument is this class object. on_message: callable 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 string which we get from the server. on_error: callable 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: callable object which is called when closed the connection. this function has one argument. The argument is this class object. on_cont_message: callback object which is called when receive continued frame data. 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: 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: this parameter is obsolete and ignored. get_mask_key: func a callable to produce new mask keys, see the WebSocket.set_mask_key's docstring for more information cookie: str cookie value. subprotocols: array of available sub protocols. default is None. NFr)urlheadercookieon_open on_messageon_dataon_erroron_closeon_pingon_pongon_cont_messager get_mask_keyr last_ping_tm last_pong_tm subprotocols)r rr r"r#r%r&r'r(r)rr*r!r-r$r r rr[s";               zWebSocketApp.__init__cCs5|j s%|jj||dkr1tddS)a send message Parameters ---------- data: Message to send. If you set opcode to OPCODE_TEXT, data must be utf-8 string or unicode. opcode: Operation code of data. default is OPCODE_TEXT. rzConnection is already closed.N)rsendZ"WebSocketConnectionClosedException)r dataopcoder r rr.s %zWebSocketApp.sendcKs/d|_|jr+|jj|d|_dS)z- Close websocket connection. FN)rrclose)r kwargsr r rr1s  zWebSocketApp.closecCsx~|j|stj|_|jry|jj|Wqtk r|}ztjdj|PWYdd}~XqXqWdS)Nz send_ping routine terminated: {}) waittimer+rZping Exceptionrwarningformat)r intervaleventpayloadexr r r _send_pings zWebSocketApp._send_pingrFcsdk rdkrdrB|rB|krBtd|sNg}|sZi}jrotddd_d_d_dfddyktjd|d |d jdk d | d |rdnd _jjt jj j dj dj d|d|d|d| djd| d| d|d| | syj} jj|rtjtjdjd||fd_jfdd}fdd}| jjj||Wnhtttfk r}z?jj|t|trYt|t SWYdd}~XnXdS)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, not send automatically. 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: http proxy host name. http_proxy_port: http proxy port. If not set, set to 80. http_no_proxy: host names, which doesn't use proxy. skip_utf8_validation: bool skip utf8 validation. host: str update host header. origin: str update origin header. dispatcher: customize reading data from socket. suppress_origin: bool suppress outputting origin header. Returns ------- teardown: bool False if caught KeyboardInterrupt, True if other exception was raised during a loop Nrz#Ensure ping_interval > ping_timeoutzsocket is already openedTcsr&jr&jjd_jrEjjj|rZ|jnd}jj |d_dS)z Tears down the connection. If close_frame is set, we will invoke the on_close handler with the statusCode and reason from there. FN) is_alivesetjoinrrr1_get_close_argsr/ _callbackr&)Z close_frameZ close_args)r9r threadr rteardowns     z*WebSocketApp.run_forever..teardownsockoptssloptZfire_cont_frameskip_utf8_validationZenable_multithreadFr r!http_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authr-hostoriginsuppress_origin proxy_typetargetargscs\jsSjjd\}}|tjkrA|S|tjkrijj|jn|tj krt j _ jj |jn|tj krjrjj|j|j|jjj|j|jn_|j}tjr)|tjkr)|jd}jj||jdjj|dS)NTzutf-8)rrZrecv_data_framerZ OPCODE_CLOSEZ OPCODE_PINGrBr'r/Z OPCODE_PONGr4r,r(Z OPCODE_CONTr)r$r0ZfinsixPY3 OPCODE_TEXTdecoder#)Zop_codeframer/)r rDr rr<s*    z&WebSocketApp.run_forever..readcsvrrtjjk}jjdk}jjk}jrr|rr|sf|rrtddS)Nrzping/pong timed outT)r4r+r,ZWebSocketTimeoutException)Zhas_timeout_expiredZ$has_pong_not_arrived_after_last_pingZhas_pong_arrived_too_late)r r r rcheckVs   z'WebSocketApp.run_forever..check)ZWebSocketExceptionrrr+r,rr*r) settimeoutrconnectrr r!r-create_dispatcherrBr" threadingEventThreadr<daemonstartrr5KeyboardInterrupt SystemExitr% isinstance)r rErFZ ping_intervalr Z ping_payloadrHrIrJrKrGrLrMZ dispatcherrNrOrrWrr )r9r r rDrCr run_forevers^4             zWebSocketApp.run_forevercCs5|p d}|jjr(t||St||S)N )rZis_sslrr )r r timeoutr r rrZks  zWebSocketApp.create_dispatchercCstjdkrA|j s:ttj|jjdkrpgSn/|j slttj|jjdkrpgS|rt|dkrdtj |ddtj |dd}|ddj d}||gSddgS) z _get_close_args extracts the code, reason from the close body if they exists, and if the self.on_close except three arguments rrNzutf-8)rfr) sys version_infor&leninspectZ getargspecrQZgetfullargspecrRbyte2intrU)r r/codereasonr r rrArs++4 zWebSocketApp._get_close_argscGs|ry|||Wnltk r}zLtjdj||tjrstj\}}}tj|WYdd}~XnXdS)Nzerror from callback {}: {}) r5rerrorr7ZisEnabledForDebugriexc_info tracebackprint_tb)r callbackrQr_tbr r rrBs zWebSocketApp._callback)rrrrrrrTr.r1r<rcrZrArBr r r rrVs,    H    )rrlrrir[r4rrrRZ_abnfrZ_corerr _exceptionsr=r__all__r robjectrr r r rs