a j@sddlmZddlZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZmZddl mZdd lmZzdd lmZWneydd lmZYn0Gd d d e Zd ddddZd ddddZeZddl mZdddddZddddddZdS)) annotationsN) AwaitableSequence)AnyCallable NamedTuple) extensionsframes) PayloadTooBig ProtocolError) BytesLike)Data) apply_maskc@seZdZUded<ded<ded<dZded<dZded <dZded <ed d d dZdd ddZ dd ddZ e dddddddddddZ dddddddd d!Z dS)"Frameboolfinz frames.OpcodeopcodebytesdataFrsv1rsv2rsv3z frames.Frame)returncCs t|j|j|j|j|j|jSN)r rrrrrrrselfr?E:\soccer\ml_env\Lib\site-packages\websockets/legacy/framing.py new_frameszFrame.new_framestrcCs t|jSr)r rrrrr__str__&sz Frame.__str__NonecCs |jSr)rcheckrrrrr#)sz Frame.checkN)max_sizer z!Callable[[int], Awaitable[bytes]]z int | Nonez%Sequence[extensions.Extension] | None)readermaskr$r rc s|dIdH}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|dIdH}td|\}n&|d kr|dIdH}td|\}|dur.||kr.t|||rB|dIdH}||IdH}|r`t||}t| ||| | | }|durg}t |D]}|j ||d}q| ||j |j |j|j|j|jS)a@ Read a WebSocket frame. Args: reader: Coroutine that reads exactly the requested number of bytes, unless the end of file is reached. 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: PayloadTooBig: If the frame exceeds ``max_size``. ProtocolError: If the frame contains incorrect values. rNz!BBTF@ zinvalid opcodezincorrect masking~z!Hz!Q)r$)structunpackr Opcode ValueErrorr r rrreverseddecoder#rrrrrr)clsr%r&r$r rZhead1Zhead2rrrrrexclengthZ mask_bitsr extensionrrrread,sP      z Frame.read)r zCallable[[bytes], Any])writer&r rcCs||jj||ddS)a Write a WebSocket frame. Args: frame: Frame to write. write: Function that writes bytes. 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. )r&r N)r serialize)rr;r&r rrrr;{sz Frame.write)__name__ __module__ __qualname____annotations__rrrpropertyrr!r# classmethodr:r;rrrrrs     Srrztuple[int, bytes])rrcCs<t|trtjj|fSt|tr0tjj|fStddS)a Convert a string or byte-like object to an opcode and a bytes-like object. This function is designed for data frames. If ``data`` is a :class:`str`, return ``OP_TEXT`` and a :class:`bytes` object encoding ``data`` in UTF-8. If ``data`` is a bytes-like object, return ``OP_BINARY`` and a bytes-like object. Raises: TypeError: If ``data`` doesn't have a supported type. data must be str or bytes-likeN) isinstancer r r2ZTEXTencoder BINARY TypeErrorrrrr prepare_datas    rIrcCs0t|tr|St|tr$t|StddS)ai Convert a string or byte-like object to bytes. This function is designed for ping and pong frames. If ``data`` is a :class:`str`, return a :class:`bytes` object encoding ``data`` in UTF-8. If ``data`` is a bytes-like object, return a :class:`bytes` object. Raises: TypeError: If ``data`` doesn't have a supported type. rCN)rDr rEr rrGrHrrr prepare_ctrls   rJ)Closeztuple[int, str]cCst|}|j|jfS)z Parse the payload from a close frame. Returns: Close code and reason. Raises: ProtocolError: If data is ill-formed. UnicodeDecodeError: If the reason isn't valid UTF-8. )rKparsecodereason)rcloserrr parse_closes rPintr )rMrNrcCst||S)z3 Serialize the payload for a close frame. )rKr<)rMrNrrrserialize_closesrR) __future__rr0collections.abcrrtypingrrrr r exceptionsr r r rZspeedupsr ImportErrorutilsrrIrJZ encode_datarKrPrRrrrrs&