Xh(crddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z ddl m Z mZmZmZmZmZmZdd lmZmZdd lmZmZmZmZdd lmZmZd d lm Z m!Z!d dl"m#Z#m$Z$e rkddl%m&Z&ddlm'Z'ddlm(Z(d dl)m*Z*d dl+m,Z,d dl-m.Z.m/Z/m0Z0d dl1m2Z2m3Z3edZ4ede5e(gdfZ6edee5e7fgdfZ8edegdfZ9edgdfZ:ede;e7gdfZ<ee dZ GddeZ=GddeZ>Gdde>e Z?Gd d!e>e!Z@dd" d)d#ZAGd$d%ZBGd&deBZCGd'd(eBZDy)*) annotationsN)IntEnum)partial)dumpsloads)select) TYPE_CHECKINGAnyCallableLiteralOptionalTypeVarUnion)CURL_SOCKET_BAD get_selector) CurlECodeCurlInfoCurlOpt CurlWsFlag)Curl CurlError) SessionClosedTimeout)not_setset_curl_options)Self)CurlHttpVersion) CurlWsFrame) CookieTypes) HeaderTypes)BrowserTypeLiteralExtraFingerprints ExtraFpDict) AsyncSession ProxySpecT WebSocket),:) separatorscDeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZy) WsCloseCodeiiiiiiiiiiiiN)__name__ __module__ __qualname__OK GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATAUNKNOWNABNORMAL_CLOSURE INVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATER BAD_GATEWAYU/home/Script/Script_env/lib/python3.12/site-packages/curl_cffi/requests/websockets.pyr.r.0sL BJNGLONOOKrCr.c.eZdZdZ d dfd ZxZS)WebSocketErrorzWebSocket-specific error.c&t|||yN)super__init__)selfmessagecode __class__s rDrJzWebSocketError.__init__Ds $'rC)r)rLstrrMz)Union[WsCloseCode, CurlECode, Literal[0]])r1r2r3__doc__rJ __classcell__rNs@rDrFrFAs'#OP(("K((rCrFceZdZdZy)WebSocketClosedzWebSocket is already closed.Nr1r2r3rPrBrCrDrTrTJs&rCrTceZdZdZy)WebSocketTimeoutzWebSocket operation timed out.NrUrBrCrDrWrWNs(rCrWtimeoutcKj}j|jd|jfd t j ||d{y7#tj $rYywxYww)Nc&jSrH) remove_reader)_fdloops rDzaselect..Wst'9'9"'=rCFT) create_future add_reader set_resultadd_done_callbackasynciowait_for TimeoutError)r^r_rYfutures`` rDaselectriRsx   !FOOB))40 =>vw///  0   s<ABA)!A'"A)&B'A))A?<B>A??BczeZdZdddd dZedZed dZeddZeddZ edd Z d Z y ) BaseWebSocketTF autoclosedebugcX||_||_d|_d|_||_d|_y)NF)_curlrm _close_code _close_reasonrnclosed)rKcurlrmrns rDrJzBaseWebSocket.__init__`s. (*.,0  rCct|jturt|j|_|jS)N)rn)rprrrnrKs rDrtzBaseWebSocket.curlhs( :: DJJ/DJzzrCc|jS)z6The WebSocket close code, if the connection is closed.)rqrvs rD close_codezBaseWebSocket.close_codensrCc|jS)z8The WebSocket close reason, if the connection is closed.)rrrvs rD close_reasonzBaseWebSocket.close_reasonss!!!rCc4tjd||zS)N!H)structpack)rMreasons rD_pack_close_framezBaseWebSocket._pack_close_framexs{{4&//rCct|dkrtj}d}||fS tjd|d}|ddj }|dkr'|tvs|dk(rt dtj||fS#t$r }t dtj|d}~wt$r }t d tj|d}~wwxYw) Nrr|ri r0zInvalid close codezInvalid close messagezInvalid close frame) lenr.r8r} unpack_fromdecoderFr6UnicodeDecodeErrorr: Exception)framerMres rD_unpack_close_framez!BaseWebSocket._unpack_close_frame|s u:>&&DF$V|! ))$6q9qr))+$;D $;tt|(,k.H.HV|& $+[-E-E $);+E+E s#,B C B&& C2C  CcFd|_|jjy)$Terminate the underlying connection.TN)rsrtcloservs rD terminatezBaseWebSocket.terminates  rCN)rtrrmboolrnr)returnz Optional[int])r Optional[str])rMintrbytesrr)rrrztuple[int, str]) r1r2r3rJpropertyrtrxrz staticmethodrrrrBrCrDrkrk_su8>%%j1  ,%%  ,!%!3!3G!<!"4++" ,s * A(&A##A(zgzip, deflate, brrrcF|j}tdid|ddd|dd|gdd|gdd|gd |d |d |d |d d| gd| d| dd| gd| d|d|d|d|d|d|d|d|d|d|d|d||jtjd|j |S) a Connect to the WebSocket. libcurl automatically handles pings and pongs. ref: https://curl.se/libcurl/c/libcurl-ws.html Args: url: url for the requests. params: query string for the requests. headers: headers to send. cookies: cookies to use. auth: HTTP basic auth, a tuple of (username, password), only basic auth is supported. timeout: how many seconds to wait before giving up. allow_redirects: whether to allow redirection. max_redirects: max redirect counts, default 30, use -1 for unlimited. proxies: dict of proxies to use, prefer to use ``proxy`` if they are the same. format: ``{"http": proxy_url, "https": proxy_url}``. proxy: proxy to use, format: "http://user@pass:proxy_url". Can't be used with `proxies` parameter. proxy_auth: HTTP basic auth for proxy, a tuple of (username, password). verify: whether to verify https certs. referer: shortcut for setting referer header. accept_encoding: shortcut for setting accept-encoding header. impersonate: which browser version to impersonate. ja3: ja3 string to impersonate. akamai: akamai string to impersonate. extra_fp: extra fingerprints options, in complement to ja3 and akamai str. default_headers: whether to set default browser headers. default_encoding: encoding for decoding response content if charset is not found in headers. Defaults to "utf-8". Can be set to a callable for automatic detection. quote: Set characters to be quoted, i.e. percent-encoded. Default safe string is ``!#$%&'()*+,/:;=?@[]~``. If set to a sting, the character will be removed from the safe string, thus quoted. If set to False, the url will be kept as is, without any automatic percent-encoding, you must encode the URL yourself. curl_options: extra curl options to use. http_version: limiting http version, defaults to http2. interface: which interface to use. cert: a tuple of (cert, key) filenames for client cert. max_recv_speed: maximum receive speed, bytes per second. curl_options: extra curl options to use. rtmethodGETurl params_listN headers_list cookies_listauthrYallow_redirects max_redirects proxies_listproxy proxy_auth verify_listrefereraccept_encoding impersonateja3akamaiextra_fpdefault_headersquote http_version interfacemax_recv_speedcert curl_optionsrrB)rtrsetoptr CONNECT_ONLYperform)rKrparamsheaderscookiesrrYrrproxiesrrverifyrrrrrrrrrrrrrrts rDconnectzWebSocket.connectsXPyy    v        , (   " v  ,! "$# $% &' () *,+ ,- .&/ 0 1 2*3 45 6&7 > G((!,  rCc|jr td|jj\}}|jt j zr; |j|\|_|_ |jr|j||fS#t$r2}|j|_|j|jd}~wwxYw)z Receive a single frame as bytes.rN)rsrTrtws_recvrrrrrqrrrFrMrrm)rKchunkrrs rD recv_fragmentzWebSocket.recv_fragmentHs ;;!"78 8yy((* u ;;)) ) 7;7O7OPU7V4 $"4~~ e|" $%66  166"  sB C -CC c g}d}|jjtj}|tk(rt dt j |j\}}|j}|j||jdk(r|tjzdk(rnXdj%||fS#t$r<}|jt j k(rt#|gggd\}}}nYd}~Td}~wwxYw)z Receive a frame as bytes. libcurl splits frames into fragments, so we have to collect all the chunks for a frame. rInvalid active socketg?NrC)rtgetinfor ACTIVESOCKETrrFrNO_CONNECTION_AVAILABLErrappend bytesleftrCONTrrMAGAINrjoin)rKchunksrsock_fdrrrr]s rDrzWebSocket.recv\s))##H$9$9: o % ')J)J  #113 u  e$??a'EJOO,Cq,H xx&& 66Y__,%gYB !,,.!,,.rCcv|j"ttj|_|jSrH)r/rreget_running_looprvs rDr_zAsyncWebSocket.loop!s+ :: %g&>&>&@ADJzzrCc4|jr td|Srrrvs rD __aiter__zAsyncWebSocket.__aiter__'rrCczK|jd{\}}|tjzrt|S7"wrH)rrrStopAsyncIterationrs rD __anext__zAsyncWebSocket.__anext__,s699;& U :## #$ $ 's ;9#;NrXczK|jr td|jjr t d|j4d{ t j |jjd|jj|d{\}}|jtjzrX |j!|x\}}\|_|_|j,r(|j+||j/d{dddd{||fS77#t j$r}td|d}~wwxYw#t&$r;}|j(|_|j+|j(d{7d}~wwxYw77|#1d{7swYfSxYww)zxReceive a single frame as bytes. Args: timeout: how many seconds to wait before giving up. rz1Concurrent call to recv_fragment() is not allowedNz#WebSocket recv_fragment() timed out)rsrTr1locked TypeErrorrerfr_run_in_executorrtrrgrWrrrrrqrrrFrMrrmr)rKrYrrrrMrLs rDrzAsyncWebSocket.recv_fragment2s ;;!"78 8 ?? ! ! #OP P?? = = U%,%5%5II--dDII4E4EF&  u {{Z--- 007MD'$HD$4d6H>>**T7>>+;<<<' = =*e|+ = '' U&'LMSTT U&()vvD$**QVV,,, =' = = = =*e|sA F;D-F;F#AD1D/D1$F##E%/F#FF# F;$F!% F;/D11E EEF# F!/FFFFF#!F;#F8)F, *F81 F;cvK|j}g}d}|jd|jjtj d{}|t k(rtdtj |j|d{\}}|j}|j||jdk(r|tjzdk(rnbdj)||fS77`#t $r@}|j"tj$k(rt'|||d{7nYd}~\d}~wwxYww)z Receive a frame as bytes. libcurl splits frames into fragments, so we have to collect all the chunks for a frame. Args: timeout: how many seconds to wait before giving up. rNrrX)r_rYrC)r_r=rtrrrrrFrrrrrrrrrrMrrir) rKrYr_rrrrrrs rDrzAsyncWebSocket.recvVs$yy,, $))##X%:%:   o % ')J)J  %)%7%7%7%HH u  e$??a'EJOO,Cq,H xx&&)  I  66Y__,!'gFFFG s[A D9 C) (D95C- C+ A C-D9+C-- D66.D1$D'%D1,D91D66D9cK|j|d{\}}|tjzstdtj |j S7Dw)zmReceive a text frame. Args: timeout: how many seconds to wait before giving up. rXNrr)rKrYrrs rDrzAsyncWebSocket.recv_strzsO !IIgI66 ez& +2J2JK K{{}7sAAAA)rrYcRK|j|d{}||S7 w)zReceive a JSON frame. Args: loads: JSON decoder, default is json.loads. timeout: how many seconds to wait before giving up. rXNr)rKrrYrs rDrzAsyncWebSocket.recv_jsons*]]7]33T{4s '% 'cnK|jr tdt|tr|j }|j 4d{|j jd|jj||d{cdddd{S7O77 #1d{7swYyxYww)rrN) rsrTrrOrr2r_r=rtrrs rDrzAsyncWebSocket.sends ;;!"78 8 gs #nn&G??  22dii''%       sZAB5 B B56B BB  B5BB5B B5 B2&B) 'B2.B5c^K|j|tjd{S7w)rNrrs rDrzAsyncWebSocket.send_binary% YYw (9(9:::: $-+-c^K|j|tjd{S7w)rNrrs rDrzAsyncWebSocket.send_bytesrCrDc^K|j|tjd{S7w)rNr rs rDr zAsyncWebSocket.send_str# YYw 8888rDr cLK|j||d{S7w)rNrrs rDrzAsyncWebSocket.send_jsons!]]5>2222s $"$c^K|j|tjd{S7w)rNrrs rDrzAsyncWebSocket.pingrGrDrCcK|j||}|j|tjd{|j y7wr)rrrrrrs rDrzAsyncWebSocket.closesC$$T73iiZ--...  /s6AAAct||jjs|jj dyy)rN)rIrr._closed push_curl)rKrNs rDrzAsyncWebSocket.terminates2 ||## LL " "4 ($rC)r.r&rtrrmrrnr)rrr)rYOptional[float]rr!)rYrNrr")rYrNrrO)rr#rYrNrr(r$r&r'r(r)r*)r1r2r3rPrJrr_r6r9rrrrrrrrrrr rrrr.r4rrrQrRs@rDr,r,s: ) ) )  )  )  -1")" ""H8<"'H<@ .3t * =L   ?I>O>O(1;,;;9NS39'2nns ))rCr,)r_zasyncio.AbstractEventLooprYrNrr)E __future__rrer}enumr functoolsrjsonrrrtypingr r r r r rraiorrconstrrrrrtrr exceptionsrrutilsrrtyping_extensionsrrr rr!rr"rr#r$r%r.r&r'r(r ON_DATA_TrO ON_MESSAGE_T ON_ERROR_T ON_OPEN_Tr ON_CLOSE_Tr.rFrTrWrirkr)r,rBrCrDr^s]" 0<<".,&'"$$OO0 A+uk:D@AI[%s *;