U dd &@sdZddlZddlZddlZddlZddlmZddlmZGdddZ Gdd d e Z Gd d d e Z Gd d d e Z Gddde Z Gddde ZGdddZGdddZGdddZGdddZe e e e dZedkreddS)zq Handlers for Content-Encoding. See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding N)brotli) DecodingErrorc@s*eZdZeedddZedddZdS)ContentDecoderdatareturncCs tdSNNotImplementedErrorselfrr3/tmp/pip-unpacked-wheel-fkuvgui5/httpx/_decoders.pydecodeszContentDecoder.decodercCs tdSr r r rrrflushszContentDecoder.flushN)__name__ __module__ __qualname__bytesrrrrrrrsrc@s.eZdZdZeedddZedddZdS) IdentityDecoderz Handle unencoded data. rcCs|Sr rr rrrrszIdentityDecoder.decodercCsdSNrrrrrrszIdentityDecoder.flushN)rrr__doc__rrrrrrrrsrc@s<eZdZdZddddZeedddZedd d ZdS) DeflateDecoderzZ Handle 'deflate' decoding. See: https://stackoverflow.com/questions/1838699 NrcCsd|_t|_dS)NT) first_attemptzlib decompressobj decompressorrrrr__init__*szDeflateDecoder.__init__rc Csx|j}d|_z|j|WStjk rr}z8|rTttj |_||WYStt ||W5d}~XYnXdSNF) rr decompressrerrorr MAX_WBITSrrstr)r rZwas_first_attemptexcrrrr.szDeflateDecoder.decodec CsDz |jWStjk r>}ztt||W5d}~XYnXdSr r rrr$rr&r r'rrrr9s zDeflateDecoder.flushrrrrr!rrrrrrrr#s rc@s<eZdZdZddddZeedddZedd d ZdS) GZipDecoderzW Handle 'gzip' decoding. See: https://stackoverflow.com/questions/1838699 NrcCsttjdB|_dS)N)rrr%r rrrrr!GszGZipDecoder.__init__rc CsFz|j|WStjk r@}ztt||W5d}~XYnXdSr )r r#rr$rr&r rr'rrrrJszGZipDecoder.decodec CsDz |jWStjk r>}ztt||W5d}~XYnXdSr r(r)rrrrPs zGZipDecoder.flushr*rrrrr+@sr+c@s<eZdZdZddddZeedddZedd d ZdS) 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' NrcCsLtdkrtddt|_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#) r ImportErrorZ Decompressorr seen_datahasattrr# _decompressprocessrrrrr!as   zBrotliDecoder.__init__rc CsR|sdSd|_z ||WStjk rL}ztt||W5d}~XYnXdSNrT)r0r2rr$rr&r-rrrrss zBrotliDecoder.decodec Cs^|js dSzt|jdr"|jWdStjk rX}ztt||W5d}~XYnXdS)Nrfinish)r0r1r r5rr$rr&r)rrrr|s  zBrotliDecoder.flushr*rrrrr.Ws  r.c@sDeZdZdZejeddddZeedddZ ed d d Z dS) MultiDecoderzE Handle the case where multiple encodings have been applied. N)childrenrcCstt||_dS)zm 'children' should be a sequence of decoders in the order in which each was applied. N)listreversedr7)r r7rrrr!szMultiDecoder.__init__rcCs|jD]}||}q|Sr )r7rr rchildrrrrs  zMultiDecoder.decodercCs&d}|jD]}|||}q |Sr)r7rrr:rrrrs zMultiDecoder.flush) rrrrtypingSequencerr!rrrrrrrr6sr6c@sReZdZdZd ejeddddZeej edddZ ej ed d d Z dS) ByteChunkerz> Handles returning byte content in fixed-size chunks. N chunk_sizercCst|_||_dSr )ioBytesIO_buffer _chunk_sizer r@rrrr!s zByteChunker.__init__contentrcsjdkr|r|gSgSj|jjkrjfddtdtjD}t|djkrjdj|Sjdj|dj|ddSngSdS)Ncsg|]}||jqSrrD.0ir valuerr sz&ByteChunker.decode..r rDrCwritetellgetvaluerangelenseektruncater rGchunksrrLrrs"        zByteChunker.decodercCs.|j}|jd|j|r*|gSgSNrrCrSrVrWrLrrrrs   zByteChunker.flush)N) rrrrr<Optionalintr!rListrrrrrrr>sr>c@sReZdZdZd ejeddddZeej edddZ ej ed d d Z dS) TextChunkerz> Handles returning text content in fixed-size chunks. Nr?cCst|_||_dSr )rAStringIOrCrDrErrrr!s zTextChunker.__init__rFcsjdkr|gSj|jjkrjfddtdtjD}t|djkrjdj|Sjdj|dj|ddSngSdS)Ncsg|]}||jqSrrHrIrLrrrNsz&TextChunker.decode..rrOrPrXrrLrrs"        zTextChunker.decodercCs.|j}|jd|j|r*|gSgSrZr[rLrrrrs   zTextChunker.flush)N) rrrrr<r\r]r!r&r^rrrrrrr_sr_c@s>eZdZdZd edddZeedddZed d d Zd S) TextDecoderz8 Handles incrementally decoding bytes into text utf-8)encodingcCst|dd|_dS)Nreplace)errors)codecsgetincrementaldecoderdecoder)r rcrrrr!szTextDecoder.__init__rcCs |j|Sr rhrr rrrrszTextDecoder.decodercCs|jddSr4rirrrrrszTextDecoder.flushN)rb) rrrrr&r!rrrrrrrrasrac@sHeZdZdZddddZeejedddZejedd d Z dS) LineDecoderz Handles incrementally reading lines from text. Has the same behaviour as the stdllib splitlines, but handling the input iteratively. NrcCsg|_d|_dSr")buffer trailing_crrrrrr! szLineDecoder.__init__)textrcCsd}|jrd|}d|_|dr4d|_|dd}|sr_rarjZSUPPORTED_DECODERSrtrrrrs.   5'':