B ]b@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZmZyddlmZWn ek rddlmZYnXGdd d eZdd lmZmZmZd d d ddZddd dddZdS)) annotationsN)Any AwaitableCallable NamedTupleOptionalSequenceTuple) extensionsframes) PayloadTooBig ProtocolError) 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|jS)N)r rrrrrrr)selfr8lib/python3.7/site-packages/websockets/legacy/framing.py new_frameszFrame.new_framestrcCs t|jS)N)rr)rrrr__str__%sz Frame.__str__NonecCs |jS)N)rcheck)rrrrr!(sz Frame.checkN)max_sizer z!Callable[[int], Awaitable[bytes]]z Optional[int]z(Optional[Sequence[extensions.Extension]])readermaskr"r rc s|dIdH}td|\}}|d@r*dnd}|d@r:dnd} |d@rJdnd} |d @rZdnd} yt|d @} Wn,tk r} ztd | Wdd} ~ XYnX|d@rdnd|krtd |d @}|dkr|dIdH}td|\}n&|d kr|dIdH}td|\}|dk r8||kr8td|d|d|rL|dIdH}||IdH}|rjt||}t| ||| | | }|dkrg}x t |D]}|j ||d}qW| ||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")structZunpackr ZOpcode ValueErrorrr rrreverseddecoder!rrrrrr)clsr#r$r"r rZhead1Zhead2rrrrrexclengthZ mask_bitsr extensionrrrread+sN   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)rr7r$r rrrr7zsz Frame.write)__name__ __module__ __qualname____annotations__rrrpropertyrrr! classmethodr6r7rrrrrs     Mr)Close prepare_ctrl prepare_datarzTuple[int, str])rrcCstt|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. ) dataclassesZastupler?parse)rrrr parse_closes rDintr)codereasonrcCst||S)z3 Serialize the payload for a close frame. )r?r8)rFrGrrrserialize_closesrH)Z __future__rrBr.typingrrrrrrr r r exceptionsr rZspeedupsr ImportErrorZutilsrr?r@Z encode_datarArDrHrrrrs $