RPfddlmZddlZddlmZmZmZmZmZm Z m Z ddl m Z m Z ddlmZmZ ddlmZGdd eZdd l mZmZmZd d Zdd Zy#e$r ddlmZY+wxYw)) annotationsN)Any AwaitableCallable NamedTupleOptionalSequenceTuple) extensionsframes) PayloadTooBig ProtocolError) apply_maskceZdZUded<ded<ded<dZded<dZded <dZded <edd Zdd Z dd Z e ddd ddZ dd ddZ y)Frameboolfinz frames.OpcodeopcodebytesdataFrsv1rsv2rsv3ctj|j|j|j|j |j |jSN)r rrrrrrrselfs S/var/www/fastapitest/venv/lib/python3.12/site-packages/websockets/legacy/framing.py new_framezFrame.new_frames=|| KK II HH II II II   c,t|jSr)strr rs r__str__z Frame.__str__#s4>>""r!c6|jjSr)r checkrs rr&z Frame.check&s~~##%%r!N)max_sizer cK|dd{}tjd|\}}|dzrdnd}|dzrdnd} |dzrdnd} |d zrdnd} tj|d z} |dzrdnd|k7r t d |d z}|dk(r)|dd{}tjd|\}n-|d k(r(|dd{}tjd|\}|||kDrt d|d|d|r|dd{}||d{}|r t|}tj| ||| | | }|g}t|D]}|j||}|j||j|j|j|j|j |j"S7#t$r} t d | d} ~ wwxYw7N7#77ԭw)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 Opcode ValueErrorrrrrreverseddecoder&rrrrrr)clsreadermaskr'r rhead1head2rrrrrexclength mask_bitsr extensions rreadz Frame.read)s 6AY}}UD1 uj(dez)tuz)tuz)tu ;]]5:#56FJ&DEd : 34 4# S=?D dD1IV s]?D dD1IV  FX$5"3F83xj PQ Q $QiIF^# dI.DLLsD$E  J!*- GI!((X(FI G  MM    NN NN NN NN   W ; 01s : ;## ($sx G-GAG-G),G-G#-G-G&AG-G)G-G+B-G- G  GG  G-&G-)G-+G-)r cJ||jj||y)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)rwriter:r s rrDz Frame.writexs 0 dnn&&DZ&HIr!)returnz frames.Frame)rEr#)rENone) r9z!Callable[[int], Awaitable[bytes]]r:rr'z Optional[int]r (Optional[Sequence[extensions.Extension]]rEr)rDzCallable[[bytes], Any]r:rr rGrErF)__name__ __module__ __qualname____annotations__rrrpropertyr r$r& classmethodrArDr!rrrs I  KD$D$D$   #& #'?C L 1L  L  L = L  L L f@D J%J J = J  Jr!r)Close prepare_ctrl prepare_datac\tj|}|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. )rOparsecodereason)rcloses r parse_closerWs% KK E ::u|| ##r!c6t||jS)z3 Serialize the payload for a close frame. )rOrC)rTrUs rserialize_closerYs v  ( ( **r!)rrrEzTuple[int, str])rTintrUr#rEr) __future__rr2typingrrrrrr r r r exceptionsrrspeedupsr ImportErrorutilsrrOrP encode_datarQrWrYrNr!rrcs\" RRR!5#% @JJ@JH $ +#"#sA A A