!nilddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z  ddl Z ddlZddlmZddlmZddlmZdd lmZej(d ZGd d ej,Zy#e $rdZ YNwxYw#e $rdZYVwxYw) ) b64encode) SimpleCookieN)JSONDecodeError) base_client) exceptions)packet)payloadzengineio.clientceZdZdZ ddZdZdZddZdZddZ dZ d Z d Z fd Z d Zd ZdZdZ ddZdZdZdZdZxZS)ClientaAn Engine.IO client. This class implements a fully compliant Engine.IO web client with support for websocket and long-polling transports. :param logger: To enable logging set to ``True`` or pass a logger object to use. To disable logging set to ``False``. The default is ``False``. Note that fatal errors are logged even when ``logger`` is ``False``. :param json: An alternative json module to use for encoding and decoding packets. Custom json modules must have ``dumps`` and ``loads`` functions that are compatible with the standard library versions. :param request_timeout: A timeout in seconds for requests. The default is 5 seconds. :param http_session: an initialized ``requests.Session`` object to be used when sending requests to the server. Use it if you need to add special client options such as proxy servers, SSL certificates, custom CA bundle, etc. :param ssl_verify: ``True`` to verify SSL certificates, or ``False`` to skip SSL certificate verification, allowing connections to servers with self signed certificates. The default is ``True``. :param handle_sigint: Set to ``True`` to automatically handle disconnection when the process is interrupted, or to ``False`` to leave interrupt handling to the calling application. Interrupt handling can only be enabled when the client instance is created in the main thread. :param websocket_extra_options: Dictionary containing additional keyword arguments passed to ``websocket.create_connection()``. :param timestamp_requests: If ``True`` a timestamp is added to the query string of Socket.IO requests as a cache-busting measure. Set to ``False`` to disable. c|jdk7r tdddg}|3t|tr|g}|Dcgc]}||vr| }}|s td|xs||_t |d|jdz||xsi|Scc}w)aConnect to an Engine.IO server. :param url: The URL of the Engine.IO server. It can include custom query string parameters if required by the server. :param headers: A dictionary with custom headers to send with the connection request. :param transports: The list of allowed transports. Valid transports are ``'polling'`` and ``'websocket'``. If not given, the polling transport is connected first, then an upgrade to websocket is attempted. :param engineio_path: The endpoint where the Engine.IO server is installed. The default value is appropriate for most cases. Example usage:: eio = engineio.Client() eio.connect('http://localhost:5000') disconnectedz%Client is not in a disconnected statepolling websocketzNo valid transports provided _connect_r)state ValueError isinstancestr transportsgetattr)selfurlheadersr engineio_pathvalid_transports transports O/home/homepc/tiktok-worker/venv/lib/python3.12/site-packages/engineio/client.pyconnectzClient.connect?s* :: 'DE E%{3  !*c*(\ 5?< &*::$wt[4??1+==> B / / r4 get_nowait task_done queue_empty)r __class__s rr>z Client._resetsT   %%' $$&##  s4AAAct|jjdy|j||d|_|jj d|jz|j d|j|jz||j}|t|tr)|jtj|xsd|jdks|jd k\rP|j |j}tjd j#|j| t%j&|j(j+d  }|j.d}|j0t2j4k7rtjd|jj dt|j6z|j6d|_|j6d|_t=|j6ddz |_t=|j6ddz |_ d|_!|xjd|j8zz c_d|_"tFjHjK||jMdd|j.ddD]}|jO|d|j:vr"d|jPvr|jS|||ry|jU|jV|_,|jU|jZ|_.y#t $rd}Y?wxYw#t,$rtjd dwxYw)zrConnectionError status_codejsonrformatr Payloadcontentdecoderpackets packet_typer OPENr(rfrgint ping_interval ping_timeoutr9rrr<appendr6_receive_packetr_connect_websocketrI _write_loopwrite_loop_task_read_loop_pollingr!) rrrrrargp open_packetpkts r_connect_pollingzClient._connect_pollings   KK  4 5 ..sM9M  EEMM#$') ) = 0@0@0IJAiil  " "fkk 1,,46 6  /#k6F6F2G G I##E*#((4  !1!1.!ABVK 0 0 ?@6I!* 488++   %%,,T2 I799QR= &C   % & $-- 'K4??,J&&sG]C $99$:J:JK"88  # #%M#   =,,138< = =sM/M MM M:c>t|jjdy|j||d}|jr3|jj d|zd}|d|jzz }n'd}||_|jj d|zd}i}|jrt}|jjD]} | j|| j<|jd d j}|jD])\} } | jd k(s|r|d z }|| z }|| =nd|vr|jj t#|jj t$s t'ddj)|jj d|jj dj+d} t-| j/d} d| z|d<t#|jj0t$r7|jj0d|jj0dd|d<n1|jj0rd|jj0i|d<|jj2r#d} |j5drJ|jj2j7d|jj2j7d} nI|jj2j7d|jj2j7d} | r{t8j:j=d| vr| nd| z}|j>|d<|j@|d <|jBs |jDr|jB|jDfnd|d!<t#|jjFtHr=d|vr|jjF|dd"<n9d"|jjFi|d<n|jjFsd|_%|jJs>d|vr%|djMd#tNjPind#tNjPi|d<|jM|jRjUd$i||d$<||d <d|d%<|jV|d&<|jM|jR tjX||j[zfi|}|rtgjhtfjjd)*j+} |jm| |jq}tgjh|-}|jrtfjtk7s|jvd)k7r|jjcd.ytgjhtfjxj+} |jm|d|_=|jj d/n2 |jq}tgjh|-}|jrtfj|k7rtej\d1|jj d2tI|jvz|jvd3|_|jvd4|_?t|jvd5d6z |_At|jvd7d6z |_Bd|_=d8|_Ctjj||jd9d:||_H|jj|j|jz|j|j|_L|j|j|_Ny#t\t^tj`f$r5|r|jjcd'Yytej\d(wxYw#tn$r/}|jjcd+tI|Yd}~yd}~wwxYw#tn$r/}|jjcd,tI|Yd}~yd}~wwxYw#tn$r/}|jjcd+tI|Yd}~yd}~wwxYw#tn$r&}tej\d0tI|zd}~wwxYw);z?Establish or upgrade to a WebSocket connection with the server.NzKwebsocket-client package not installed, only polling transport is availableFrz Attempting WebSocket upgrade to Trkz#Attempting WebSocket connection to ;)headersepcookiez; Authorizationz&Only basic authentication is supportedz{}:{}rrrczBasic )certfilekeyfilessloptrzws://r:httpwsshttpsz://z scheme://http_proxy_hosthttp_proxy_porthttp_proxy_authca_certs cert_reqsrenable_multithreadr`z*WebSocket upgrade failed: connection errorzConnection errorprober'z7WebSocket upgrade failed: unexpected send exception: %sz7WebSocket upgrade failed: unexpected recv exception: %sencoded_packetz(WebSocket upgrade failed: no PONG packetz WebSocket upgrade was successfulzUnexpected recv exception: zno OPEN packetz#WebSocket connection accepted with rfrgrhrirjr.rr1)Orrmrnrorfrqrprrcookiesvaluenameoutputstripitemslowerauthrtuplerrxencoderr{certproxies startswithgeturllibparseurlparsehostnameportusernamepasswordverifyr ssl_verifyupdatessl CERT_NONEwebsocket_extra_optionspoprtcreate_connectionrsruOSErrorWebSocketExceptionwarningrr r*PINGr, Exceptionrecvr}PONGr(UPGRADEr9r~rgrrrrrr<rr6r: settimeoutrIrr_read_loop_websocketr!)rrrr websocket_urlupgrader extra_optionsckrrr basic_auth proxy_url parsed_urlr:rerrs rrzClient._connect_websockets   KK  ? @..sM;O 88 KK  2]B DG Wtxx/ /MG)DM KK  5 E G  99B))++ /"(,,6;; /iirsi399;G!(  <<>X-4u$G  g-$))..2L!$))..%8$%MNN$^^IINN1%tyy~~a'8::@&/&z299'B +3j+@($))..%0 $ q 1#yy~~a0+2 h'+5tyy~~*F h'yy    ++G4 $ 1 1 5 5dii//33F;!=I!% 1 1 5 5tyy0044W=!?I!'!6!6%*i%7 (94"6J8B7J7JM"347AM"34&..*2E2E$,,j.A.AB!""34 $))**C0},:>)):J:JM(+J7/9499;K;K.LM(+YY%%"'=(h'.. S]]/KL+6 *F h' t3377"EF") h") h.2 *+#'#7#7 i T99: E,, 7 7 99L=JLB  fkk8??AA   GGI --q1C&++-W1D ##>@ fnn-446A  &1D " KK  ? @ <GGI!--q9K&&&++5 001ABB KK  5K|jtjtj|jy |jtjk(r(|j!d|j"j$y |jtj&k(ry |jj)d|jy ) z(Handle incoming packets from the server.UNKNOWNzReceived packet %s data %smessageTr1)r?r7z%Received unexpected packet of type %sN)r}lenr packet_namesrmrqrr(bytesr+r6rr)r*rr3r0r7SERVER_DISCONNECTNOOPrn)rr packet_names rrzClient._receive_packetsV%8%8!99))#//:?H   (+&sxx7CHHZ I ??fnn ,    388t  D __ +   fmmFKKB C __ , OO$t{{/L/LO M __ +  KK  E!oo /r%c |jdk7ry|jj||jj dt j |jt|jts|jydy)z(Queue a packet to be sent to the server.r.NzSending packet %s data %sr) rr4r5rmrqr rr}rr(r)rrs rr)zClient._send_packetsi :: $  s  '    0&sxx7CHH I>H Ir%cn|jtj|_|jsd|j_ |jj |||||S#tj j$r3}|jjd|||t|cYd}~Sd}~wwxYw)NF)rr(r`z+HTTP %s request to %s failed with error %s.) rrlSessionrrrequestrRequestExceptionrmrqr)rmethodrrbodyr`excs rrrzClient._send_requests 99  ((*DI$DII  99$$VS'-4%6 6""33  KK  J#S# /s8O sA$$B4(B/)B4/B4ch|jdd}||jvr6|r |j|j|g|S |j||Sy#t$r+|dk(r$t |dk(r|j|cYSwxYw#|j j |dzYyxYw)zInvoke an event handler.r2Fr0rz handler errorN)rhandlersrI TypeErrorrrm exception)reventrBrCr2s rr6zClient._trigger_eventsJJ{E2 DMM !1t11$--2FNNN D "3t}}U3T:: "%" L0 #D Q$84==#7#99!"DKK))%2B*BCs$A0B B B  B B1c|jdk(r|jr|jjd|jz|j d|j|j zt|j|jdz}|t|tr;|jj|xsd|jjdn|jdks|jd k\rB|jjd |j|jjdno t!j"|j$j'd  }|j*D]}|j-||jdk(r|jr|jr5|jjd|jj/|jdk(rX|j1d|j2j4d t6j8j;||j=|jjdy#t($r:|jjd |jjdYwxYw#t($rY}wxYw)z-Read packets by polling the Engine.IO server.r.zSending polling GET request to r_r`N*Connection refused by the server, abortingrarb6Unexpected status code %s in server response, abortingrcrdz'Unexpected packet from server, aborting"Waiting for write loop task to endr0Fr1Exiting read loop task)rrrmrqrprrrsmaxrrrrrr4r5rvr ryrzr{rr|rr"r6r7TRANSPORT_ERRORrr<r=r>)rrrrs rrzClient._read_loop_pollingsCjjK'D,@,@ KK  1DMMA C""t}}t'>'>'@@D..0A0ABQF#HAyJq#. ##EEG t$}}s"amms&: ##%9:;--I t$ OOAII4D4DW4MN yy *$$S) */jjK'D,@,@4    KK  A B  % % ' :: $    dkk.I.I*/  1 --44T: KKM 12'  ##=? t$     s$/I6+J<6?J98J9< KKc|jdk(rd} |jj}t|dk(r*|jjst j  t%j&| }|j)||jdk(r|j*r5|jj!d |j*j-|jdk(rX|j/d |j0j2d  t4j6j9||j=|jj!dy#t j$r:|jjd|jjdYt j $r:|jjd|jjdY[t$r}t|tur+|jdk(r|jj!dn%|jj!dt#||jjdYd}~d}~wwxYw#t$rK}|jj!d t#||jjdYd}~)rrrrs rrzClient._read_loop_websocket sJjjK'A GGLLNq6Q;tww'8'8#FFHH* mm15   %EjjK'H    KK  A B  % % ' :: $    dkk.I.I*/  1 --44T: KKM 12Q66  ##@B t$??  ##?A t$ 7g%!''Q,KK$$CEKK$$KA  t$     FAP t$    sRAE!&J K!!A J-AJ8JA>'>:;4&  $$&tzz~~E~'BCr{*")#2, ,,.%%2OOG4&&DMM +\: 00'2#+CJJ((*+9 1c 2KK''IIK==3&!--3*>KK'')=>?mmM+/D( &/),:: GG//? GGLL8 ,,. /_jjK'v 23g##  !!"CD  ++H"DD'2KK''CE s6I/9+J A4J5/*JJJ21J259K10K1)NNz engine.io)FN)r)NNN)__name__ __module__ __qualname____doc__rr$r,r0rIrLrQrTrWr>rrrr)rrr6rrr __classcell__)r]s@rr r s"F59)!/F'D0&#, 0;%zfP/&I37 D*'3R13f?4r%r )base64r http.cookiesrloggingr4rrErKr engineio.jsonrrl ImportErrorrrrrr r getLoggerdefault_logger BaseClientr r%rrs%  )"""#45` 4[ # #` 4HIs"A5B5A?>A?B  B