U Wi,@s.dZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z GdddZ Gd d d e Z Gd d d e ZGd dde ZGddde ZGddde ZGddde ZGdddZGdddZGdddZGdddZe eeeedZedkrede dkr*eddS) zq Handlers for Content-Encoding. See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding ) annotationsN)brotlizstd) DecodingErrorc@s*eZdZdddddZddddZdS) ContentDecoderbytesdatareturncCs tdSNNotImplementedErrorselfr rU/home/afrizal/public_html/fastapi/venv/lib/python3.8/site-packages/httpx/_decoders.pydecodeszContentDecoder.decoder cCs tdSr r rrrrflushszContentDecoder.flushN)__name__ __module__ __qualname__rrrrrrrsrc@s.eZdZdZdddddZddddZd S) IdentityDecoderz Handle unencoded data. rr cCs|Sr rrrrrrszIdentityDecoder.decodercCsdSNrrrrrr"szIdentityDecoder.flushN)rrr__doc__rrrrrrrsrc@s<eZdZdZddddZddddd Zddd d Zd S) DeflateDecoderzZ Handle 'deflate' decoding. See: https://stackoverflow.com/questions/1838699 NonercCsd|_t|_dS)NT) first_attemptzlib decompressobj decompressorrrrr__init__-szDeflateDecoder.__init__rr c Csx|j}d|_z|j|WStjk rr}z8|rTttj |_||WYStt ||W5d}~XYnXdSNF) r r# decompressr!errorr" MAX_WBITSrrstr)rr Zwas_first_attemptexcrrrr1szDeflateDecoder.decodec CsDz |jWStjk r>}ztt||W5d}~XYnXdSr r#rr!r'rr)rr*rrrr<s zDeflateDecoder.flushNrrrrr$rrrrrrr&s 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$JszGZipDecoder.__init__rr c CsFz|j|WStjk r@}ztt||W5d}~XYnXdSr )r#r&r!r'rr)rr r*rrrrMszGZipDecoder.decodec CsDz |jWStjk r>}ztt||W5d}~XYnXdSr r+r,rrrrSs zGZipDecoder.flushNr-rrrrr.Csr.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&) r ImportErrorZ Decompressorr# seen_datahasattrr& _decompressprocessrrrrr$ds   zBrotliDecoder.__init__rr c CsR|sdSd|_z ||WStjk rL}ztt||W5d}~XYnXdSNrT)r3r5rr'rr)r0rrrrvs zBrotliDecoder.decodec Cs^|js dSzt|jdr"|jWdStjk rX}ztt||W5d}~XYnXdS)Nrfinish)r3r4r#r8rr'rr)r,rrrrs  zBrotliDecoder.flushNr-rrrrr1Zs  r1c@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|_dS)NzXUsing 'ZStandardDecoder', ...Make sure to install httpx using `pip install httpx[zstd]`.)rr2ZstdDecompressorr"r#rrrrr$szZStandardDecoder.__init__rr c Cstdk s tt}zP||j||jjrb|jjrb|jj}t |_||j|q(Wn2tj k r}zt t ||W5d}~XYnX|Sr )rAssertionErrorioBytesIOwriter#r&eof unused_datar:r"Z ZstdErrorrr)getvalue)rr outputr@r*rrrrs  zZStandardDecoder.decodecCs"|j}|jjstdt|S)NzZstandard data is incomplete)r#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)listreversedrE)rrErrrr$szMultiDecoder.__init__rr cCs|jD]}||}q|Sr )rErrr childrrrrs  zMultiDecoder.decodercCs&d}|jD]}|||}q |Sr)rErrrHrrrrs zMultiDecoder.flushNr-rrrrrDsrDc@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_sizerrMrrrr$s zByteChunker.__init__rz list[bytes]contentr csjdkr|r|gSgSj|jjkrjfddtdtjD}t|djkrjdj|Sjdj|dj|ddSngSdS)Ncsg|]}||jqSrrO.0irvaluerr sz&ByteChunker.decode..r rOrNr>tellrArangelenseektruncaterrRchunksrrWrrs"        zByteChunker.decodercCs.|j}|jd|j|r*|gSgSNrrNrAr_r`rWrrrrs   zByteChunker.flush)Nr-rrrrrJsrJc@s@eZdZdZddddddZdd d d d Zd d ddZdS) TextChunkerz> Handles returning text content in fixed-size chunks. NrKrrLcCst|_||_dSr )r<StringIOrNrOrPrrrr$s zTextChunker.__init__r) list[str]rQcsjdkr|r|gSgSj|jjkrjfddtdtjD}t|djkrjdj|Sjdj|dj|ddSngSdS)Ncsg|]}||jqSrrSrTrWrrrYsz&TextChunker.decode..rrZr[rarrWrrs"        zTextChunker.decodercCs.|j}|jd|j|r*|gSgSrcrdrWrrrrs   zTextChunker.flush)Nr-rrrrresrec@s@eZdZdZddddddZddd d d Zdd d dZdS) TextDecoderz8 Handles incrementally decoding bytes into text utf-8r)r)encodingr cCst|dd|_dS)Nreplace)errors)codecsgetincrementaldecoderdecoder)rrjrrrr$!szTextDecoder.__init__rr cCs |j|Sr rorrrrrr$szTextDecoder.decodercCs|jddSr7rprrrrr'szTextDecoder.flushN)rir-rrrrrhsrhc@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$3szLineDecoder.__init__r)rg)textr cCsd}|jrd|}d|_|dr4d|_|dd}|ss8   5&''=