a j3@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z ddlmZmZzddlmZWneyddlmZYn0gdZGd d d ejZejZejZejZejZej Z!ej"Z#eeefZ$ee!e#fZ%Gd d d ejZ&e&j'd e&j(de&j)de&j*de&j+de&j,de&j-de&j.de&j/de&j0de&j1de&j2de&j3de&j4de&j5diZ6de7d<e&j'e&j(e&j)e&j*e&j-e&j.e&j/e&j0e&j1e&j2e&j3e&j4h Z8e&j'e&j(e&j+hZ9e:e;eGdddZ?ej>Gd d!d!Z@dd"lAmBZBdS)#) annotationsN) GeneratorSequence)CallableUnion) PayloadTooBig ProtocolError) apply_mask) OpcodeOP_CONTOP_TEXT OP_BINARYOP_CLOSEOP_PINGOP_PONG DATA_OPCODES CTRL_OPCODES CloseCodeFrameClosec@s$eZdZdZd\ZZZd\ZZZ dS)r z#Opcode values for WebSocket frames.)rr) N) __name__ __module__ __qualname____doc__CONTTEXTBINARYCLOSEPINGPONGr%r%7E:\soccer\ml_env\Lib\site-packages\websockets/frames.pyr %s r c@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS)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%r&r7s 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_EXPLANATIONSc@seZdZUdZded<ded<dZded<d Zded <d Zded <d Zded <e e j d dZ ddddZedddddddddddZdddddddd Zd!dd"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 opcodez#Union[bytes, bytearray, memoryview]dataTboolfinFrsv1rsv2rsv3ZWEBSOCKETS_MAX_LOG_SIZEZ75strreturnc Csd}t|jdt|jdkr"dnd}|jr4dnd}|jturXtt|j}n8|jtur|j}t||j dkr|j ddd}d |dd |d || dg}d d d |D}n|jt urt t |j}n|jrztt|j}d}Wnttfy|j}t||j dkrl|j ddd}d |dd |d || dg}d dd |D}d}Yn0nd}t||j kr|j dd}|dd |d|| d}d td|||g}|jjd |d|dS)zE Return a human-readable representation of a frame. Nz byters continuedrs css|]}|dVqdS02xNr%.0byter%r%r& rFz Frame.__str__..textcss|]}|dVqdSrHr%rJr%r%r&rMrFbinaryz''z...z, z [])lenr9r;r8r reprbytesdecoder MAX_LOG_SIZEjoinrr?rparseUnicodeDecodeErrorAttributeErrorfiltername)selfZcodinglengthZ non_finalr9rOcutmetadatar%r%r&__str__s<&  & & "z Frame.__str__N)max_size extensionsz-Callable[[int], Generator[None, None, bytes]]z int | Nonez%Sequence[extensions.Extension] | NonezGenerator[None, None, Frame]) read_exactmaskrarbrAc cs|dEdH}td|\}}|d@r*dnd}|d@r:dnd} |d@rJdnd} |d @rZdnd} zt|d @} Wn.ty} ztd | WYd} ~ n d} ~ 00|d@rdnd|krtd |d @}|dkr|dEdH}td|\}n&|d kr|dEdH}td|\}|dur,||kr,t|||r@|dEdH}||EdH}|r^t||}|| ||| | | }|dur~g}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. PayloadTooBig: If the frame's payload size exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rN!BBTF@ zinvalid opcodezincorrect masking~!Hrz!Q)ra) structunpackr ValueErrorr rr reversedrTcheck)clsrcrdrarbr9head1head2r;r<r=r>r8excr] mask_bytesframe extensionr%r%r&rWsB      z Frame.parserbrS)rdrbrAc Cs(||durg}|D]}||}qt}|jr:dnd|jrFdndB|jrTdndB|jrbdndB|jB}|rvdnd}t |j }|dkr| t d|||Bn>|d kr| t d ||dB|n| t d ||d B||rtd }| ||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. Nrfrrgrhrirlreiz!BBHz!BBQrkrn)rsencodeioBytesIOr;r<r=r>r8rQr9writeropacksecretsZ token_bytesr getvalue) r\rdrbrzoutputrurvr]rxr9r%r%r& serializes>          zFrame.serializeNonecCsL|js|js|jrtd|jtvrHt|jdkr:td|jsHtddS)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 r8rrQr9r;r\r%r%r&rsOs z Frame.check)rrrr__annotations__r;r<r=r>intosenvirongetrUr` classmethodrWrrsr%r%r%r&r|s     2M;rc@s`eZdZUdZded<ded<ddddZed dd d d Zd dd dZddddZ dS)rz Code and reason for WebSocket close frames. Attributes: code: Close code. reason: Close reason. rcoder?reasonr@cCszd|jkrdkr nnd}n.d|jkr6dkr@nnd}nt|jd}|jd|d}|jrv|d |j}|S) zU Return a human-readable representation of a close code and reason. i registeredz private useunknownz ()rG)rr7rr)r\Z explanationresultr%r%r&r`osz Close.__str__rS)r9rAcCslt|dkrHtd|dd\}|dd}|||}||St|dkr`|tjdStddS)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. rrmNrrBzclose frame too short)rQrorprTrsrr+r )rtr9rrcloser%r%r&rWs    z Close.parsecCs |td|j|jS)z: Serialize the payload of a close frame. rm)rsrorrrr|rr%r%r&rszClose.serializercCs.|jtvs*d|jkr dks*ntddS)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 rr%r%r&rss"z Close.checkN) rrrrrr`rrWrrsr%r%r%r&ras  rr{)C __future__r dataclassesenumr}rrrocollections.abcrrtypingrr exceptionsrr Zspeedupsr ImportErrorutils__all__IntEnumr rr r r r!rr"rr#rr$rrrrr'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r7rrZOK_CLOSE_CODESrS bytearray memoryviewZ BytesLike dataclassrrrBrbr%r%r%r&s     eL