DQcj8dZ ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl TddlmZdgZGd d ZGd d ZGd deZdS)z N)ABNF) WebSocketgetdefaulttimeout)*)_logging WebSocketAppceZdZdZdZdZdS) Dispatcherz Dispatcher c"||_||_dSNapp ping_timeoutselfrrs .lib/python3.11/site-packages/websocket/_app.py__init__zDispatcher.__init__.(c|jjrVtj|jjjfdd|j\}}}|r |sdS||jjTdSdSN)r keep_runningselectsockr)rr read_callbackcheck_callbackrwes rreadzDispatcher.read2sh# mX]'*BD4EGGGAq! $}E N    h#     rN)__name__ __module__ __qualname____doc__rr"rrrr r *s<)))rr c$eZdZdZdZdZdZdS) SSLDispatcherz SSLDispatcher c"||_||_dSr rrs rrzSSLDispatcher.__init__?rrc|jjr:|}|r |sdS||jj8dSdSr )rrr)rrrrrs rr"zSSLDispatcher.readCshh#  A $}E N    h#     rc|jjj}|r|gStj|fdd|j\}}}|Sr)rrpendingrr)rrrr r!s rrzSSLDispatcher.selectKsJx}! <<>> 7N-"b$2CDD1arN)r#r$r%r&rr"rrrrr(r(;sK)))rr(cpeZdZdZ ddZejfdZdZdZ dd Z d Z d Z d Z dS)r z_ Higher level of APIs are provided. The interface is like JavaScript WebSocket object. NTc||_||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)rr/r0r2r3r5r6r7r8r9rr:r1r=r4s rrzWebSocketApp.__init__Ysv &:ff   $       .!( (rcp|jr|j||dkrtddS)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)rsend"WebSocketConnectionClosedException)rdataopcodes rr?zWebSocketApp.sendsIy 1DINN488A= 14/11 1 1 1rc Xd|_|jr|jjdi|d|_dSdS)z- Close websocket connection. FNr)rrclose)rkwargss rrDzWebSocketApp.closesE" 9  DIO % %f % % %DIII  rcP||stj|_|jrZ |jn?#t $r2}t jd|Yd}~dSd}~wwxYw||dSdS)Nz send_ping routine terminated: {}) waittimer;rping Exceptionrwarningformat)rintervaleventexs r _send_pingzWebSocketApp._send_pings**X&&  $ D y INN$$$$ $%G%N%Nr%R%RSSSEEEEE **X&&     sA B 'BB rFcBdkrdr|r|krtd|sg}|si}jrtddd_d_d_d fd  t j||jdu| |rdnd_jtj j j j ||||j| | | | | s} j|rYt%jt%jj|f dfd }fd }| jj||dS#t2t4t6f$r[}j|t;|t6rt;|t4 cYd}~Sd}~wwxYw)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. 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 openedTcBr<r(d_jrj|r|jnd}jj g|Rd_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_alivesetjoinrrrD_get_close_argsrA _callbackr6) close_frame close_argsrNrthreads rteardownz*WebSocketApp.run_forever..teardown s &//++   %D y " !!!--$/9   T;;J DN4= 6: 6 6 6 6DIIIrF)sockoptssloptfire_cont_frameskip_utf8_validationenable_multithread) r0r1http_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authr=hostoriginsuppress_origin proxy_type)targetargscNjs Sjd\}}|tjkr |S|tjkr"j|jn(|tj kr9tj _ j |jn|tj krZjrSj|j|j|jj|j|jnu|j}t$jr%|tjkr|d}j||jdj|dS)NTutf-8)rrrecv_data_framer OPCODE_CLOSE OPCODE_PINGrWr7rA OPCODE_PONGrHr<r8 OPCODE_CONTr9r4rBfinsixPY3 OPCODE_TEXTdecoder3)op_codeframerArr[s rr"z&WebSocketApp.run_forever..read7s(&#8::%!%!:!:4!@!@d//:#8E??* 00:NN4<<<<< 00 :(, D%NN4<<<<< 00 :T5I :NN4<#(<<<<NN4#7#(:uy::::!:Dw47d.>#>4#{{733NN4<u|TJJJNN4?D999trcratjjz k}jjz dk}jjz k}jr|r|s|rtddS)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..checkQs O*.)++8I*IL*X';?;LtO`;`cd;d8040ADDU0UXd0d-)O 3O!EOIbO88MNNNtrr )WebSocketExceptionrrr;r<rr:r9 settimeoutrconnectr/r0r1r=create_dispatcherrWr2 threadingEventThreadrP setDaemonstartr"rJKeyboardInterrupt SystemExitr5 isinstance)rr\r] ping_intervalrrarbrcrdr_rerf dispatcherrgrhr"r~r!rNr[rZs` ` @@@r run_foreverzWebSocketApp.run_foreversb  (9 L  LM Lm|.K L$%JKK K G F 9 A$%?@@ @         $G 8!!76 $ 4D @%9+8#C44e EEEDI I !2!4!4 5 5 5 I  T[ / /} /d>O&/%  ' ' ' B!33LAA NN4< ( ( ( !))")?-1GIII  &&&       4       OODIND% 8 8 8 8 8,j9 8 8 8 NN4=! , , ,!Z((  HJJJ!!%6777 7 7 7 7 7 7  8s 'EF,,HAHHHc||pd}|jrt||St||S)N )ris_sslr(r )rrtimeouts rrzWebSocketApp.create_dispatcherfsB$" 9     0 w// /$(((rctjdkr9|jr/tt j|jjdkrgSn8|jr/tt j|jjdkrgS|rpt|dkr]dtj |ddztj |ddz}|dd 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 )rrrrNrl) sys version_infor6leninspect getargspecrjgetfullargspecrsbyte2intrv)rrAcodereasons rrVzWebSocketApp._get_close_argsms  f $ = C(:4=(I(I(N$O$OST$T   = C(>t}(M(M(R$S$SWX$X   "CIIN "d1Q3i0003<QqS 3J3JJD!""X__W--F&> !d|rc*|r ||g|RdS#t$rw}tjd||tjr1t j\}}}tj|Yd}~dSYd}~dSd}~wwxYwdS)Nzerror from callback {}: {}) rJrerrorrLisEnabledForDebugrexc_info tracebackprint_tb)rcallbackrjr!_tbs rrWzWebSocketApp._callbacks  + +%%%%%%% + + +;BB8QOOPPP-//+"|~~HAq"&r*********++++++ +  + +s  BA&B  B) NNNNNNNNTNNNN)NNrNNNNNFNNNFN)r#r$r%r&rrrur?rDrPrrrVrWrrrr r Ts$(9=6:!%>B" L)L)L)L)\!% 01111"0426:>8<).7;6: Y8Y8Y8Y8v)))( + + + + +r)r&rrrrrHrrs_abnfr_corerr _exceptionsr__all__r r(objectr rrrrs;(   ////////  "2v+v+v+v+v+6v+v+v+v+v+r