o 6Wj / @sdZddlmZddlZddlZddlZddlZddlmZzddl Z Wne y?zddl Z Wn e y<dZ YnwYnwzddl Z Wn e yQdZ YnwGdddZ Gdd d e ZGd d d e ZGd d d e ZGddde ZGddde ZGddde ZGdddZGdddZGdddZGdddZeeeeedZe durede dureddSdS)zq Handlers for Content-Encoding. See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding ) annotationsN) DecodingErrorc@s eZdZd ddZd ddZdS) ContentDecoderdatabytesreturncCtNNotImplementedErrorselfrr^/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/httpx/_decoders.pydecode%zContentDecoder.decodecCr r r rrrrflush(rzContentDecoder.flushNrrrrrr)__name__ __module__ __qualname__rrrrrrr$s rc@s$eZdZdZd ddZd ddZd S) IdentityDecoderz Handle unencoded data. rrrcCs|Sr rr rrrr1zIdentityDecoder.decodecCsdSNrrrrrr4rzIdentityDecoder.flushNrr)rrr__doc__rrrrrrr,s rc@.eZdZdZd ddZddd Zdd d Zd S)DeflateDecoderzZ Handle 'deflate' decoding. See: https://stackoverflow.com/questions/1838699 rNonecCsd|_t|_dSNT) first_attemptzlib decompressobj decompressorrrrr__init__?szDeflateDecoder.__init__rrc Csn|j}d|_z|j|WStjy6}z|r+ttj |_||WYd}~Stt ||d}~wwNF) r#r& decompressr$errorr% MAX_WBITSrrstr)rrwas_first_attemptexcrrrrCszDeflateDecoder.decodec C6z|jWStjy}ztt||d}~wwr r&rr$r*rr,rr.rrrrN  zDeflateDecoder.flushNrr!rrrrrrr'rrrrrrr 8s    r c@r) GZipDecoderzW Handle 'gzip' decoding. See: https://stackoverflow.com/questions/1838699 rr!cCsttjdB|_dS)N)r$r%r+r&rrrrr'\zGZipDecoder.__init__rrc Cs8z|j|WStjy}ztt||d}~wwr )r&r)r$r*rr,rrr.rrrr_s zGZipDecoder.decodec Cr/r r0r1rrrrer2zGZipDecoder.flushNr3rrr4rrrrr5Us   r5c@r) 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' rr!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 ImportError Decompressorr& seen_datahasattrr) _decompressprocessrrrrr'vs  zBrotliDecoder.__init__rrc CsD|sdSd|_z||WStjy!}ztt||d}~wwNrT)r=r?r:r*rr,r8rrrrs zBrotliDecoder.decodec CsP|jsdSzt|jdr|jWdStjy'}ztt||d}~ww)Nrfinish)r=r>r&rBr:r*rr,r1rrrrs  zBrotliDecoder.flushNr3rrr4rrrrr9ls   r9c@r)ZStandardDecoderz Handle 'zstd' RFC 8878 decoding. Requires `pip install zstandard`. Can be installed as a dependency of httpx using `pip install httpx[zstd]`. rr!cCs*tdur tddt|_d|_dS)NzXUsing 'ZStandardDecoder', ...Make sure to install httpx using `pip install httpx[zstd]`.F) zstandardr;ZstdDecompressorr%r&r=rrrrr's zZStandardDecoder.__init__rrc CstdusJd|_t}zA||j||jjr@|jjrJ|jj}t |_||j||jjrE|jjsW|SW|SW|SW|Stj yb}zt t ||d}~wwr")rDr=ioBytesIOwriter&r)eof unused_datarEr% ZstdErrorrr,getvalue)rroutputrJr.rrrrs* zZStandardDecoder.decodecCs,|jsdS|j}|jjstdt|S)NrzZstandard data is incomplete)r=r&rrIrr)rretrrrrs  zZStandardDecoder.flushNr3rrr4rrrrrCs   rCc@s.eZdZdZdddZdd d Zdd d ZdS) MultiDecoderzE Handle the case where multiple encodings have been applied. childrentyping.Sequence[ContentDecoder]rr!cCstt||_dS)zm 'children' should be a sequence of decoders in the order in which each was applied. N)listreversedrP)rrPrrrr'szMultiDecoder.__init__rrcCs|jD]}||}q|Sr )rPrrrchildrrrrs  zMultiDecoder.decodecCs&d}|jD] }|||}q|Sr)rPrrrTrrrrs zMultiDecoder.flushN)rPrQrr!rrr4rrrrrOs   rOc@0eZdZdZddddZdd d ZdddZdS) ByteChunkerz> Handles returning byte content in fixed-size chunks. N chunk_size int | Nonerr!cCt|_||_dSr )rFrG_buffer _chunk_sizerrXrrrr'  zByteChunker.__init__contentr list[bytes]cjdur |r |gSgSj|jjkr_jfddtdtjD}t|djkrFjdj|Sjdj|dj|ddSgS)Ncg|] }||jqSrr\.0irvaluerr z&ByteChunker.decode..r r\r[rHtellrLrangelenseektruncaterr_chunksrrgrr"         zByteChunker.decodecC.|j}|jd|j|r|gSgSNrr[rLrprqrgrrrr   zByteChunker.flushr rXrYrr!)r_rrr`)rr`r4rrrrrW   rWc@rV) TextChunkerz> Handles returning text content in fixed-size chunks. NrXrYrr!cCrZr )rFStringIOr[r\r]rrrr'r^zTextChunker.__init__r_r, list[str]cra)Ncrbrrcrdrgrrrirjz&TextChunker.decode..rrkrlrrrrgrrrtzTextChunker.decodecCrurvrwrgrrrr+rxzTextChunker.flushr ry)r_r,rr}rr}r4rrrrr{ rzr{c@s0eZdZdZddddZdd d Zdd dZdS) TextDecoderz8 Handles incrementally decoding bytes into text utf-8encodingr,rr!cCst|dd|_dS)Nreplace)errors)codecsgetincrementaldecoderdecoder)rrrrrr'7r7zTextDecoder.__init__rrcCs |j|Sr rrr rrrr:s zTextDecoder.decodecCs|jddSrArrrrrr=szTextDecoder.flushN)r)rr,rr!)rrrr,)rr,r4rrrrr2s   rc@s.eZdZdZdddZdd d Zdd d Zd S) LineDecoderz Handles incrementally reading lines from text. Has the same behaviour as the stdllib splitlines, but handling the input iteratively. rr!cCsg|_d|_dSr()buffer trailing_crrrrrr'Is zLineDecoder.__init__textr,r}cCsd}|jr d|}d|_|drd|_|dd}|sgS|d|v}|}t|dkr:|s:|j|dgS|jrQd|j|dg|dd}g|_|sY|g|_|S) Nu …

 FTrkrr)rendswith splitlinesrorappendjoinpop)rr NEWLINE_CHARStrailing_newlinelinesrrrrMs(   " zLineDecoder.decodecCs.|js|jsgSd|jg}g|_d|_|S)NrF)rrr)rrrrrrss zLineDecoder.flushNr3)rr,rr}r~r4rrrrrAs   &r)identitygzipdeflatebrzstdrr)r __future__rrrFtypingr$ _exceptionsrr:r; brotlicffirDrrr r5r9rCrOrWr{rrSUPPORTED_DECODERSrrrrrsX         5*''=