iRddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZddl d gZdad ed dfd ZGd dZGddeZGddeZGddZGdd Zy)N)CallableAny)_logging)ABNF) parse_url) WebSocketgetdefaulttimeout)* WebSocketAppreconnectIntervalreturnc|ayN) RECONNECT)r s H/var/www/html/aiagents/fe/lib/python3.12/site-packages/websocket/_app.py setReconnectr)s!IcLeZdZdZdededdfdZdededdfd Z ded eddfd Z y) DispatcherBasez DispatcherBase app ping_timeoutrNc ||_||_yr)rr)selfrrs r__init__zDispatcherBase.__init__2s(rsecondscallbackc<tj||yr)timesleeprrrs rtimeoutzDispatcherBase.timeout6s 7 r reconnectorc 4 tjdj|tt j t j||dy#t$r,}tjdj||d}~wwxYw)NzQreconnect() - retrying in {seconds_count} seconds [{frame_count} frames in stack]) seconds_count frame_countT) reconnectingzUser exited {err}err) rinfoformatleninspectstackrr KeyboardInterrupt)rrr#es r reconnectzDispatcherBase.reconnect:s}  MMmtt%3w}}3GuI J JJw  T *   MM-444; <G sAA"" B+'BB) __name__ __module__ __qualname____doc__rfloatrintrr"r1rrrr.sU)C)u))sh48rrc<eZdZdZdejdededdfdZy) Dispatcherz Dispatcher sock read_callbackcheck_callbackrNctj}|j|jjjtj  |jj rC|j|j}|r|sn||jj rC|jy#|jwxYwr) selectorsDefaultSelectorregisterrr; EVENT_READ keep_runningselectrcloserr;r<r=selrs rreadzDispatcher.readIs'') TXX]]'')=)=> ((''JJt001(? (('' IIKCIIKs AB>>C)r2r3r4r5socketrrIr8rrr:r:Es.  x QY ^b rr:c\eZdZdZdejdededdfdZdejfd Z y) SSLDispatcherz SSLDispatcher r;r<r=rNc|jjj}tj}|j |tj  |jj r:|j||}|r|sn||jj r:|jy#|jwxYwr) rr;r?r@rArBrCrDrErFs rrIzSSLDispatcher.read[sxx}}!!'') T9//0 ((''KKc*(? (('' IIKCIIKs AB77C rGc|jjj}|jr|gS|j|j}t |dkDr|ddSy)Nr)rr;pendingrDrr,)rr;rGrHs rrDzSSLDispatcher.selectisVxx}}!! <<>7N JJt(( ) q6A:Q47N r) r2r3r4r5rJrrIr?r@rDr8rrrLrLWs@  x QY ^b y88rrLcxeZdZdZdededdfdZdejded eddfd Z d e d eddfd Z d e deddfdZ y)WrappedDispatcherz WrappedDispatcher r dispatcherrNcf||_||_||_|jd|jy)N)rrrRsignalabort)rrrrRs rrzWrappedDispatcher.__init__xs.($!Z--.rr;r<r=c|jj|||jxr|j|j|yyr)rRrIrr")rr;r<r=s rrIzWrappedDispatcher.read~s8 T=1 Mdll4+<+rc rd|_|jr$|jjdi|d|_yy)z- Close websocket connection. FNr8)rCr;rE)rkwargss rrEzWebSocketApp.closes4" 99 DIIOO %f %DI rcdx|_|_tj|_tj |j |_d|j_|jjy)Nr)targetT) rjrkrsEventrmThread _send_pingrldaemonstartrs r_start_ping_threadzWebSocketApp._start_ping_threadsW011D-"*$++4??C"&  rc|jr|jj|jr5|jjr|jj ddx|_|_y)Nr)rmsetrlis_alivejoinrjrkrs r_stop_ping_threadzWebSocketApp._stop_ping_thread sW >> NN      0 0 9 9 ;    ! !! $011D-rcv|jj|js|jdury|jj|js|jdur|jrTt j |_ tjd|jj|j|jj|js|jduryyyy#t$r/}tjdj|Yd}~kd}~wwxYw)NFTz Sending pingzFailed to send ping: {err}r() rmwaitrnrCr;rrjrdebugpingro Exceptionr+)rr0s rrzWebSocketApp._send_pings >>  t11 2d6G6G56P ..%%d&8&89d>O>OSW>Wyy$(IIK!ONN>2IINN4#4#45 ..%%d&8&89d>O>OSW>W9>W9 !ONN#?#F#F1#F#MNNOs:D D8 %D33D8sockoptssloptrnrrohttp_proxy_hosthttp_proxy_port http_no_proxyhttp_proxy_authhttp_proxy_timeoutskip_utf8_validationhostoriginrRsuppress_origin proxy_typer1c t||dkr td||dkr td|r|r||kr tdsgsijr td|_|_|_d_ddtffd dd td df    fd d tf fd d tffd ddtd td tffd tj|tjd smrkj r_tjdjt!t#j$j'j r_sj,S#t(tf$r6}tj*dj|Yd}~Ud}~wwxYw#swwxYw)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 openedT close_framectj5jr dddyd_dddjd_jrjj j |r|nd\}}d_jj||y#1swYxYw)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. NTF) rurrrrCr;rE_get_close_args _callbackr`)rclose_status_code close_reasonrs rteardownz*WebSocketApp.run_forever..teardownss,, .)) . .*.& .  " " $ %D yy !.2.B.B* /6 + |DI NN4==*;\ J . .sB.B..B7r'rc|r&jrjjtjjdud_jj t  jjjjj  j  jtjdjrj!j#j$j'jj y#t(t*t,t.t0f$r}||Yd}~yd}~wwxYw)NT)rrfire_cont_framerenable_multithread) r\rerrrrrrfrrrrrJzWebsocket connected)r;shutdownr rdrc settimeoutr connectr[r\rerfrprr*rnrrr]rIrzConnectionRefusedErrorr/ SystemExitr)r'r0checkrRhandleDisconnectrrrrrrrrrIrrrrrs rsetSockz)WebSocketApp.run_forever..setSocks2  ""$!!!76 $ 4 4D @%9#' )DI II !2!4 5 2 !!HHT[[$3$3=$3HZ!%!2!2fo)$2F2F"H 34%%++-t||, e<68NPacmoxy 2 L11 2s:CE E:' E55E:cjsS jjd\}}|t j k(r|S|t jk(r'jj|jy|t jk(r@tj_ jj|jy|t jk(rzj rnjj"|j|j$|j&jj |j|j&y|j}|t j(k(rs|j+d}jj"||j$djj,|y#ttf$r}r |cYd}~S|d}~wwxYw)NTutf-8)rCr;recv_data_framerzr/r OPCODE_CLOSE OPCODE_PINGrrarv OPCODE_PONGrrkrb OPCODE_CONTrcrgrwfin OPCODE_TEXTdecoder^) op_codeframer0rvcustom_dispatcherrrrrs rrIz&WebSocketApp.run_forever..reads$$z! !%!:!:4!@$+++&D,,,t||UZZ8 D,,,$(IIK!t||UZZ8D,,,1E1Et||UZZ$||UYY8t33$zz5996 zzd...7K;;w/Dt||T5<<Ft5378IJ $+A..G  s#G H/ H8H>HHc8jrtjjz jkD}jjz dk}jjz jkD}jr|r|s|r t dy)Nrzping/pong timed outT)rrrjrkWebSocketTimeoutException)has_timeout_expired$has_pong_not_arrived_after_last_pinghas_pong_arrived_too_laters rrz'WebSocketApp.run_forever..checks!!&*iikD4E4E&EHYHY&Y#7;7H7H4K\K\7\_`7`4,0,=,=@Q@Q,QTXTeTe,e)%%+=AZ34IJJrr0cd_j|sjj|t |t t frr{tjdj|rStjdjttjjyytjdj|y)NTz{err} - reconnectr(zCCalling custom dispatcher reconnect [{frame_count} frames in stack]r&z{err} - goodbye)rqrrr_ isinstancer/rrr*r+rr,r-r.r1error)r0r'rrRr1rrrs rrz2WebSocketApp.run_forever..handleDisconnects#D   " " $t}}a0!/<=  188Q8?@$NN#h#o#o|AHANANAP}Q#o$RS((G<%077A7>? rrz? ?  $):$%@A A Mm|.K$%JK KGF 99$%?@ @*((  K$ K: 2$ 24 2 2 2@ d  B t     $  (!,++L*iPTPXPXFYZ[F\]   I$''NN#a#h#huxzAzGzGzIvJ#h$KL((G<''% "9-  MM;BBqBI J JJ % %s+,A5F77G<,G72G?7G<<G?? H is_sslc`|r t|||S|xsd}|r t||St||S)N )rQrLr:)rrrRrr"s rrzWebSocketApp.create_dispatchers; $T<D D$"  w/ /$((rrc|jr|sddgS|jr\t|jdk\rDd|jdz|jdz}|jddjd}||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) NrTrrr)r`rvr,r)rrrreasons rrzWebSocketApp._get_close_args s}}K$<    K$4$4 5 : #k&6&6q&9 9Krs   /&   "C"D" .$N:++*_+_+r