File: //usr/local/lib/python3.10/dist-packages/httpx/__pycache__/_decoders.cpython-310.pyc
o
���g / � @ s� d Z ddlmZ ddlZddlZddlZddlZddlmZ zddl Z W n e
y? zddl Z W n e
y< dZ Y nw Y nw zddl Z W n e
yQ dZ Y nw G dd� d�Z
G dd � d e
�ZG d
d � d e
�ZG d d
� d
e
�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� d�ZG dd� d�ZG dd� d�ZG dd� d�Zeeeeed �Ze du r�e�d � e du r�e�d � dS dS ) zq
Handlers for Content-Encoding.
See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding
� )� annotationsN� )�
DecodingErrorc @ s e Zd Zd dd�Zd
dd�ZdS ) �ContentDecoder�data�bytes�returnc C � t � ��N��NotImplementedError��selfr � r �:/usr/local/lib/python3.10/dist-packages/httpx/_decoders.py�decode% � zContentDecoder.decodec C r r
r �r r r r �flush( r zContentDecoder.flushN�r r r r �r r )�__name__�
__module__� __qualname__r r r r r r r $ s
r c @ s$ e Zd ZdZd
dd�Zd dd�Zd S ) �IdentityDecoderz
Handle unencoded data.
r r r c C s |S r
r r
r r r r 1 � zIdentityDecoder.decodec C s dS �N� r r r r r r 4 r zIdentityDecoder.flushNr r )r r r �__doc__r r r r r r r , s
r c @ �. e Zd ZdZd
dd�Zddd �Zdd
d �Zd S )�DeflateDecoderzZ
Handle 'deflate' decoding.
See: https://stackoverflow.com/questions/1838699
r �Nonec C s d| _ t�� | _d S �NT)�
first_attempt�zlib�
decompressobj� decompressorr r r r �__init__? s zDeflateDecoder.__init__r r c
C sn | j }d| _ z| j�|�W S tjy6 } z |r+t�tj �| _| �|�W Y d }~S tt |��|�d }~ww �NF)
r# r& �
decompressr$ �errorr% � MAX_WBITSr r �str)r r �was_first_attempt�excr r r r C s ��zDeflateDecoder.decodec
C �6 z| j �� W S tjy } ztt|��|�d }~ww r
�r& r r$ r* r r, �r r. r r r r N � ��zDeflateDecoder.flushN�r r! r r �r r r r r' r r r r r r r 8 s
r c @ r )� GZipDecoderzW
Handle 'gzip' decoding.
See: https://stackoverflow.com/questions/1838699
r r! c C s t �t jdB �| _d S )N� )r$ r% r+ r& r r r r r' \ � zGZipDecoder.__init__r r c
C s8 z| j �|�W S tjy } ztt|��|�d }~ww r
)r& r) r$ r* r r, �r r r. r r r r _ s ��zGZipDecoder.decodec
C r/ r
r0 r1 r r r r e r2 zGZipDecoder.flushNr3 r r r4 r r r r r5 U s
r5 c @ 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'
r r! c C sN t d u r td�d �t �� | _d| _| t| jd�r | jj| _d S | jj| _d S )Nz�Using '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_data�hasattrr) � _decompress�processr r r r r' v s ��
zBrotliDecoder.__init__r r c
C sD |sdS d| _ z| �|�W S tjy! } ztt|��|�d }~ww �Nr T)r= r? r: r* r r, r8 r r r r � s ��zBrotliDecoder.decodec
C sP | j sdS zt| jd�r| j�� W dS tjy' } ztt|��|�d }~ww )Nr �finish)r= r> r& rB r: r* r r, r1 r r r r � s
��zBrotliDecoder.flushNr3 r r r4 r r r r r9 l s
r9 c @ r )�ZStandardDecoderz�
Handle 'zstd' RFC 8878 decoding.
Requires `pip install zstandard`.
Can be installed as a dependency of httpx using `pip install httpx[zstd]`.
r r! c C s* t d u r td�d �t �� �� | _d| _d S )NzXUsing 'ZStandardDecoder', ...Make sure to install httpx using `pip install httpx[zstd]`.F)� zstandardr; �ZstdDecompressorr% r& r= r r r r r' � s ��
zZStandardDecoder.__init__r r c
C s� t d usJ �d| _t�� }zA|�| j�|�� | jjr@| jjrJ| jj}t � � �
� | _|�| j�|�� | jjrE| jjs W |�� S W |�� S W |�� S W |�� S t j yb } zt t
|��|�d }~ww r" )rD r= �io�BytesIO�writer& r) �eof� unused_datarE r% � ZstdErrorr r, �getvalue)r r �outputrJ r. r r r r � s* �������zZStandardDecoder.decodec C s, | j sdS | j�� }| jjstd��t|�S )Nr z Zstandard data is incomplete)r= r& r rI r r )r �retr r r r � s
zZStandardDecoder.flushNr3 r r r4 r r r r rC � s
rC c @ s. e Zd ZdZddd�Zdd
d �Zdd d
�ZdS )� MultiDecoderzE
Handle the case where multiple encodings have been applied.
�children� typing.Sequence[ContentDecoder]r r! c C s t t|��| _dS )zm
'children' should be a sequence of decoders in the order in which
each was applied.
N)�list�reversedrP )r rP r r r r' � s zMultiDecoder.__init__r r c C s | j D ]}|�|�}q|S r
)rP r �r r �childr r r r � s
zMultiDecoder.decodec C s& d}| j D ] }|�|�|�� }q|S r )rP r r rT r r r r � s
zMultiDecoder.flushN)rP rQ r r! r r r4 r r r r rO � s
rO c @ �0 e Zd ZdZdddd�Zdd d
�Zddd�ZdS )� ByteChunkerz>
Handles returning byte content in fixed-size chunks.
N�
chunk_size�
int | Noner r! c C � t �� | _|| _d S r
)rF rG �_buffer� _chunk_size�r rX r r r r' � �
zByteChunker.__init__�contentr � list[bytes]c �� � j d u r |r
|gS g S � j�|� � j�� � j kr_� j�� �� �fdd�tdt��� j �D �}t|d �� j krF� j�d� � j�� |S � j�d� � j�|d � � j�� |d d� S g S )Nc � g | ] }�||� j � �qS r �r\ ��.0�i�r �valuer r �
<listcomp>� � ��z&ByteChunker.decode.<locals>.<listcomp>r ���� r\ r[ rH �tellrL �range�len�seek�truncate�r r_ �chunksr rg r r � �"
�
zByteChunker.decodec C �. | j �� }| j �d� | j �� |r|gS g S �Nr �r[ rL rp rq rg r r r r �
zByteChunker.flushr
�rX rY r r! )r_ r r r` )r r` r4 r r r r rW � �
rW c @ rV )� TextChunkerz>
Handles returning text content in fixed-size chunks.
NrX rY r r! c C rZ r
)rF �StringIOr[ r\ r] r r r r' r^ zTextChunker.__init__r_ r, � list[str]c ra )Nc rb r rc rd rg r r ri rj z&TextChunker.decode.<locals>.<listcomp>r rk rl rr r rg r r rt zTextChunker.decodec C ru rv rw rg r r r r + rx zTextChunker.flushr
ry )r_ r, r r} �r r} r4 r r r r r{ rz r{ c @ s0 e Zd ZdZdddd�Zdd d �Zdd
d�ZdS )� TextDecoderz8
Handles incrementally decoding bytes into text
�utf-8�encodingr, r r! c C s t �|�dd�| _d S )N�replace)�errors)�codecs�getincrementaldecoder�decoder)r r� r r r r' 7 r7 zTextDecoder.__init__r r c C s | j �|�S r
�r� r r
r r r r : s zTextDecoder.decodec C s | j �dd�S rA r� r r r r r = s zTextDecoder.flushN)r� )r� r, r r! )r r r r, )r r, r4 r r r r r 2 s
r c @ s. e Zd ZdZddd�Zdd 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.
r r! c C s g | _ d| _d S r( )�buffer� trailing_crr r r r r' I s
zLineDecoder.__init__�textr, r} c C s� d}| j r d| }d| _ |�d�rd| _ |d d� }|s g S |d |v }|�� }t|�dkr:|s:| j�|d � g S | jrQd�| j�|d g|dd � }g | _|sY|�� g| _|S ) Nu
�
FTrk r r � )r� �endswith�
splitlinesro r� �append�join�pop)r r� �
NEWLINE_CHARS�trailing_newline�linesr r r r M s(
" zLineDecoder.decodec C s. | j s| jsg S d�| j �g}g | _ d| _|S )Nr� F)r� r� r� )r r� r r r r s s zLineDecoder.flushNr3 )r� r, r r} r~ r4 r r r r r� A s
&r� )�identity�gzip�deflate�br�zstdr� r� )r �
__future__r r� rF �typingr$ � _exceptionsr r: r; �
brotlicffirD r r r r5 r9 rC rO rW r{ r r� �SUPPORTED_DECODERSr� r r r r �<module> sX ���
� 5*''=�
�