o [h@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZmZzddlmZWneyGddlmZYnwGdd d eZdd lmZmZmZdddZdddZdS)) annotationsN)Any AwaitableCallable NamedTupleOptionalSequenceTuple) extensionsframes) PayloadTooBig ProtocolError) 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 rrrrrrrselfr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/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& Optional[int]r (Optional[Sequence[extensions.Extension]]c 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||krtd|d|d|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. r Nz!BBTF@ zinvalid opcodezincorrect masking~z!Hz!Qzover size limit (z > z bytes))r&)structunpackr ZOpcode ValueErrorrr rrreverseddecoder%rrrrrr)clsr'r)r&r rZhead1Zhead2rrrrrexclengthZ 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)rr?r)r rrrr?zsz Frame.write)rr)rr!)rr$) r'r(r)rr&r*r r+rr)r?r@r)rr r+rr$)__name__ __module__ __qualname____annotations__rrrpropertyr r"r% classmethodr>r?rrrrrs"      Sr)Close prepare_ctrl prepare_datarrrTuple[int, str]cCstt|S)z Parse the payload from a close frame. Returns: Tuple[int, str]: close code and reason. Raises: ProtocolError: if data is ill-formed. UnicodeDecodeError: if the reason isn't valid UTF-8. ) dataclassesastuplerHparse)rrrr parse_closes rOcodeintreasonr!cCst||S)z3 Serialize the payload for a close frame. )rHrA)rPrRrrrserialize_closesrS)rrrrK)rPrQrRr!rr) __future__rrLr5typingrrrrrrr r r exceptionsr rZspeedupsr ImportErrorutilsrrHrIZ encode_datarJrOrSrrrrs $