o a:@sdZ ddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl Tddl m Z dgZGd d d ZGd d d ZGd ddZdS)z N)ABNF) WebSocketgetdefaulttimeout)*)_logging WebSocketAppc@s eZdZdZddZddZdS) Dispatcherz Dispatcher cC||_||_dSNapp ping_timeoutselfr rr0/usr/lib/python3/dist-packages/websocket/_app.py__init__) zDispatcher.__init__cCs\|jjr,t}||jjjtj||j}|r|sdS|| |jjsdSdSr ) r keep_running selectorsDefaultSelectorregistersock EVENT_READselectrclose)rr read_callbackcheck_callbackselrrrrread-s zDispatcher.readN)__name__ __module__ __qualname____doc__rr!rrrrr %s r c@s(eZdZdZddZddZddZdS) SSLDispatcherz SSLDispatcher cCr r r rrrrr>rzSSLDispatcher.__init__cCs4|jjr|}|r|sdS||jjsdSdSr )r rr)rrrrr rrrr!BszSSLDispatcher.readcCs^|jjj}|r |gSt}||tj||j}| t |dkr-|ddSdS)Nr) r rZpendingrrrrrrrlen)rrrr rrrrJs    zSSLDispatcher.selectN)r"r#r$r%rr!rrrrrr&:s  r&c@sxeZdZdZ       dddZejfddZdd Zd d Z       dddZ ddZ ddZ ddZ dS)rz_ Higher level of APIs are provided. The interface is like JavaScript WebSocket object. NTcCsv||_|dur |ng|_| |_||_||_||_||_||_||_||_ | |_ 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. NFr)urlheadercookieon_open on_messageon_dataon_erroron_closeon_pingon_pongon_cont_messager get_mask_keyr last_ping_tm last_pong_tm subprotocols)rr(r)r+r,r.r/r0r1r2rr3r*r6r-rrrr^s"? 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)rsendZ"WebSocketConnectionClosedException)rdataopcoderrrr7s  zWebSocketApp.sendcKs,d|_|jr|jjdi|d|_dSdS)z- Close websocket connection. FNr)rrr)rkwargsrrrrs  zWebSocketApp.closec Cst||s8t|_|jr1z|j|Wnty0}ztd|WYd}~dSd}~ww||rdSdS)Nz send_ping routine terminated: {}) waittimer4rZping ExceptionrZwarningformat)rintervaleventZpayloadexrrr _send_pings  zWebSocketApp._send_pingrFcsdur dkr td|dur|dkrtdr$|r$|kr$td|s(g}|s,i}jr3tddd_d_d_dfdd zktj||jdu| dd _jt jj j j j |||| j| | ||d | s{} j|rttjj||fd d_fd d}fdd}| jj||WdStttfy}zj|t|trʂt|t WYd}~Sd}~ww)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. 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. Returns ------- teardown: bool False if caught KeyboardInterrupt, True if other exception was raised during a loop NrzEnsure ping_timeout > 0zEnsure ping_interval >= 0z#Ensure ping_interval > ping_timeoutzsocket is already openedTcsbrrd_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) Zis_alivesetjoinrrr_get_close_args _callbackr/) close_frameclose_status_codeZ close_reason)r@rthreadrrteardowns   z*WebSocketApp.run_forever..teardown)sockoptssloptZfire_cont_frameskip_utf8_validationZenable_multithread) r)r*http_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authr6hostoriginsuppress_origin proxy_type)targetargscsjsSjd\}}|tjkr|S|tjkr&j|jdS|tj kr:t _ j |jdS|tj krZjrZj|j|j|jj|j|jdS|j}|tjkrg|d}j||jdj|dS)NTutf-8)rrZrecv_data_framerZ OPCODE_CLOSEZ OPCODE_PINGrGr0r8Z OPCODE_PONGr<r5r1Z OPCODE_CONTr2r-r9Zfin OPCODE_TEXTdecoder,)Zop_codeframer8)rrKrrr!Ks4        z&WebSocketApp.run_forever..readcsTr(tjk}jjdk}jjk}jr(|r(|s$|r(tddS)Nrzping/pong timed outT)r<r4r5ZWebSocketTimeoutException)Zhas_timeout_expiredZ$has_pong_not_arrived_after_last_pingZhas_pong_arrived_too_late)rrrrcheckesz'WebSocketApp.run_forever..checkr )ZWebSocketExceptionrrr4r5rr3r2Z settimeoutrZconnectr(r)r*r6create_dispatcherrGr+ threadingZEventZThreadrBZdaemonstartr!r=KeyboardInterrupt SystemExitr. isinstance)rrLrMZ ping_intervalrZ ping_payloadrOrPrQrRrNrSrT dispatcherrUrVr!r]er)r@rrrKrJr run_foreversl4      zWebSocketApp.run_forevercCs&|pd}|jrt||St||S)N )rZis_sslr&r )rrZtimeoutrrrr^zs   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) NrrrY)r/r8r'r[)rrHrIreasonrrrrFs zWebSocketApp._get_close_argsc Gsx|r:z ||g|RWdSty9}z td|||jr.|||WYd}~dSWYd}~dSd}~wwdS)Nzerror from callback {}: {})r=rerrorr>r.)rcallbackrXrerrrrGszWebSocketApp._callback) NNNNNNNNTNNNN)NNrNrCNNNNFNNNFN)r"r#r$r%rrrZr7rrBrfr^rFrGrrrrrYs4 R   ' )r%rsysr_r< tracebackZ_abnfrZ_corerr _exceptionsrCr__all__r r&rrrrrs