RPf`ZddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZmZddlmZmZmZgd Zd Zed d Dcgc] }e| c}ZGd deZ ddZ ddZGddeZ ddZGddeZ ddZ ycc}w)) annotationsN)AnyDictListOptionalSequenceTupleUnion) exceptionsframes) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescjeZdZdZedZ d d dZd dZdd d dZd dZ y)rz) Per-Message Deflate extension. permessage-deflateNc|i}|dvsJ|dvsJd|cxkrdksJJd|cxkrdksJJd|vsJ||_||_||_||_||_|js&t j |j |_|js1t jdd|j i|j|_ d|_ y) z? Configure the Per-Message Deflate extension. N)FTrwbitsrF) remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingszlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr"r#r$r%r&s b/var/www/fastapitest/venv/lib/python3.12/site-packages/websockets/extensions/permessage_deflate.py__init__zPerMessageDeflate.__init__!s  $ " )]:::(M999*0b00000)/R/////////*D')B&&<#%:"!2..--T5P5P4PQDL--++111595K5KDL !&c pd|jd|jd|jd|jd S)Nz-PerMessageDeflate(remote_no_context_takeover=z, local_no_context_takeover=z, remote_max_window_bits=z, local_max_window_bits=))r"r#r$r%r-s r.__repr__zPerMessageDeflate.__repr__JsV**.*I*I)JK))-)G)G(HI&&*&A&A%BC%%)%?%?$@  C r0)max_sizec8|jtjvr|S|jtjur"|js|S|j rrd|_nj|j s|Stj|d}|j sd|_|jr&tj|j |_ |j}|j r |tz }|dn|} |jj!||}|jj(rt%j*d|d |j r|jr|` tj|| S#tj"$r}t%j&d|d}~wwxYw) z, Decode an incoming frame. Frsv1Tr Nrzdecompression failedzover size limit (? > z bytes)data)opcoder CTRL_OPCODESOP_CONTr,finr8 dataclassesreplacer"r'r(r$r)r:_EMPTY_UNCOMPRESSED_BLOCK decompresserrorr ProtocolErrorunconsumed_tail PayloadTooBig)r-framer5r: max_lengthexcs r.decodezPerMessageDeflate.decodeSsc <<6.. .L <<6>> )(( yy(-%:: ''E:E99(,%..#119T9T8TU zz 99 - -D"*Q  L<<**4FFcj|jtjvr|S|jtjurTt j |d}|j r1tjdd|j i|j|_ |jj|j|jjtjz}|j r|j#t$r|dd}|j r|j r|` t j ||S)z, Encode an outgoing frame. Tr7rNr9r!)r;r r<r=r?r@r#r'r*r%r&r+compressr:flush Z_SYNC_FLUSHr>endswithrA)r-rGr:s r.encodezPerMessageDeflate.encodes <<6.. .L <zIClientPerMessageDeflateFactory.process_response_params..TH5uzzTYY&H#&zreceived duplicate Frlz#expected server_no_context_takeoverTzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr) anyr NegotiationErrorr^rnr`rarbrcrr&r-reaccepted_extensionsr`rarbrcs` r.process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsIsd H4GH H--0CDII;.OP P % 8  & & " "  * *- 112WXX  * *--1*  & & . &- 112STT'$*E*EE 112VWW  & & .%1 112UVV  ( (D 0 &-)-)D)D&'$*E*EE 112VWW & & " (b " (b  " "   r0)FFNTN) r`rSrarSrbrYrcOptional[Union[int, bool]]r&Optional[Dict[str, Any]]rUrV)rUList[ExtensionParameter])reSequence[ExtensionParameter]rSequence[Extension]rUr) rZr[r\r]rr^r/rvrr!r0r.rrs. - .D,1+004=A6: 3$(3%)3!. 3 !; 3 4 3 3@  g ,g 1g   g r0rch|g}td|Dst|tddigz}|S)z Enable Per-Message Deflate with default settings in client extensions. If the extension is already present, perhaps with non-default settings, the configuration isn't changed. c3VK|]!}|jtjk(#ywrR)r^r)r{extension_factorys r.r}z3enable_client_permessage_deflate..s*  "@"E"EE')memLevel)r&)rlistr extensionss r.rrsQ !+ *% *#-q/ )  r0cdeZdZdZedZ d ddZ ddZy) ra Server-side extension factory for the Per-Message Deflate extension. Parameters behave as described in `section 7.1 of RFC 7692`_. .. _section 7.1 of RFC 7692: https://www.rfc-editor.org/rfc/rfc7692.html#section-7.1 Set them to :obj:`True` to include them in the negotiation offer without a value or to an integer value to include them with this value. Args: server_no_context_takeover: prevent server from using context takeover. client_no_context_takeover: prevent client from using context takeover. server_max_window_bits: maximum size of the server's LZ77 sliding window in bits, between 8 and 15. client_max_window_bits: maximum size of the client's LZ77 sliding window in bits, between 8 and 15. compress_settings: additional keyword arguments for :func:`zlib.compressobj`, excluding ``wbits``. require_client_max_window_bits: do not enable compression at all if client doesn't advertise support for ``client_max_window_bits``; the default behavior is to enable compression without enforcing ``client_max_window_bits``. rNc(|"d|cxkrdkstdtd|"d|cxkrdkstdtd|d|vr td| |r td||_||_||_||_||_||_y) rqNrrrrrsrzLcompress_settings must not include wbits, set server_max_window_bits insteadzVrequire_client_max_window_bits is enabled, but client_max_window_bits isn't configured)rtr`rarbrcr&require_client_max_window_bits)r-r`rarbrcr&rs r.r/z'ServerPerMessageDeflateFactory.__init__s'.!7M2SQS2SNO O3TNO O&.!7M2SQS2SNO O3TNO O  (W8I-I5  " ).L>  +E'*D'&<#&<#!2.L+r0cxtfd|Dr"tjdjt |d\}}}}j r|sd}j r|sd}jn*| j}n|jkDr j}j|dur\j}nO|!jrAtjd|dur j}nj|kr j}t||||t|||xsd|xsdjfS)zi Process request parameters. Return response params and an extension instance. c3PK|]}|jjk(ywrRryrzs r.r}zHServerPerMessageDeflateFactory.process_request_params..r~rzskipped duplicate Trzrequired client_max_window_bitsr) rr rr^rnr`rarbrcrrfrr&rs` r.process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params sr H4GH H--0B499+.NO O $ 7  & & " ""  * *--1*  * *--1*  & & . &-)-)D)D&'$*E*EE)-)D)D&  & & .%-)-)D)D&&-66$556WXX'4/)-)D)D&,,/EE)-)D)D& **&&   **&,"&,"&&    r0)FFNNNF)r`rSrarSrbrYrcrYr&rrrSrUrV)rerrrrUz2Tuple[List[ExtensionParameter], PerMessageDeflate])rZr[r\r]rr^r/rr!r0r.rrs4 - .D,1+004046:/4!M$(!M%)!M!. !M !. !M 4 !M)-!M !MFo ,o 1o  < o r0rcl|g}td|Dst|tddddigz}|S)z Enable Per-Message Deflate with default settings in server extensions. If the extension is already present, perhaps with non-default settings, the configuration isn't changed. c3VK|]!}|jtjk(#ywrR)r^r)r{ ext_factorys r.r}z3enable_server_permessage_deflate..s*  :???r rr)rbrcr&)rrrrs r.rr}sW % *% *')')#-q/ )   r0) r`rSrarSrbrYrcrrUr)rerrlrSrUzrs" DDD!6KK 016q">3t9>R5 R5j $ $*7   0: (:8<:A:zm %;m `:%0o %;o d:%O?s B(