U
|
[ ±dË+ ã @ sh d Z ddlZddlmZmZmZ ddlmZmZm Z m
|
Z
|
m Z G dd dZ G dd d e Z e je iZdS )
|
z>
|
wsproto/extensions
|
~~~~~~~~~~~~~~~~~~
|
|
WebSocket extensions.
|
é N)ÚOptionalÚTupleÚUnioné )Ú CloseReasonÚ FrameDecoderÚ FrameProtocolÚOpcodeÚRsvBitsc @ sú e Zd ZU eed< edddZeeef dddZee eeef ddd Z
|
ed
|
dd d Z ee e f eeeeeef d ddZee e f eeeef dddZee e f eeeed
|
f dddZee e f eeeeeeef dddZd
|
S )Ú ExtensionÚname©Úreturnc C s dS ©NF© ©Úselfr r úId:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\wsproto/extensions.pyÚenabled s zExtension.enabledc C s d S ©Nr r r r r Úoffer s zExtension.offer©r r c C s d S r r ©r r r r r Úaccept s zExtension.acceptNc C s d S r r r r r r Úfinalize s zExtension.finalize©ÚprotoÚopcodeÚrsvÚpayload_lengthr c C s t dddS r )r
|
)r r r r r r r r Úframe_inbound_header s z Extension.frame_inbound_header©r Údatar c C s |S r r ©r r r" r r r Úframe_inbound_payload_data& s z$Extension.frame_inbound_payload_data©r Úfinr c C s d S r r )r r r& r r r Úframe_inbound_complete+ s z Extension.frame_inbound_complete©r r r r" r& r c C s ||fS r r )r r r r r" r& r r r Úframe_outbound0 s zExtension.frame_outbound)Ú__name__Ú
|
__module__Ú __qualname__ÚstrÚ__annotations__Úboolr r r r r r r r r r
|
Úintr r Úbytesr$ r' r r) r r r r r s4
|
|
|
ú
|
|
|
þ
|
þ
|
|
ùr c @ s¨ e Zd ZdZdZdZd-eee eee ddddZ e
|
edd d
|
Z e j edd d d
|
Z e
|
edd dZ e j edd ddZ eedddZedddZeeef dddZeddddZeeee ee f ddd Zeeedef dd d Zeeef eeeeeef d d d!Z eeef e ee ef d"d#d$Z eeef eee edf d%d&d'Z eeef eee eeee f d(d)d*Z edd+d,Z!dS ).ÚPerMessageDeflatezpermessage-deflateé FN)Úclient_no_context_takeoverÚclient_max_window_bitsÚserver_no_context_takeoverÚserver_max_window_bitsr c C sZ || _ || _| j| _| j| _|d k r*|| _|d k r8|| _d | _d | _ d | _
|
d | _ d| _ d S r ) r4 r6 Ú DEFAULT_CLIENT_MAX_WINDOW_BITSÚ_client_max_window_bitsÚ DEFAULT_SERVER_MAX_WINDOW_BITSÚ_server_max_window_bitsr5 r7 Ú _compressorÚ _decompressorÚ_inbound_is_compressibleÚ_inbound_compressedÚ_enabled)r r4 r5 r6 r7 r r r Ú__init__A s zPerMessageDeflate.__init__r c C s | j S r )r9 r r r r r5 \ s z(PerMessageDeflate.client_max_window_bits)Úvaluer c C s" |dk s|dkrt d|| _d S ©Né r3 z.Window size must be between 9 and 15 inclusive)Ú
|
ValueErrorr9 ©r rB r r r r5 ` s c C s | j S r )r; r r r r r7 f s z(PerMessageDeflate.server_max_window_bitsc C s" |dk s|dkrt d|| _d S rC )rE r; rF r r r r7 j s )r r c C s |t jt jt jfkS r )r ZTEXTÚBINARYÚ CONTINUATION)r r r r r Ú_compressible_opcodep s z&PerMessageDeflate._compressible_opcodec C s | j S r )r@ r r r r r s s zPerMessageDeflate.enabledc C s> d| j d| j g}| jr$| d¡ | jr4| d¡ d |¡S )Núclient_max_window_bits=%dúserver_max_window_bits=%dr4 r6 ú; )r5 r7 r4 Úappendr6 Újoin)r Ú
|
parametersr r r r v s þ
|
|
zPerMessageDeflate.offerr c C s dd | d¡D }|dd
D ]r}| d¡r6d| _q | d¡rHd| _q | d¡rnt| d d¡d ¡ | _q | d
|
¡r t| d d¡d ¡ | _q d| _d S ) Nc S s g | ] }| ¡ qS r ©Ústrip©Ú.0Úbr r r Ú
|
<listcomp> s z.PerMessageDeflate.finalize.<locals>.<listcomp>ú;r r4 Tr6 r5 ú=r7 ) ÚsplitÚ
|
startswithr4 r6 r0 rQ r5 r7 r@ )r r ÚbitsÚbitr r r r s
|
|
|
|
zPerMessageDeflate.finalize)Úparamsr c C sÀ d }d }dd | d¡D }|dd
D ]}| d¡r>d| _q(| d¡rPd| _q(| d¡rd |kr|t| d d¡d ¡ }q¶| j}q(| d
|
¡r(d |kr°t| d d¡d ¡ }q(| j}q(||fS ) Nc S s g | ] }| ¡ qS r rP rR r r r rU s z3PerMessageDeflate._parse_params.<locals>.<listcomp>rV r r4 Tr6 r5 rW r7 )rX rY r4 r6 r0 rQ r5 r7 )r r\ r5 r7 rZ r[ r r r Ú _parse_params s"
|
|
|
|
z PerMessageDeflate._parse_paramsc C s | |¡\}}g }| jr"| d¡ | jr2| d¡ z<|d k rP| d| ¡ || _|d k rl| d| ¡ || _W n tk
|
r Y d S X d| _d |¡S d S )Nr4 r6 rJ rK TrL ) r] r4 rM r6 r5 r7 rE r@ rN )r r r5 r7 rO r r r r ¨ s"
|
|
|
zPerMessageDeflate.acceptr c C s |j r| ¡ rtjS |j r*|tjkr*tjS | |¡| _| jd kr|j | _| jr| jsXt |j
|
rf| j }n| j }| j d krt t| ¡| _ tdddS )NTF)Zrsv1Z iscontrolr ZPROTOCOL_ERRORr rH rI r> r? ÚAssertionErrorÚclientr7 r5 r= ÚzlibÚ decompressobjr0 r
|
)r r r r r rZ r r r r ¾ s
|
|
|
z&PerMessageDeflate.frame_inbound_headerr! c C sR | j r | js|S | jd k s tz| j t|¡W S tjk
|
rL tj Y S X d S r )
|
r? r> r= r^ Ú
|
decompressr1 r` Úerrorr ÚINVALID_FRAME_PAYLOAD_DATAr# r r r r$ Ù s z,PerMessageDeflate.frame_inbound_payload_datar% c C s |sd S | j sd | _d S | js(d | _d S | jd k s6tz | j d¡}|| j ¡ 7 }W n tjk
|
rp tj Y S X |j
|
r| j }n| j }|rd | _d | _|S )Ns ÿÿ) r> r? r= r^ rb Úflushr` rc r rd r_ r6 r4 )r r r& r" Úno_context_takeoverr r r r' å s* z(PerMessageDeflate.frame_inbound_completer( c C sÎ | |¡s||fS |tjk r0td|dd
}| jd krv|tjk sHt|jrV| j}n| j}t
|
t j t j t | ¡| _| j t|¡}|rÆ|| j t j¡7 }|d d
}|jr¶| j}n| j}|rÆd | _||fS )NTr éüÿÿÿ)T)rI r rH r
|
r< r^ r_ r5 r7 r` Ú compressobjÚZ_DEFAULT_COMPRESSIONÚDEFLATEDr0 Úcompressr1 re Ú Z_SYNC_FLUSHr4 r6 )r r r r r" r& rZ rf r r r r) s0
|
|
|
ÿ z PerMessageDeflate.frame_outboundc C sR d| j g}| jr | d¡ | d| j ¡ | jr<| d¡ d | jjd |¡¡S )NrJ r4 rK r6 z<{} {}>rL ) r5 r4 rM r7 r6 ÚformatÚ __class__r* rN )r Údescrr r r Ú__repr__, s
|
|
zPerMessageDeflate.__repr__)FNFN)"r* r+ r, r r8 r: r/ r r0 rA Úpropertyr5 Úsetterr7 r rI r r r- r r r r] r r r r
|
r r r1 r$ r' r) rp r r r r r2 ; sd ûú
|
|
ú
|
|
þ
|
þ !
|
|
ù (r2 )Ú__doc__r` Útypingr r r Zframe_protocolr r r r r
|
r r2 r ÚSUPPORTED_EXTENSIONSr r r r Ú<module> s -
|
