o [h]@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m Z mZddlmZmZddlmZmZmZgd Zd Zd d ed dDZGdddeZd/ddZd0d d!ZGd"d#d#eZd1d'd(ZGd)d*d*eZd2d-d.ZdS)3) annotationsN)AnyDictListOptionalSequenceTupleUnion) exceptionsframes) ExtensionNameExtensionParameter)ClientExtensionFactory ExtensionServerExtensionFactory)PerMessageDeflateClientPerMessageDeflateFactory enable_client_permessage_deflateServerPerMessageDeflateFactory enable_server_permessage_deflatescCsg|]}t|qS)str).0bitsrr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/websockets/extensions/permessage_deflate.py src@sJeZdZdZedZ ddddZdddZdddddZd ddZ dS)!rz) Per-Message Deflate extension. permessage-deflateNremote_no_context_takeoverboollocal_no_context_takeoverremote_max_window_bitsintlocal_max_window_bitscompress_settingsOptional[Dict[Any, Any]]returnNonecCs|duri}|dvs J|dvsJd|krdks JJd|kr+dks.JJd|vs4J||_||_||_||_||_|jsOtj|j d|_|js`tjdd|j i|j|_ d|_ dS) z? Configure the Per-Message Deflate extension. N)FTrwbitsr,Fr) r!r#r$r&r'zlib decompressobjdecoder compressobjencoderdecode_cont_data)selfr!r#r$r&r'rrr__init__!s*     zPerMessageDeflate.__init__rc Cs&d|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&r4rrr__repr__JszPerMessageDeflate.__repr__)max_sizeframe frames.Framer9 Optional[int]cCs|jtjvr|S|jtjur|js|S|jrd|_n|js|Stj|dd}|js,d|_|j r8t j |j d|_ |j}|jrB|t7}|durHdn|}|j ||}|j jr^td|d|jrf|j rf|` tj||d S) z, Decode an incoming frame. Frsv1Tr-Nrzover size limit (? > z bytes)data)opcoder CTRL_OPCODESOP_CONTr3finr> dataclassesreplacer!r.r/r$r0r@_EMPTY_UNCOMPRESSED_BLOCK decompressunconsumed_tailr Z PayloadTooBig)r4r:r9r@ max_lengthrrrdecodeSs2   zPerMessageDeflate.decodecCs|jtjvr|S|jtjur&tj|dd}|jr&tjdd|j i|j |_ |j |j |j tj}|jrB|trB|dd}|jrJ|jrJ|` tj||dS)z, Encode an outgoing frame. Tr=r,Nr?r)rAr rBrCrErFr#r.r1r&r'r2compressr@flush Z_SYNC_FLUSHrDendswithrG)r4r:r@rrrencodes    zPerMessageDeflate.encodeN) r!r"r#r"r$r%r&r%r'r(r)r*)r)r)r:r;r9r<r)r;)r:r;r)r;) __name__ __module__ __qualname____doc__r namer5r8rKrQrrrrrs ) 7rserver_no_context_takeoverr"client_no_context_takeoverserver_max_window_bitsr<client_max_window_bitsOptional[Union[int, bool]]r)List[ExtensionParameter]cCsfg}|r |d|r|d|r|dt|f|dur&|d|S|r1|dt|f|S)zS Build a list of ``(name, value)`` pairs for some compression parameters. )rXN)rYNrZT)r[Nr[)appendr)rXrYrZr[paramsrrr_build_parameterss    r`r_Sequence[ExtensionParameter] is_server|r1t||dur8d}q t|||dkrZ|durKt||tvrTt|}q t|||dkr|durgt||rp|durpd}q |tvryt|}q t||t|||||fS)z Extract compression parameters from a list of ``(name, value)`` pairs. If ``is_server`` is :obj:`True`, ``client_max_window_bits`` may be provided without a value. This is only allowed in handshake requests. FNrXTrYrZr[)r ZDuplicateParameterZInvalidParameterValue_MAX_WINDOW_BITS_VALUESr%ZInvalidParameterName)r_rbrXrYrZr[rWvaluerrr_extract_parameterssJ              rfc@sBeZdZdZedZ     ddddZdddZd ddZdS)!ra Client-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, or :obj:`True` to indicate support without setting a limit. compress_settings: additional keyword arguments for :func:`zlib.compressobj`, excluding ``wbits``. r FNTrXr"rYrZr<r[r\r'Optional[Dict[str, Any]]r)r*cCs|dusd|krdkstdtd|dus0|dus0d|kr+dks0tdtd|durtd |jkrHtd jdurW|durVtd njdur]n|durej}n |jkrotd t |||pud |pxd j S)zV Process response parameters. Return an extension instance. c3|] }|jjkVqdSrRrWrotherr7rr RzIClientPerMessageDeflateFactory.process_response_params..zreceived duplicate Frbz#expected server_no_context_takeoverTNzexpected server_max_window_bitsz"unsupported server_max_window_bitsz!unexpected client_max_window_bitsz"unsupported client_max_window_bitsr+) anyr NegotiationErrorrWrfrXrYrZr[rr'r4r_rorXrYrZr[rr7rprocess_response_paramsGsJ             z6ClientPerMessageDeflateFactory.process_response_params)FFNTN) rXr"rYr"rZr<r[r\r'rgr)r*)r)r])r_rarorpr)r) rSrTrUrVr rWr5rnr{rrrrrs  r extensions*Optional[Sequence[ClientExtensionFactory]] Sequence[ClientExtensionFactory]cCs:|durg}tdd|Dst|tddidg}|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. Ncs|] }|jtjkVqdSrR)rWr)rZextension_factoryrrrru  z3enable_client_permessage_deflate..memLevel)r')rxlistrr|rrrrs rc@s8eZdZdZedZ     ddddZdddZdS)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, or :obj:`True` to indicate support without setting a limit. compress_settings: additional keyword arguments for :func:`zlib.compressobj`, excluding ``wbits``. r FNrXr"rYrZr<r[r'rgr)r*cCs|dusd|krdkstdtd|dus,d|kr'dks,tdtd|dur8d|vr8td||_||_||_||_||_dS)rhNrr+rirjr,zLcompress_settings must not include wbits, set server_max_window_bits insteadrkrmrrrr5s"  z'ServerPerMessageDeflateFactory.__init__r_rarorp2Tuple[List[ExtensionParameter], PerMessageDeflate]cstfdd|Drtdjt|dd\}}}}jr%|s%d}jr,|s,d}jdur2n|dur:j}n|jkrBj}jdurO|durNj}n|durXtd|dur`j}nj|krhj}t ||||t |||ptd|pwdj fS) zi Process request parameters. Return response params and an extension instance. c3rqrRrrrsr7rrru rvzHServerPerMessageDeflateFactory.process_request_params..zskipped duplicate TrwNzrequired client_max_window_bitsr+) rxr ryrWrfrXrYrZr[r`rr'rzrr7rprocess_request_paramssX        z5ServerPerMessageDeflateFactory.process_request_params)FFNNN) rXr"rYr"rZr<r[r<r'rgr)r*)r_rarorpr)r)rSrTrUrVr rWr5rrrrrrs r*Optional[Sequence[ServerExtensionFactory]] Sequence[ServerExtensionFactory]cCs>|durg}tdd|Dst|tddddidg}|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. NcsrrR)rWr)rZ ext_factoryrrrru|rz3enable_server_permessage_deflate.. rr)rZr[r')rxrrrrrrrps r) rXr"rYr"rZr<r[r\r)r])r_rarbr"r)rc)r|r}r)r~)r|rr)r) __future__rrEr.typingrrrrrrr r r r rbaserrr__all__rGrangerdrr`rfrrrrrrrrs( $  > 1(