o KK&h@sddlmZddlZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZmZddl mZdd lmZzdd lmZWneyOdd lmZYnwGd d d e Zd ddZd!ddZeZddl mZd"ddZd#ddZdS)$) annotationsN) AwaitableSequence)AnyCallable NamedTuple) extensionsframes) PayloadTooBig ProtocolError) BytesLike)Data) 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.OpcodeopcodebytesdataFrsv1rsv2rsv3return frames.FramecCs t|j|j|j|j|j|jSN)r rrrrrrrselfrL/var/www/html/venv/lib/python3.10/site-packages/websockets/legacy/framing.py new_frameszFrame.new_framestrcCs t|jSr)r!r rrrr__str__& z Frame.__str__NonecCs |jSr)r checkrrrrr%)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)rrCr)r rrrrC{sz Frame.write)rr)rr!)rr$) r'r(r)rr&r*r r+rr)rCrDr)rr r+rr$)__name__ __module__ __qualname____annotations__rrrpropertyr r"r% classmethodrBrCrrrrrs"      Srrrrtuple[int, bytes]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 r7TEXTencoder BINARY TypeErrorrrrr prepare_datas   rTrcCs,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. rM)rNr!rPr rrRrSrrr 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. )rVrE)rYrZrrrserialize_closesr^)rrrrL)rrrr)rrrrW)rYr]rZr!rr) __future__rr5collections.abcrrtypingrrrr r exceptionsr r r rspeedupsr ImportErrorutilsrrTrU encode_datarVr\r^rrrrs*