o
�J�h / � @ 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 �MC:\pinokio\api\whisper-webui.git\app\env\lib\site-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 Zwas_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
r/ r0 r r r r e r1 zGZipDecoder.flushNr2 r r r3 r r r r r4 U s
r4 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� ImportErrorZ 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= r9 r* r r, r7 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& r@ r9 r* r r, r0 r r r r � s
��zBrotliDecoder.flushNr2 r r r3 r r r r r8 l s
r8 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" )rB r; �io�BytesIO�writer& r) �eof� unused_datarC r% Z ZstdErrorr r, �getvalue)r r �outputrH 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 rG r r )r �retr r r r � s
zZStandardDecoder.flushNr2 r r r3 r r r r rA � s
rA 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�reversedrM )r rM r r r r' � s zMultiDecoder.__init__r r c C s | j D ]}|�|�}q|S r
)rM r �r r �childr r r r � s
zMultiDecoder.decodec C s& d}| j D ] }|�|�|�� }q|S r )rM r r rQ r r r r � s
zMultiDecoder.flushN)rM rN r r! r r r3 r r r r rL � s
rL 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
)rD rE �_buffer� _chunk_size�r rU 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 �rY ��.0�i�r �valuer r �
<listcomp>� � ��z&ByteChunker.decode.<locals>.<listcomp>r ������ rY rX rF �tellrI �range�len�seek�truncate�r r\ �chunksr rd r r � �"
�
zByteChunker.decodec C �. | j �� }| j �d� | j �� |r|gS g S �Nr �rX rI rm rn rd r r r r �
zByteChunker.flushr
�rU rV r r! )r\ r r r] )r r] r3 r r r r rT � �
rT c @ rS )� TextChunkerz>
Handles returning text content in fixed-size chunks.
NrU rV r r! c C rW r
)rD �StringIOrX rY rZ r r r r' r[ zTextChunker.__init__r\ r, � list[str]c r^ )Nc r_ r r` ra rd r r rf rg z&TextChunker.decode.<locals>.<listcomp>r rh ri ro r rd r r rq zTextChunker.decodec C rr rs rt rd r r r r + ru zTextChunker.flushr
rv )r\ r, r rz �r rz r3 r r r r rx rw rx 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 r6 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 r? r� r r r r r = s zTextDecoder.flushN)r} )r~ r, r r! )r r r r, )r r, r3 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, rz 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
�
FTrh r r � )r� �endswith�
splitlinesrl r� �append�join�pop)r r� Z
NEWLINE_CHARSZtrailing_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.flushNr2 )r� r, r rz r{ r3 r r r r r� A s
&r� )�identity�gzipZdeflate�br�zstdr� r� )r �
__future__r r� rD �typingr$ � _exceptionsr r9 r: Z
brotlicffirB r r r r4 r8 rA rL rT rx r| r� �SUPPORTED_DECODERSr� r r r r �<module> sX ���
� 5*''=�
�