A>qjEAUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddlmZmZddl mZ ddlmZgd ZGd d ej.Zej2Zej4Zej6Zej8Zej:Zej<ZeeefZeeefZ Gd d ej.Z!e!jDde!jFde!jHde!jJde!jLde!jNde!jPde!jRde!jTde!jVde!jXde!jZde!j\de!j^de!j`diZ1de2d<e!jDe!jFe!jHe!jJe!jPe!jRe!jTe!jVe!jXe!jZe!j\e!j^h Z3e!jDe!jFe!jLhZ4ejjGdd Z6ejjGd!d"Z7 d% d&d#Z8dd$l9m:Z:y#e$r ddlmZY9wxYw)') annotationsN) GeneratorSequence)CallableSelf) PayloadTooBig ProtocolError) BytesLike) apply_mask)Opcode CloseCodeFrameClosec(eZdZdZd\ZZZd\ZZZ y)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGP/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/websockets/frames.pyr r s-)D$(E4r!r cLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZy)rz-Close code values for WebSocket close frames.iiiiiiiiiiiiiiiN)rrrrNORMAL_CLOSURE GOING_AWAYPROTOCOL_ERRORUNSUPPORTED_DATANO_STATUS_RCVDABNORMAL_CLOSURE INVALID_DATAPOLICY_VIOLATIONMESSAGE_TOO_BIGMANDATORY_EXTENSIONINTERNAL_ERRORSERVICE_RESTARTTRY_AGAIN_LATER BAD_GATEWAY TLS_HANDSHAKEr r!r"rr0sT7NJNNLONOOKMr!rOKz going awayzprotocol errorzunsupported datazno status received [internal]zabnormal closure [internal]zinvalid frame payload datazpolicy violationzmessage too bigzmandatory extensionzinternal errorzservice restartztry again laterz bad gatewayz TLS handshake failure [internal]zdict[int, str]CLOSE_CODE_EXPLANATIONSceZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <e e jjd dZ eded ediZddZddZeddd ddZdd ddZddZy)raN WebSocket frame. Attributes: opcode: Opcode. data: Payload data. fin: FIN bit. rsv1: RSV1 bit. rsv2: RSV2 bit. rsv3: RSV3 bit. Only these fields are needed. The MASK bit, payload length and masking-key are handled on the fly when parsing and serializing frames. r opcoder dataTboolfinFrsv1rsv2rsv3WEBSOCKETS_MAX_LOG_SIZE75c|jj|j}|j\}}||k(rdn|rdnd}t |j dt |j dk(rdnd}|j rdnd}djtd |||g}|jjd |d |d S) ax Return a human-readable representation of a frame. This function is intended for logging and debugging. It doesn't aim to support round-tripping because payloads can be too long for displaying conveniently. Instead, it shows the beginning and the end. It's robust to incorrect data. It attempts to decode UTF-8 payloads whenever possible, even for binary frames and control frames, because those frequently contain UTF-8 data. It applies the same logic to continuation frames, because we don't know if they continue a text frame or a binary frame. textbinaryz byters continuedz, N z []) DEFAULT_IS_TEXTgetr6 _data_reprlenr7r9joinfiltername)self expect_text data_repris_text data_typelength non_finalmetadatas r"__str__z Frame.__str__s**..t{{; !__. 7%0Bvh  N#5s499~/B(LM((B 99VD9fi*HIJ++""#1YKr(1==r!c|js'd|jj|jfS|jtur+ t t j|jdfSt|jd|jzkDrt|jdd|jzdz}t|jd|jzdzd}t||jtk7xrt||j }|rzt!||zj#d }n[t|j}t||jtk7|j }|rt!|j#d }|sr|j}t||jdzkDr3|jdzd z dz}dj%|dd|zd|| dg}dj%d|D}t|jkDr$|jdzd z }|dd|zdz|| dz}||fS#ttf$rYwxYw)z Return a human-readable representation of the payload. Also returns whether the payload is text. The representation is elided to fit ``MAX_LOG_SIZE``. This is a helper for the __str__ method. z''TNr)must_start_clean)must_end_cleanreplace)errors)r[r\rr!rsrEc3$K|]}|d yw)02xNr ).0bytes r" z#Frame._data_repr..s B4D: Bsz...)r7rGrHr6rstrrparser UnicodeDecodeErrorrJ MAX_LOG_SIZEbytesis_utf8_fragmentrr9reprdecoderK)rN data_startdata_endrQrPr7rBcuts r"rIzFrame._data_reprsTyy--11$++>> > ;;%  5;;tyy12D88 tyy>A 1 11 1tyy)E1t/@/@+@A+EFGJTYYrD,=,='='B'DEFH&!%!4##xx  *x"7!?!?y!?!QR #D&!%!4#xxG  I!>? YYF6{T..!33((A-1a76)AG#4k63$%="QR B6 BBI y>D-- -##q(1,C!)AG,u4y#7GGI'!!e"#56  s)I I I N)max_size extensionsc#K|dEd{}tjd|\}}|dzrdnd}|dzrdnd} |dzrdnd} |d zrdnd} t|d z} |dzrdnd|k7r t d |d z}|dk(r)|dEd{}tjd|\}n-|d k(r(|dEd{}tjd|\}|||kDr t |||r|dEd{}||Ed{}|r t |}|| ||| | | }|g}t|D]}|j||}|j|S7Q#t$r} t d | d} ~ wwxYw7777~w)a Parse a WebSocket frame. This is a generator-based coroutine. Args: read_exact: Generator-based coroutine that reads the requested bytes or raises an exception if there isn't enough data. mask: Whether the frame should be masked i.e. whether the read happens on the server side. max_size: Maximum payload size in bytes. extensions: List of extensions, applied in reverse order. Raises: EOFError: If the connection is closed without a full WebSocket frame. PayloadTooBig: If the frame's payload size exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rN!BBTF@ zinvalid opcodezincorrect masking~!Hrz!QrX)ro) structunpackr ValueErrorr r r reversedrkcheck)cls read_exactmaskrorpr7head1head2r9r:r;r<r6excrS mask_bytesframe extensions r"rez Frame.parses:%Q-'}}UD1 uj(dez)tuz)tuz)tu ;EJ./FJ&DEd : 34 4# S=(m+D dD1IV s](m+D dD1IV  FX$51 1 $.qM1J%V,, dJ/DFD#tT48  J!*- ?I$$UX$>E ?   W( ; 01s : ;,, 2-sw FEAFE",F E? -F9F:B C e^ LLVUECKH I LLVUECKH I  ,,Q/J LL $ dii4D99D T  r!c|js|js |jr td|jt vr;t |jdkDr td|js tdyy)z Check that reserved bits and opcode have acceptable values. Raises: ProtocolError: If a reserved bit or the opcode is invalid. zreserved bits must be 0}zcontrol frame too longzfragmented control frameN) r:r;r<r r6 CTRL_OPCODESrJr7r9rNs r"rz Frame.checkwsh 99 TYY 9: : ;;, &499~##$<==88#$>?? 'r!returnrd)rztuple[str, bool | None]) rz9Callable[[int], Generator[None, None, bytes | bytearray]]rr8roz int | Nonerp%Sequence[extensions.Extension] | NonerzGenerator[None, None, Frame])rr8rprrrhrNone)rrrr__annotations__r9r:r;r<intosenvironrHrgrrrrGrVrI classmethodrerrr r!r"rrrs N OCD$D$D$rzz~~&?FGLT65%>O>2F"P $<@ GMG G  G : G &GGZ=A :!:!: :!  :!x@r!rcPeZdZUdZded<ded<d dZed dZd dZdd Z y )rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. zCloseCode | intcoderdreasoncd|jcxkrdkrnnd}n;d|jcxkrdkrnnd}n tj|jd}|jd|d}|jr|d |j}|S) zU Return a human-readable representation of a close code and reason. i registeredz private useunknownz ()rE)rr4rHr)rN explanationresults r"rVz Close.__str__s| 499 #t #&K TYY % %'K155diiKKII;b Q/ ;;xq .F r!cFt|tr tdt|dk\rIt j d|dd\}|ddj }|||}|j|St|dk(r|tjdStd)z Parse the payload of a close frame. Args: data: Payload of the close frame. Raises: ProtocolError: If data is ill-formed. UnicodeDecodeError: If the reason isn't valid UTF-8. z.only compressed outgoing frames use memoryviewrrzNrr@zclose frame too short) isinstance memoryviewAssertionErrorrJr{r|rkrrr(r )rr7rrcloses r"rez Close.parses dJ ' !QR R t9>mmD$r(3GT!"X__&Ff%E KKML Y!^y//4 4 78 8r!c|jtjd|j|jj zS)z: Serialize the payload of a close frame. rz)rr{rrrrrs r"rzClose.serializes4 {{4+dkk.@.@.BBBr!c|jtvs-d|jcxkrdkstdtdyy)z Check that the close code has a valid value for a close frame. Raises: ProtocolError: If the close code is invalid. rrzinvalid status codeN)rEXTERNAL_CLOSE_CODESr rs r"rz Close.checksH 11TTYY5M5M 56 66N 56 66N1r!Nr)r7r rr)rrhr) rrrrrrVrrerrr r!r"rrs8  K$992C 7r!rcdt|}}|s.tdt|}||kr||}|dzdk(r|dz } |s|dz}tt|dz |}||k\r^||}|dzdk(r|dz}|dzdk(rd}n#|dzdk(rd}n|d zdk(rd}n |d zd k(rd}nd}|t||z kr t|} |||j} d t | vS#t $rYy wxYw) z*Guess if data is a fragment of UTF-8 text.rrYrsrrXrz\xF)rJminmaxrkrjrf) r7r[r\startend max_startrbmin_endseq_lenrAs r"ririsICI3E 3t9% i;Dj J.    qc$i!mU+Wn9Dj J.qj J. "j0 "j0 "j0#d)c/)$i 'E#%%' DJ&& sC!! C-,C-r)FF)r7rhr[r8r\r8rr8); __future__r dataclassesenumrrrr{collections.abcrrtypingrr exceptionsr r r speedupsr ImportErrorutils__all__IntEnumr rrrrrr DATA_OPCODESrrr$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r4rrOK_CLOSE_CODES dataclassrrrir@rpr r!r"rs"  /!4"$  )T\\){{ {{   {{ {{T6! dD  .d , .  2 =  = 8  2 0 !!#8 . 0 0 = ?+,       !!     "   S@S@S@l J7J7J7^# E' E'E'E' E'R]"!!"sI00 I?>I?