A>qjdrddlmZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZddlmZmZmZddlmZmZmZd d lmZmZmZgd Zd Zed dDcgc] }e| c}ZGddeZ ddZ ddZ!GddeZ" ddZ#GddeZ$ ddZ%ycc}w)) annotationsN)Sequence)AnyLiteral)DuplicateParameterInvalidParameterNameInvalidParameterValueNegotiationError PayloadTooBig ProtocolError)CONT CTRL_OPCODESFrame) BytesLike 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)FTrwbitsr#F) remote_no_context_takeoverlocal_no_context_takeoverremote_max_window_bitslocal_max_window_bitscompress_settingszlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr&r'r(r)r*s g/opt/rentech/trading_bot/.venv/lib/python3.12/site-packages/websockets/extensions/permessage_deflate.py__init__zPerMessageDeflate.__init__)s  $ " )]:::(M999*0b00000)/R/////////*D')B&&<#%:"!2..--T5P5P4PQDL--++111((DL!&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)r1s r2__repr__zPerMessageDeflate.__repr__SsV**.*I*I)JK))-)G)G(HI&&*&A&A%BC%%)%?%?$@  C r4)max_sizec|jtvr|S|jtur9|js|S|jr t d|j r[d|_nS|js|S|j sd|_|jr&tj|j |_ |j r5t|jdkrt|jtz}n |j}|dn|} |jj!||}|jj"r t%d||j rDt|jdk\r,|jj!tdr t%d||j r|jr|` t)|j||j d|j*|j,S#tj&$r}t d |d}~wwxYw) z, Decode an incoming frame. z"RSV1 bit set in continuation frameFTr$iNrrzdecompression failed)opcoderrr0rsv1r finr&r+r,r(r-lendatabytes_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr errorrrsv2rsv3)r1framer9r? max_lengthexcs r2decodezPerMessageDeflate.decode\s <<< 'L <<4 (( zz#$HIIyy(-%:: 99(,%..#119T9T8TU 99UZZ4/$'@@D::D"*Q  A<<**4 memoryviewrrErF)r1rGr?s r2encodezPerMessageDeflate.encodes% <<< 'L <zIClientPerMessageDeflateFactory.process_response_params..}H5uzzTYY&H#&zreceived duplicate Frhz#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 r]rjr_r`rarbrr*r1rdaccepted_extensionsr_r`rarbs` r2process_response_paramsz6ClientPerMessageDeflateFactory.process_response_paramsrs@ H4GH H"%8 #DE E % 8  & & " "  * *-&'LMM  * *--1*  & & . &-&'HII'$*E*EE&'KLL  & & .%1&'JKK  ( (D 0 &-)-)D)D&'$*E*EE&'KLL & & " (b " (b  " "   r4)FFNTN) r_rSr`rSrarXrbint | Literal[True] | Noner*dict[str, Any] | NonerUrV)rUSequence[ExtensionParameter])rdrrSequence[Extension]rUr) rYrZr[r\rr]r3rrrr%r4r2rr,s. - .D,1+0-1=A37 3$(3%)3!+ 3 !; 3 1 3 3@  g ,g 1g   g r4rch|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)rwextension_factorys r2ryz3enable_client_permessage_deflate..s*  "@"E"EE')memLevel)r*)r}listr extensionss r2rrsQ !+ *% *#-q/ )  r4cdeZdZdZedZ 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://datatracker.ietf.org/doc/html/rfc7692#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``. r Nc(|"d|cxkrdkstdtd|"d|cxkrdkstdtd|d|vr td| |r td||_||_||_||_||_||_y) rmNrr"rnror#zLcompress_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)rpr_r`rarbr*require_client_max_window_bits)r1r_r`rarbr*rs r2r3z'ServerPerMessageDeflateFactory.__init__s'.!7M2SQS2SNO O3TNO O&.!7M2SQS2SNO O3TNO O  (W8I-I5  " ).L>  +E'*D'&<#&<#!2.L+r4cPtfd|Drtdjt|d\}}}}jr|sd}j r|sd}j n*| j }n|j kDr j }j|durRj}nE|jr7td|dur j}nj|kr j}t||||t|||xsd|xsdjfS)zi Process request parameters. Return response params and an extension instance. c3PK|]}|jjk(ywrRrurvs r2ryzHServerPerMessageDeflateFactory.process_request_params..?rzr{zskipped duplicate Tr|zrequired client_max_window_bitsr") r}r r]rjr_r`rarbrrerr*r~s` r2process_request_paramsz5ServerPerMessageDeflateFactory.process_request_params4sf H4GH H"%7 {#CD D $ 7  & & " ""  * *--1*  * *--1*  & & . &-)-)D)D&'$*E*EE)-)D)D&  & & .%-)-)D)D&&-66*+LMM'4/)-)D)D&,,/EE)-)D)D& **&&   **&,"&,"&&    r4)FFNNNF)r_rSr`rSrarXrbrXr*rrrSrUrV)rdrrrrUz2tuple[list[ExtensionParameter], PerMessageDeflate])rYrZr[r\rr]r3rr%r4r2rrs4 - .D,1+0-1-137/4!M$(!M%)!M!+ !M !+ !M 1 !M)-!M !MFo ,o 1o  < o r4rcl|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)rw ext_factorys r2ryz3enable_server_permessage_deflate..s*  :???r rr)rarbr*)r}rrrs r2rrsW % *% *')')#-q/ )   r4) r_rSr`rSrarXrbrrUzlist[ExtensionParameter])rdrrhrSrUz9tuple[bool, bool, int | None, int | Literal[True] | None])rz'Sequence[ClientExtensionFactory] | NonerUz Sequence[ClientExtensionFactory])rz'Sequence[ServerExtensionFactory] | NonerUz Sequence[ServerExtensionFactory])& __future__rr+collections.abcrtypingrr exceptionsrr r r r r framesrrrrrrbaserrr__all__rArangerWrgrrerjrrrr)bitss0r2rs " $/.AAKK 016q">3t9>s s l $ $'7   0: (:8<:>:zm %;m `7%0o %;o d7%Q?sB4