o _i@sddlmZddlZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZmZddlmZmZzdd lmZWneyKdd lmZYnwGd d d e ZdddZd ddZeZddl mZd!ddZd"ddZdS)#) annotationsN) AwaitableSequence)AnyCallable NamedTuple) extensionsframes) PayloadTooBig ProtocolError) BytesLikeDataLike) apply_maskc@seZdZUded<ded<ded<dZded<dZded <dZded <ed%d dZd&ddZ d'ddZ e dddd(ddZ dd d)d#d$Z dS)*Frameboolfinz frames.Opcodeopcoder dataFrsv1rsv2rsv3return frames.FramecCs t|j|j|j|j|j|jSN)r rrrrrrrselfr[/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/websockets/legacy/framing.py new_frameszFrame.new_framestrcCs t|jSr)r rrrrr__str__% z Frame.__str__NonecCs |jSr)rcheckrrrrr$(r"z Frame.checkN)max_sizer reader!Callable[[int], Awaitable[bytes]]maskr% int | Noner %Sequence[extensions.Extension] | Nonec s|dIdH}td|\}}|d@rdnd}|d@rdnd} |d@r&dnd} |d @r.dnd} z t|d @} WntyJ} ztd | d} ~ ww|d@rQdnd|krYtd |d @}|dkrp|dIdH}td|\}n|d kr|dIdH}td|\}|dur||krt|||r|dIdH}||IdH}|rt||}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 rhead1head2rrrrrexclength mask_bitsr extensionrrrread+sV    z Frame.read)r writeCallable[[bytes], Any]cCs||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)rrBr(r rrrrBzsz Frame.write)rr)rr )rr#) r&r'r(rr%r)r r*rr)rBrCr(rr r*rr#)__name__ __module__ __qualname____annotations__rrrpropertyrr!r$ classmethodrArBrrrrrs"      Srrrrtuple[int, BytesLike]cCs8t|tr tjj|fSt|trtjj|fStd)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-like) isinstancer r r6TEXTencoder BINARY TypeErrorrrrr prepare_datas   rSbytescCs,t|tr |St|trt|Std)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. rL)rMr rOr rTrQrRrrr prepare_ctrls  rU)Closetuple[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. )rVparsecodereason)rcloserrr parse_closes r\rYintrZr cCst||S)z3 Serialize the payload for a close frame. )rVrD)rYrZrrrserialize_closesr^)rrrrK)rrrrT)rrTrrW)rYr]rZr rrT) __future__rr4collections.abcrrtypingrrrr r exceptionsr r r rspeedupsr ImportErrorutilsrrSrU encode_datarVr\r^rrrrs(