5dZ ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl ddlmZdgZGd d ZGd d ZGd deZy)aL websocket - WebSocket client library for Python Copyright (C) 2010 Hiroki Ohtani(liris) This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1335 USA N)ABNF) WebSocketgetdefaulttimeout)*)_logging WebSocketAppceZdZdZdZy) Dispatcherc ||_||_yNapp ping_timeoutselfrrs !lib/third_party/websocket/_app.py__init__zDispatcher.__init__,(c4|jjjrxtj|jjjfdd|j\}}}|r|sy||jjjrwyyN)rsock connectedselectr)rr read_callbackcheck_callbackrwes rreadzDispatcher.read0slhhmm%%mmXX]]''*BD4E4EGGAq!$   hhmm%%rN)__name__ __module__ __qualname__rr"rrrr r +s )rr ceZdZdZdZdZy) SSLDispactherc ||_||_yr rrs rrzSSLDispacther.__init__:rrc|jjjrC|j}|r|sy||jjjrByyr )rrrr)rrrrrs rr"zSSLDispacther.read>sBhhmm%% A$   hhmm%%rc|jjj}|jr|gStj|fdd|j\}}}|Sr)rrpendingrr)rrrr r!s rrzSSLDispacther.selectFsLxx}}!! <<>7N--"b$2C2CD1arN)r#r$r%rr"rrrrr'r'9s)rr'cxeZdZdZ d dZej fdZdZdZ d dZ dZ d Z d Z y) r zc Higher level of APIs are provided. The interface is like JavaScript WebSocket object. Nc||_||ng|_| |_||_||_||_||_||_||_||_ | |_ d|_ | |_ d|_ d|_d|_| |_y)a url: websocket url. header: 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: a callable to produce new mask keys, see the WebSocket.set_mask_key's docstring for more information subprotocols: array of available sub protocols. default is None. NFr)urlheadercookieon_open on_messageon_dataon_erroron_closeon_pingon_pongon_cont_message keep_running get_mask_keyr last_ping_tm last_pong_tm subprotocols)rr.r/r1r2r4r5r6r7r8r9r:r0r=r3s rrzWebSocketApp.__init__TsT & 2f   $       .!( (rcp|jr|jj||dk(r tdy)z send message. 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)rsend"WebSocketConnectionClosedException)rdataopcodes rr?zWebSocketApp.sends6yyDIINN48A=4/1 1>rc rd|_|jr$|jjdi|d|_yy)z- close websocket connection. FNr)r9rclose)rkwargss rrDzWebSocketApp.closes4" 99 DIIOO %f %DI rc@|j|sStj|_|jr |jj |j|sRyy#t $r.}t jdj|Yd}~yd}~wwxYw)Nz send_ping routine terminated: {}) waittimer;rping Exceptionrwarningformat)rintervaleventexs r _send_pingzWebSocketApp._send_pingss**X& $ D yyIINN$ **X& !$$%G%N%Nr%RSsA&& B/$BBcLdkrdr|r|kr td|sg}|si}jr tddd_d_d_d fd  t j ||jdu| |rdnd_jjtjjjjj||||j| | | | | sj} j!j"|rXt%j&t%j(j*|f j-dj/fd }fd }| j1jj||y#t2t4t6f$rO}j!j8|t;|t6rt;|t4 cYd}~Sd}~wwxYw)a@ run event loop for WebSocket framework. This loop is infinite loop and is alive during websocket is available. sockopt: values for socket.setsockopt. sockopt must be tuple and each element is argument of sock.setsockopt. sslopt: ssl socket optional dict. ping_interval: automatically send "ping" command every specified period(second) if set to 0, not send automatically. ping_timeout: timeout(second) if the pong message is not received. 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: skip utf8 validation. host: update host header. origin: update origin header. dispatcher: customize reading data from socket. suppress_origin: suppress outputting origin header. Returns ------- False if caught KeyboardInterrupt True if other exception was raised during a loop Nrz#Ensure ping_interval > ping_timeoutzsocket is already openedTcJr0jr jjd_jrjj j |r |jnd}jjg|d_y)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) isAlivesetjoinr9rrD_get_close_argsrA _callbackr5) close_frame close_argsrNrthreads rteardownz*WebSocketApp.run_forever..teardowns{ &..*  %D yy !--$/   T;J DNN4== 6: 6DIrF)sockoptssloptfire_cont_frameskip_utf8_validationenable_multithread) r/r0http_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authr=hostoriginsuppress_origin proxy_type)targetargscjsSjjd\}}|tjk(r|S|tj k(r'j j|jy|tjk(r@tj_ j j|jy|tjk(rzjrnj j|j|j |j"j j|j|j"y|j}t$j&r$|tj(k(r|j+d}j j||j dj j,|y)NTutf-8)r9rrecv_data_framer OPCODE_CLOSE OPCODE_PINGrWr6rA OPCODE_PONGrHr<r7 OPCODE_CONTr8r3rBfinsixPY3 OPCODE_TEXTdecoder2)op_codeframerArr[s rr"z&WebSocketApp.run_forever..read se((#:%!%!:!:4!@d///#E?* 0 00NN4<<<  0 00(, D%NN4<<< 0 00T5I5INN4<<#(<<<NN4#7#7#(::uyy: !::Dww7d.>.>#>#{{73NN4<<u||TJNN4??D9rcrytjjz kD}jjz dk}jjz kD}jr|r|s|r tdy)Nrzping/pong timed outT)rHr;r<WebSocketTimeoutException)has_timeout_expired$has_pong_not_arrived_after_last_pinghas_pong_arrived_too_laterrs rcheckz'WebSocketApp.run_forever..check%s} *.))+8I8I*IL*X';?;L;LtO`O`;`cd;d8040A0ADDUDU0UXd0d-)) 3!EIb78MNNrr )WebSocketExceptionrr9r;r<rr:r8 settimeoutrconnectr.r/r0r=create_dispatcherrWr1 threadingEventThreadrP setDaemonstartr"rJKeyboardInterrupt SystemExitr4 isinstance)rr\r] ping_intervalrrarbrcrdr_rerf dispatcherrgrhr"r~r!rNr[rZs` ` @@@r run_foreverzWebSocketApp.run_foreversB  # (9L Mm|.K$%JK KGF 99$%?@ @  "G 8!!!76 $ 4 4D @%9+84e EDI II !2!4 5 II  T[[ / /} /d>O>O&/%  '!33LA NN4<< (!)"))??-1GI  &  4  OODIINND% 8,j9 8 NN4==! ,!Z( J!!%677 7  8s "EGH#AHH#H#cr|xsd}|jjr t||St||S)N )ris_sslr'r )rrtimeouts rrzWebSocketApp.create_dispatcher:s6$" 99    w/ /$((rctjdkrC|jr5tt j |jj dk7rEgS|jr5tt j|jj dk7rgS|rZt|dk\rLdtj|ddztj|ddz}|ddjd}||gSddgS) z this functions extracts the code, reason from the close body if they exists, and if the self.on_close except three arguments )rrrrNrl) sys version_infor5leninspect getargspecrjgetfullargspecrsbyte2intrv)rrAcodereasons rrVzWebSocketApp._get_close_argsAs   f $==C(:(:4==(I(N(N$OST$T ==C(>(>t}}(M(R(R$SWX$X CINd1Qi003<<Qq 3JJD!"X__W-F&> !d|rcV|r& tj|r||y||g|yy#t$ru}tjdj ||tj r2tj\}}}tj|Yd}~yYd}~yd}~wwxYw)Nzerror from callback {}: {}) rismethodrJrerrorrLisEnabledForDebugrexc_info tracebackprint_tb)rcallbackrjr!_tbs rrWzWebSocketApp._callbackSs  +##H-dOT)D)  +;BB8QOP--/"||~HAq"&&r**0 +s* * B(A&B##B() NNNNNNNNTNNNN)NNrNNNNNFNNNFN)r#r$r%__doc__rrrur?rDrPrrrVrWrrrr r Nsk $(9=6:!%>B" ;)z!% 0 0 10426:>8<).7;8< H8T)$ +r)rrrrrrHrrs_abnfr_corerr _exceptionsr__all__r r'objectr rrrrs^,   /    *Q+6Q+r