U g@sddlmZddlZddlmZmZmZmZmZddl m Z m Z ddl m Z mZddl mZddlmZzdd lmZWn ek rdd lmZYnXGd d d eZd d dddZd ddddZeZddl mZdddddZddddddZdS)) annotationsN)Any AwaitableCallable NamedTupleSequence) 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=/tmp/pip-unpacked-wheel-dx_q7dq3/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,tk r} ztd | W5d} ~ XYnX|d@rdnd|krtd |d @}|dkr|dIdH}td|\}n&|d kr|dIdH}td|\}|dk r8||kr8td|d|d|rL|dIdH}||IdH}|rjt||}t| ||| | | }|dkrg}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!Qzover size limit (z > z bytes))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;zsz 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 ZBINARY TypeErrorrrrr prepare_datas    rHrcCs0t|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 rrFrGrrr prepare_ctrls   rI)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. )rJparsecodereason)rcloserrr parse_closes rOintr )rLrMrcCst||S)z3 Serialize the payload for a close frame. )rJr<)rLrMrrrserialize_closesrQ) __future__rr0typingrrrrrr r exceptionsr r r rZspeedupsr ImportErrorutilsrrHrIZ encode_datarJrOrQrrrrs$