tniddlmZddlZddlmZmZddlmZmZm Z ddl m Z m Z ddl mZmZddlmZmZ dd lmZn#e$r dd lmZYnwxYwGd d e ZddZddZeZddl mZddZddZdS)) annotationsN) AwaitableSequence)AnyCallable NamedTuple) extensionsframes) PayloadTooBig ProtocolError) BytesLikeDataLike) apply_maskceZdZUded<ded<ded<dZded<dZded <dZded <ed d Zd!dZ d"dZ e dddd#dZ ddd$dZ dS)%Frameboolfinz frames.OpcodeopcoderdataFrsv1rsv2rsv3return frames.Framecptj|j|j|j|j|j|jSN)r rrrrrrrselfs V/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/websockets/legacy/framing.py new_framezFrame.new_frames4| K I H I I I    strc*t|jSr)r#r!rs r __str__z Frame.__str__%s4>"""r"Nonec4|jSr)r!checkrs r r(z Frame.check(s~##%%%r"N)max_sizer reader!Callable[[int], Awaitable[bytes]]maskr) int | Noner %Sequence[extensions.Extension] | NonecK|dd{V}tjd|\}}|dzrdnd}|dzrdnd} |dzrdnd} |d zrdnd} tj|d z} n"#t$r} t d | d} ~ wwxYw|dzrdnd|krt d |d z}|dkr)|dd{V}tjd|\}n.|d kr(|dd{V}tjd|\}|||krt |||r|dd{V}||d{V}|rt||}tj| ||| | | }|g}t|D]}| ||}| ||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!Q)r))structunpackr Opcode ValueErrorr r rrreverseddecoder(rrrrrr)clsr*r,r)r rhead1head2rrrrrexclength mask_bitsr! extensions r readz Frame.read+s6VAYY}UD11 uj(3ddez)4ttuz)4ttuz)4ttu ;]5:#566FF ; ; ; 011s : ;J& 1DDEd : : 344 4# S==??????D dD11IVV s]]??????D dD11IV  FX$5$511 1  ($fQiiIVF^^######  /dI..DLsD$EE  J!*-- G GI!((X(FFIIs M   N N N N    sA)) B3BB)r writeCallable[[bytes], Any]cP||j||dS)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)rrGr,r s r rGz Frame.writezs/0 dn&&DZ&HHIIIIIr")rr)rr#)rr&) r*r+r,rr)r-r r.rr)rGrHr,rr r.rr&)__name__ __module__ __qualname____annotations__rrrpropertyr!r%r( classmethodrFrGr"r rrs IIIOOODDD    X ####&&&& $<@ L L L L L [L f=A JJJJJJJJr"rrrrtuple[int, BytesLike]ct|tr%tjj|fSt|t rtjj|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 r;TEXTencoderBINARY TypeErrorrs r prepare_datar[s_ $:}!4;;==00 D) $ $:}#T))8999r"bytesct|tr|St|trt |St d)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. rT)rUr#rWrr\rYrZs r prepare_ctrlr^sQ$:{{}} D) $ $:T{{8999r")Closetuple[int, str]cFtj|}|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. )r_parsecodereason)rcloses r parse_closerfs" K  E :u| ##r"rcintrdr#cFt||S)z3 Serialize the payload for a close frame. )r_rJ)rcrds r serialize_closeris v   ( ( * **r")rrrrR)rrrr\)rr\rr`)rcrgrdr#rr\) __future__rr9collections.abcrrtypingrrrr r exceptionsr r rrspeedupsr ImportErrorutilsrr[r^ encode_datar_rfrirQr"r rss"""""" ////////,,,,,,,,,,!!!!!!!!55555555((((((((#%%%%%%%###""""""""#@J@J@J@J@JJ@J@J@JF::::0::::0  $ $ $ $ ++++++s= A  A