o `^h1@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZzddlmZWneyHddlmZYnwgdZGdd d ejZejZejZejZejZejZej Z!eeefZ"eee!fZ#Gd d d ejZ$e$j%d e$j&d e$j'de$j(de$j)de$j*de$j+de$j,de$j-de$j.de$j/de$j0de$j1de$j2de$j3diZ4de5d<e$j%e$j&e$j'e$j(e$j+e$j,e$j-e$j.e$j/e$j0e$j1e$j2h Z6e$j%e$j&e$j)hZ7e8e9e:fZ;ejdd!l?m@Z@dS)") annotationsN)Callable GeneratorSequence) PayloadTooBig ProtocolError) apply_mask) OpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODESFrameClosec@s$eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGr#r#O/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/websockets/frames.pyr #s r c@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS) CloseCodez-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#r$r%5s"r%OKz 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_EXPLANATIONSc@seZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <e e j d dZ d%ddZedddd&ddZddd'd d!Zd(d#d$ZdS))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 opcodebytesdataTboolfinFrsv1rsv2rsv3WEBSOCKETS_MAX_LOG_SIZE75returnstrc Csd}t|jdt|jdkrdnd}|jrdnd}|jtur)t|j}n|jtura|j}t||jdkrV|jddd}d |dd |d || dg}d d d |D}n^|jt urot t |j}nP|jrz t|j}d}WnAttfy|j}t||jdkr|jddd}d |dd |d || dg}d dd |D}d}Ynwd}t||jkr|jdd}|dd |d|| d}d td|||g}|jjd |d|dS)zE Return a human-readable representation of a frame. Nz byters continuedrs cs|]}|dVqdS02xNr#.0byter#r#r$ z Frame.__str__..textcsrIrJr#rLr#r#r$rOrPbinaryz''z...z, z [])lenr9r;r7r reprdecoder MAX_LOG_SIZEjoinrrBrparseUnicodeDecodeErrorAttributeErrorfiltername)selfcodinglength non_finalr9rRcutmetadatar#r#r$__str__s>&  & &"z Frame.__str__N)max_size extensions read_exact-Callable[[int], Generator[None, None, bytes]]maskre int | Nonerf%Sequence[extensions.Extension] | NoneGenerator[None, None, Frame]c cs|dEdH}td|\}}|d@rdnd}|d@rdnd} |d@r&dnd} |d @r.dnd} zt|d @} WntyI} ztd | d} ~ ww|d@rPdnd|krXtd |d @}|dkro|dEdH}td|\}n|d kr|dEdH}td|\}|dur||krtd|d|d|r|dEdH}||EdH}|rt||}|| ||| | | }|durg}t|D] }|j||d}q| |S)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. UnicodeDecodeError: If the frame contains invalid UTF-8. PayloadTooBig: If the frame's payload size exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rN!BBTF@ zinvalid opcodezincorrect masking~!Hrz!Qzover size limit (z > z bytes))re) structunpackr ValueErrorrrr reversedrVcheck)clsrgrirerfr9head1head2r;r<r=r>r7excr` mask_bytesframe extensionr#r#r$rYsH   z Frame.parserfc Cs&||dur g}|D]}||}q t}|jrdnd|jr#dndB|jr*dndB|jr1dndB|jB}|r;dnd}t |j }|dkrS| t d|||Bn|d kre| t d ||dB|n | t d ||d B||r~td }| ||rt|j |} n|j } | | |S)aH Serialize a WebSocket frame. Args: mask: Whether the frame should be masked i.e. whether the write happens on the client side. extensions: List of extensions, applied in order. Raises: ProtocolError: If the frame contains incorrect values. Nrnrrorprqrtrmiz!BBHz!BBQrsrv)r{encodeioBytesIOr;r<r=r>r7rTr9writerwpacksecrets token_bytesr getvalue) r^rirfroutputr}r~r`rr9r#r#r$ serializes>          zFrame.serializeNonecCsP|js |js |jr td|jtvr$t|jdkrtd|js&tddSdS)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>rr7rrTr9r;r^r#r#r$r{Ns z Frame.checkrArB) rgrhrir:rerjrfrkrArl)rir:rfrkrAr8rAr)rrrr__annotations__r;r<r=r>intosenvirongetrWrd classmethodrYrr{r#r#r#r$rzs"      2N;rc@sNeZdZUdZded<ded<dddZedd d Zdd dZdddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. rcoderBreasonrAcCszd|jkr dkrnnd}nd|jkrdkr nnd}nt|jd}|jd|d}|jr;|d |j}|S) zU Return a human-readable representation of a close code and reason. i registeredz private useunknownz ()rH)rr6rr)r^ explanationresultr#r#r$rdnsz Close.__str__r9r8cCsht|dkr$td|dd\}|dd}|||}||St|dkr0|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. rruNrrCzclose frame too short)rTrwrxrVr{r%r*r)r|r9rrcloser#r#r$rYs    z Close.parsecCs |td|j|jS)z: Serialize the payload of a close frame. ru)r{rwrrrrrr#r#r$rszClose.serializercCs8|jtvsd|jkrdkstdtddSdS)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_CODESrrr#r#r$r{s z Close.checkNr)r9r8rAr)rAr8r) rrrrrrdrrYrr{r#r#r#r$r`s     rr)A __future__r dataclassesenumrrrrwtypingrrr exceptionsrrspeedupsr ImportErrorutils__all__IntEnumr rr rr rr r rr!rr"rrrr%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r6rrOK_CLOSE_CODESr8 bytearray memoryview BytesLike dataclassrrrCrfr#r#r#r$s     fL