o Tj: @s(UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z zddl Z Wne yHzddl Z Wn e yEdZ YnwYnwejrgejdkrZddlmZmZnddlmZmZded <n&zejdkrvddlmZmZnddlmZmZd ZWn e yd ZYnwd ZGd ddejZGdddZGdddZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ Gdd d Z!Gd!d"d"Z"Gd#d$d$Z#Gd%d&d&Z$eeeeed'Z%d(ed)<e dure%&d*ese%&d+dSdS),zq Handlers for Content-Encoding. See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding ) annotationsN) DecodingError))ZstdDecompressor ZstdErrorbool_zstandard_installedTFic@s.eZdZed ddZddd Zd d d Zd S) DecompressorreturnbytescCdSNselfrrS/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/httpx2/_decoders.pyunconsumed_tail8szDecompressor.unconsumed_taildata max_lengthintcCrrr)rrrrrr decompress;zDecompressor.decompresscCrrrrrrrflush=rzDecompressor.flushNr r )rr rrr r )__name__ __module__ __qualname__propertyrrrrrrrr 7s   r c@s.eZdZdZdddZdd d Zdd dZdS)ZlibDecompressorz Drain a `zlib`/`gzip` decompressor in bounded pieces so a small compressed input cannot inflate to an unbounded buffer in a single call. decompressorr r NonecCs ||_dSr)r!)rr!rrr__init__F zZlibDecompressor.__init__rr typing.Iterator[bytes]ccs8|j|t}|r|V|j|jjt}|s dSdSr)r!rMAX_DECODE_CHUNK_SIZEr)rr decompressedrrrrIs  zZlibDecompressor.decompresscCs |jSr)r!rrrrrrOr$zZlibDecompressor.flushN)r!r r r"rr r r%r)rrr__doc__r#rrrrrrr @s   r c@s eZdZd ddZd ddZd S) ContentDecoderrr r r%cCtrNotImplementedErrorrrrrrdecodeTzContentDecoder.decodecCr+rr,rrrrrWr0zContentDecoder.flushNr(r r%)rrrr/rrrrrr*Ss r*c@s$eZdZdZd ddZd dd Zd S) IdentityDecoderz Handle unencoded data. rr r r%ccs |VdSrrr.rrrr/`s zIdentityDecoder.decodeccsdEdHdSNrrrrrrrcszIdentityDecoder.flushNr(r1)rrrr)r/rrrrrr2[s r2c@.eZdZdZdddZdd d Zdd d Zd S)DeflateDecoderzZ Handle 'deflate' decoding. See: https://stackoverflow.com/questions/1838699 r r"cCsd|_tt|_dS)NT) first_attemptr zlib decompressobjr!rrrrr#nszDeflateDecoder.__init__rr r%c cs|j}d|_z |j|EdHWdStjyC}z#|r1tttj |_||EdHnt t ||WYd}~dSd}~wwNF) r6r!rr7errorr r8 MAX_WBITSr/rstr)rrwas_first_attemptexcrrrr/rszDeflateDecoder.decodec c>z |jVWdStjy}ztt||d}~wwrr!rr7r:rr<rr>rrrr~zDeflateDecoder.flushNr r"r(r1rrrr)r#r/rrrrrr5gs    r5c@r4) GZipDecoderzW Handle 'gzip' decoding. See: https://stackoverflow.com/questions/1838699 r r"cCstttjdB|_dS)N)r r7r8r;r!rrrrr#szGZipDecoder.__init__rr r%c csDz |j|EdHWdStjy!}ztt||d}~wwr)r!rr7r:rr<)rrr>rrrr/szGZipDecoder.decodec cr?rr@rArrrrrBzGZipDecoder.flushNrCr(r1rDrrrrrEs   rEc@r4) BrotliDecoderaT Handle 'brotli' decoding. Requires `pip install brotlipy`. See: https://brotlipy.readthedocs.io/ or `pip install brotli`. See https://github.com/google/brotli Supports both 'brotlipy' and 'Brotli' packages since they share an import name. The top branches are for 'brotlipy' and bottom branches for 'Brotli' r r"cCsNtdur tddt|_d|_|t|jdr |jj|_dS|jj|_dS)NzUsing 'BrotliDecoder', but neither of the 'brotlicffi' or 'brotli' packages have been installed. Make sure to install httpx using `pip install httpx[brotli]`.Fr) brotli ImportErrorr r! seen_datahasattrr _decompressprocessrrrrr#s  zBrotliDecoder.__init__rr r%c csx|sdSd|_ztd}|j||d}|r%|V|jd|d}|sWdSWdStjy;}ztt||d}~ww)NT)output_buffer_limit)rJr&rLrHr:rr<)rrrOr'r>rrrr/szBrotliDecoder.decodec cs^|jsdSz t|jdr|jWntjy'}ztt||d}~wwdEdHdS)Nfinishr)rJrKr!rQrHr:rr<rArrrrs  zBrotliDecoder.flushNrCr(r1rDrrrrrGs  rGc@r4)ZStandardDecoderznHandle 'zstd' RFC 8878 decoding. On Python 3.13 and below, this requires `pip install httpx2[zstd]`. r r"cCs tstddt|_d|_dS)NzXUsing 'ZStandardDecoder', ...Make sure to install httpx using `pip install httpx[zstd]`.F)r rIrr!rJrrrrr#s zZStandardDecoder.__init__rr r%c cs|sdSd|_zA|jjr|jj|}t|_ |j|t}|r6|V|jjs,|jjr-n |jdt}|s!|jjr>|jjsAWdS|jj}t|_qty\}zt t ||d}~ww)NTrP) rJr!eof unused_datarrr& needs_inputrrr<)rrr'r>rrrr/s2  zZStandardDecoder.decodeccs*|jsdS|jjstddEdHdS)NzZstandard data is incompleter)rJr!rSrrrrrrs zZStandardDecoder.flushNrCr(r1rDrrrrrRs   rRc@sJeZdZUdZdZded<dd d ZdddZdddZe dddZ dS) MultiDecoderzE Handle the case where multiple encodings have been applied. ztyping.ClassVar[int]max_decode_links encodingstyping.Sequence[str]r r"cCsFdd|D}t||jkrtd|jdddt|D|_dS)zk 'encodings' should be the content codings in the order in which each was applied. cSsg|]}|tvr|qSrSUPPORTED_DECODERS).0encodingrrr sz)MultiDecoder.__init__..zCannot apply more than z content encodings.cSsg|]}t|qSrr[)r]codingrrrr_sN)lenrXrreversedchildren)rrYcodingsrrrr# szMultiDecoder.__init__rr r%ccs4t|f}|jD] }||j|}q |EdHdSr)iterrc_piper/)rrstreamschildrrrr/s   zMultiDecoder.decodeccs>td}|jD]}t||j||}q|EdHdSr3)rerc itertoolschainrfr/r)rrgrhrrrrs  zMultiDecoder.flushr/0typing.Callable[[bytes], typing.Iterator[bytes]]upstreamccs|D] }||EdHqdSrr)r/rlchunkrrrrf"szMultiDecoder._pipeN)rYrZr r"r(r1)r/rkrlr%r r%) rrrr)rX__annotations__r#r/r staticmethodrfrrrrrVs    rVc@0eZdZdZddddZdd d ZdddZdS) ByteChunkerz> Handles returning byte content in fixed-size chunks. N chunk_size int | Noner r"cCt|_||_dSr)ioBytesIO_buffer _chunk_sizerrrrrrr#0  zByteChunker.__init__contentr list[bytes]cjdur |r |gSgSj|jjkr_jfddtdtjD}t|djkrFjdj|Sjdj|dj|ddSgS)Ncg|] }||jqSrrxr]irvaluerrr_;z&ByteChunker.decode..r rxrwwritetellgetvaluerangeraseektruncaterr{chunksrrrr/4   "     zByteChunker.decodecC.|j}|jd|j|r|gSgSNrrwrrrrrrrrH   zByteChunker.flushrrrrsr r")r{r r r|)r r|rDrrrrrq+   rqc@rp) TextChunkerz> Handles returning text content in fixed-size chunks. Nrrrsr r"cCrtr)ruStringIOrwrxryrrrr#TrzzTextChunker.__init__r{r< list[str]cr})Ncr~rrrrrrr__rz&TextChunker.decode..rrrrrrrr/XrzTextChunker.decodecCrrrrrrrrlrzTextChunker.flushrr)r{r<r rr rrDrrrrrOrrc@s0eZdZdZddddZdd d Zdd dZdS) TextDecoderz8 Handles incrementally decoding bytes into text utf-8r^r<r r"cCst|dd|_dS)Nreplace)errors)codecsgetincrementaldecoderdecoder)rr^rrrr#xszTextDecoder.__init__rr cCs |j|Srrr/r.rrrr/{s zTextDecoder.decodecCs|jddS)NrPTrrrrrr~szTextDecoder.flushN)r)r^r<r r")rr r r<)r r<rDrrrrrss   rc@r4) LineDecoderz Handles incrementally reading lines from text. Has the same behaviour as the stdllib splitlines, but handling the input iteratively. r r"cCsg|_d|_dSr9)buffer trailing_crrrrrr#s zLineDecoder.__init__textr<rcCsd}|jr d|}d|_|drd|_|dd}|sgS|d|v}|}t|dkr:|s:|j|dgS|jrQd|j|dg|dd}g|_|sY|g|_|S) Nu …

 FTrrr)rendswith splitlinesrarappendjoinpop)rr NEWLINE_CHARStrailing_newlinelinesrrrr/s(   " zLineDecoder.decodecCs.|js|jsgSd|jg}g|_d|_|S)NrF)rrr)rrrrrrs zLineDecoder.flushNrC)rr<r rrrDrrrrrs   &r)identitygzipdeflatebrzstdzdict[str, type[ContentDecoder]]r\rr)'r) __future__rrrurisystypingr7 _exceptionsrrHrI brotlicffi TYPE_CHECKING version_infocompression.zstdrrbackports.zstdrnr r&Protocolr r r*r2r5rErGrRrVrqrrrr\rrrrrsr            :.'$$=