o Tje"@sUddlmZddlZddlmZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZdZejdZded<dd d ZGd ddZGdddZGdddeZdddZdddZdS)) annotationsN)NoReturn)HeadersMutableHeaders)ASGIAppMessageReceiveScopeSend) zapplication/gzipzapplication/x-gzipzapplication/zipzaudio/*z font/woffz font/woff2z image/avifz image/gifz image/jpegz image/pngz image/webpztext/event-streamzvideo/*_gzip_capacity_limiterz,anyio.lowlevel.RunVar[anyio.CapacityLimiter]returnanyio.CapacityLimitercCs6ztWStytd}t||YSw)zDReturn the capacity limiter used for worker-thread GZip compression.()r get LookupErroranyioCapacityLimitersetlimiterr\/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/starlette/middleware/gzip.py_get_gzip_capacity_limiter s    rc@s.eZdZ   deddddZdddZdS)GZipMiddleware exclude_content_typesappr minimum_sizeint compresslevelthread_minimum_sizertuple[str, ...]r NonecCs&||_||_||_||_t||_dSN)rr r"r#_normalize_content_typesrselfrr r"r#rrrr__init__-s  zGZipMiddleware.__init__scoper receiversendr cs|ddkr||||IdHdSt|d}d|ddvr/t|j|j|j|j|jd}n t|j|j|jd}||||IdHdS) Ntypehttp)r+gzipAccept-Encoding)r"r#rr) rrr GZipResponderr r"r#rIdentityResponder)r)r+r,r-headers responderrrr__call__<s  zGZipMiddleware.__call__N)rrr rrr r!r"r!r#r!rr$r r%r+r r,rr-r r r%)__name__ __module__ __qualname__DEFAULT_EXCLUDED_CONTENT_TYPESr*r7rrrrr,src@sDeZdZUded<edd!d d Zd"ddZd#ddZd$ddZd S)%r4strcontent_encodingrrrr r!rr$r r%cCs>||_||_t||_t|_i|_d|_d|_d|_ d|_ dS)NF) rr r'runattached_sendr-initial_messagestartedcontent_encoding_setcontent_type_is_excludedpartial_response)r)rr rrrrr*Ts  zIdentityResponder.__init__r+r r,rr-r cs"||_||||jIdHdSr&)r-rsend_with_compression)r)r+r,r-rrrr7eszIdentityResponder.__call__messagercs|d}|dkrD||_t|jdd}d|v|_|ddk|_|dd d d }||d d d h}||j  |_ dS|dkrj|jsQ|jsQ|j rj|j s`d|_ | |jIdH| |IdHdS|dkr|j sd|_ |dd}|dd}t ||jkr|s| |jIdH| |IdHdS|s|j|ddIdH}t|jdd}|d||dkr|j|d<tt ||d<||d<| |jIdH| |IdHdS|j|ddIdH}t|jdd}|d||dkr |j|d<|d=||d<| |jIdH| |IdHdS|dkrE|dd}|dd}|j||dIdH|d<| |IdHdS|dkr]| |jIdH| |IdHdSdS)Nr.zhttp.response.startr5)rawzcontent-encodingstatusz content-typer2;r/z/*zhttp.response.bodyTbody more_bodyF)rOr1zContent-EncodingzContent-Lengthzhttp.response.pathsend)rArrCrEr partitionstriplower isdisjointrrDrBr-lenr apply_compressionradd_vary_headerr?r>)r)rG message_typer5 media_type media_typesrMrOrrrrFisp            z'IdentityResponder.send_with_compressionrMbytesrOboolcs|S)zApply compression on the response body. If more_body is False, the compression stream is finalized. Compression resources are only allocated once a body is actually compressed. rr)rMrOrrrrUsz#IdentityResponder.apply_compressionN)rrr r!rr$r r%r9)rGrr r%rMrZrOr[r rZ) r:r;r<__annotations__r=r*r7rFrUrrrrr4Qs   Br4csPeZdZdZ ddeddfddZedddZdddZdddZ Z S) r3r0rr)r#rrrr r!r"r#rr$r r%cs(tj|||d||_||_d|_dS)Nr)superr*r"r# _compressorr( __class__rrr*s  zGZipResponder.__init__zlib._CompresscCs*|jdurt|jtjdtj|_|jS)N)r`zlib compressobjr"DEFLATED MAX_WBITS)r)rrr compressors zGZipResponder.compressorrMrZrOr[cs>t||jkrt}tjj|j|||dIdHS|||S)Nr)rTr#rr to_threadrun_sync_compress_body)r)rMrOrrrrrUs  zGZipResponder.apply_compressioncCs4|r|j||jtjS|j||jSr&)ricompressflushre Z_SYNC_FLUSHr\rrrrlszGZipResponder._compress_body)rr8)r rcr]) r:r;r<r?r=r*propertyrirUrl __classcell__rrrarr3s  r3rGrrcs td)Nzsend awaitable not set) RuntimeError)rGrrrr@sr@ content_typesr$cCstdd|DS)Ncss&|]}|ddVqdS)rKrN)rPrQrR).0 content_typerrr s$z+_normalize_content_types..)tuple)rsrrrr'sr')r r )rGrr r)rsr$r r$) __future__rretypingranyio.lowlevelranyio.to_threadstarlette.datastructuresrrstarlette.typesrrrr r r=lowlevelRunVarr r^rrr4r3r@r'rrrrs   %c %