U g / @sdZddlmZddlZddlZddlZddlZddlmZz ddl Z Wn:e k rz ddl Z Wne k r|dZ YnXYnXz ddl Z Wne k rdZ YnXGdddZ Gdd d e ZGd d d e ZGd d d e ZGddde ZGddde ZGddde ZGdddZGdddZGdddZGdddZeeeeedZe dkrtede dkreddS)zq Handlers for Content-Encoding. See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding ) annotationsN) DecodingErrorc@s*eZdZdddddZddddZdS) ContentDecoderbytesdatareturncCs tdSNNotImplementedErrorselfrr3/tmp/pip-unpacked-wheel-ql4n0x43/httpx/_decoders.pydecode%szContentDecoder.decoder cCs tdSr r rrrrflush(szContentDecoder.flushN)__name__ __module__ __qualname__rrrrrrr$src@s.eZdZdZdddddZddddZd S) IdentityDecoderz Handle unencoded data. rrcCs|Sr rr rrrr1szIdentityDecoder.decodercCsdSNrrrrrr4szIdentityDecoder.flushN)rrr__doc__rrrrrrr,src@s<eZdZdZddddZddddd Zddd d Zd S) DeflateDecoderzZ Handle 'deflate' decoding. See: https://stackoverflow.com/questions/1838699 NonercCsd|_t|_dSNT) first_attemptzlib decompressobj decompressorrrrr__init__?szDeflateDecoder.__init__rrc Csx|j}d|_z|j|WStjk rr}z8|rTttj |_||WYStt ||W5d}~XYnXdSNF) rr" decompressr errorr! MAX_WBITSrrstr)rrZwas_first_attemptexcrrrrCszDeflateDecoder.decodec CsDz |jWStjk r>}ztt||W5d}~XYnXdSr r"rr r&rr(rr)rrrrNs zDeflateDecoder.flushNrrrrr#rrrrrrr8s rc@s<eZdZdZddddZddddd Zddd d Zd S) GZipDecoderzW Handle 'gzip' decoding. See: https://stackoverflow.com/questions/1838699 rrcCsttjdB|_dS)N)r r!r'r"rrrrr#\szGZipDecoder.__init__rrc CsFz|j|WStjk r@}ztt||W5d}~XYnXdSr )r"r%r r&rr(rrr)rrrr_szGZipDecoder.decodec CsDz |jWStjk r>}ztt||W5d}~XYnXdSr r*r+rrrres zGZipDecoder.flushNr,rrrrr-Usr-c@s<eZdZdZddddZddddd Zddd d Zd S) 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' rrcCsLtdkrtddt|_d|_|t|jdr>|jj|_n |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 ImportErrorZ Decompressorr" seen_datahasattrr% _decompressprocessrrrrr#vs   zBrotliDecoder.__init__rrc CsR|sdSd|_z ||WStjk rL}ztt||W5d}~XYnXdSNrT)r3r5r1r&rr(r/rrrrs zBrotliDecoder.decodec Cs^|js dSzt|jdr"|jWdStjk rX}ztt||W5d}~XYnXdS)Nrfinish)r3r4r"r8r1r&rr(r+rrrrs  zBrotliDecoder.flushNr,rrrrr0ls  r0c@s<eZdZdZddddZddddd Zddd d Zd S) ZStandardDecoderz Handle 'zstd' RFC 8878 decoding. Requires `pip install zstandard`. Can be installed as a dependency of httpx using `pip install httpx[zstd]`. rrcCs*tdkrtddt|_d|_dS)NzXUsing 'ZStandardDecoder', ...Make sure to install httpx using `pip install httpx[zstd]`.F) zstandardr2ZstdDecompressorr!r"r3rrrrr#szZStandardDecoder.__init__rrc Cstdk s td|_t}zP||j||jjrh|jj rh|jj }t |_||j|q.Wn2tj k r}zt t||W5d}~XYnX|Sr)r:AssertionErrorr3ioBytesIOwriter"r%eof unused_datar;r!Z ZstdErrorrr(getvalue)rroutputrAr)rrrrs  zZStandardDecoder.decodecCs,|js dS|j}|jjs$tdt|S)NrzZstandard data is incomplete)r3r"rr@rr)rretrrrrs  zZStandardDecoder.flushNr,rrrrr9s r9c@s>eZdZdZdddddZdddd d Zdd d d ZdS) MultiDecoderzE Handle the case where multiple encodings have been applied. ztyping.Sequence[ContentDecoder]r)childrenr cCstt||_dS)zm 'children' should be a sequence of decoders in the order in which each was applied. N)listreversedrF)rrFrrrr#szMultiDecoder.__init__rrcCs|jD]}||}q|Sr )rFrrrchildrrrrs  zMultiDecoder.decodercCs&d}|jD]}|||}q |Sr)rFrrrIrrrrs zMultiDecoder.flushNr,rrrrrEsrEc@s@eZdZdZddddddZdd d d d Zd d ddZdS) ByteChunkerz> Handles returning byte content in fixed-size chunks. N int | Noner chunk_sizer cCst|_||_dSr )r=r>_buffer _chunk_sizerrNrrrr#s zByteChunker.__init__rz list[bytes]contentr csjdkr|r|gSgSj|jjkrjfddtdtjD}t|djkrjdj|Sjdj|dj|ddSngSdS)Ncsg|]}||jqSrrP.0irvaluerr sz&ByteChunker.decode..r rPrOr?tellrBrangelenseektruncaterrSchunksrrXrrs"        zByteChunker.decodercCs.|j}|jd|j|r*|gSgSNrrOrBr`rarXrrrrs   zByteChunker.flush)Nr,rrrrrKsrKc@s@eZdZdZddddddZdd d d d Zd d ddZdS) TextChunkerz> Handles returning text content in fixed-size chunks. NrLrrMcCst|_||_dSr )r=StringIOrOrPrQrrrr#s zTextChunker.__init__r( list[str]rRcsjdkr|r|gSgSj|jjkrjfddtdtjD}t|djkrjdj|Sjdj|dj|ddSngSdS)Ncsg|]}||jqSrrTrUrXrrrZsz&TextChunker.decode..rr[r\rbrrXrrs"        zTextChunker.decodercCs.|j}|jd|j|r*|gSgSrdrerXrrrr+s   zTextChunker.flush)Nr,rrrrrf srfc@s@eZdZdZddddddZddd d d Zdd d dZdS) TextDecoderz8 Handles incrementally decoding bytes into text utf-8r(r)encodingr cCst|dd|_dS)Nreplace)errors)codecsgetincrementaldecoderdecoder)rrkrrrr#7szTextDecoder.__init__rrcCs |j|Sr rprr rrrr:szTextDecoder.decodercCs|jddSr7rqrrrrr=szTextDecoder.flushN)rjr,rrrrri2sric@s<eZdZdZddddZdddd d Zddd d Zd S) LineDecoderz Handles incrementally reading lines from text. Has the same behaviour as the stdllib splitlines, but handling the input iteratively. rrcCsg|_d|_dSr$)buffer trailing_crrrrrr#IszLineDecoder.__init__r(rh)textr cCsd}|jrd|}d|_|dr4d|_|dd}|ssL       5*''=