o
    Zh&                     @  s  d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl	m
Z
 G dd dZG d	d
 d
eZG dd deZG dd deZG dd deZG dd deZG dd dZG dd dZG dd dZG dd dZeeeedZedu red dS dS )zq
Handlers for Content-Encoding.

See: https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Encoding
    )annotationsN   )brotli)DecodingErrorc                   @  s    e Zd Zd	ddZd
ddZdS )ContentDecoderdatabytesreturnc                 C     t  NNotImplementedErrorselfr    r   F/var/www/html/lang_env/lib/python3.10/site-packages/httpx/_decoders.pydecode      zContentDecoder.decodec                 C  r
   r   r   r   r   r   r   flush   r   zContentDecoder.flushNr   r   r	   r   r	   r   )__name__
__module____qualname__r   r   r   r   r   r   r      s    
r   c                   @  s$   e Zd ZdZd
ddZdddZd	S )IdentityDecoderz 
    Handle unencoded data.
    r   r   r	   c                 C  s   |S r   r   r   r   r   r   r         zIdentityDecoder.decodec                 C  s   dS N    r   r   r   r   r   r   !   r   zIdentityDecoder.flushNr   r   )r   r   r   __doc__r   r   r   r   r   r   r      s    
r   c                   @  .   e Zd ZdZdddZddd	Zdd
dZd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   zDeflateDecoder.__init__r   r   c              
   C  sn   | j }d| _ z| j|W S  tjy6 } z|r+ttj | _| |W  Y d }~S tt	||d }~ww NF)
r#   r&   
decompressr$   errorr%   	MAX_WBITSr   r   str)r   r   Zwas_first_attemptexcr   r   r   r   0   s   zDeflateDecoder.decodec              
   C  6   z| j  W S  tjy } ztt||d }~ww r   r&   r   r$   r*   r   r,   r   r-   r   r   r   r   ;      zDeflateDecoder.flushNr	   r"   r   r   r   r   r   r   r'   r   r   r   r   r   r   r!   %   s
    

r!   c                   @  r    )GZipDecoderzW
    Handle 'gzip' decoding.

    See: https://stackoverflow.com/questions/1838699
    r	   r"   c                 C  s   t t jdB | _d S )N   )r$   r%   r+   r&   r   r   r   r   r'   I      zGZipDecoder.__init__r   r   c              
   C  s8   z| j |W S  tjy } ztt||d }~ww r   )r&   r)   r$   r*   r   r,   r   r   r-   r   r   r   r   L   s   zGZipDecoder.decodec              
   C  r.   r   r/   r0   r   r   r   r   R   r1   zGZipDecoder.flushNr2   r   r   r3   r   r   r   r   r4   B   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	tdd t  | _d| _|  t| jdr | jj| _d S | jj| _d S )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_datahasattrr)   _decompressprocessr   r   r   r   r'   c   s   
zBrotliDecoder.__init__r   r   c              
   C  sD   |sdS d| _ z| |W S  tjy! } ztt||d }~ww Nr   T)r:   r<   r   r*   r   r,   r7   r   r   r   r   u   s   zBrotliDecoder.decodec              
   C  sP   | j sdS zt| jdr| j  W dS  tjy' } ztt||d }~ww )Nr   finish)r:   r;   r&   r?   r   r*   r   r,   r0   r   r   r   r   ~   s   
zBrotliDecoder.flushNr2   r   r   r3   r   r   r   r   r8   Y   s
    
	
	r8   c                   @  s.   e Zd ZdZdddZdd
dZdddZdS )MultiDecoderzE
    Handle the case where multiple encodings have been applied.
    childrentyping.Sequence[ContentDecoder]r	   r"   c                 C  s   t t|| _dS )zm
        'children' should be a sequence of decoders in the order in which
        each was applied.
        N)listreversedrA   )r   rA   r   r   r   r'      s   zMultiDecoder.__init__r   r   c                 C  s   | j D ]}||}q|S r   )rA   r   r   r   childr   r   r   r      s   
zMultiDecoder.decodec                 C  s&   d}| j D ]}|||  }q|S r   )rA   r   r   rE   r   r   r   r      s   
zMultiDecoder.flushN)rA   rB   r	   r"   r   r   r3   r   r   r   r   r@      s
    

r@   c                   @  0   e Zd ZdZddddZdddZdddZdS )ByteChunkerz>
    Handles returning byte content in fixed-size chunks.
    N
chunk_size
int | Noner	   r"   c                 C     t  | _|| _d S r   )ioBytesIO_buffer_chunk_sizer   rI   r   r   r   r'         

zByteChunker.__init__contentr   list[bytes]c                       j d u r|r
|gS g S  j|  j  j kr_ j  fddtdt j D }t|d  j krF jd  j  |S  jd  j|d   j  |d d S g S )Nc                      g | ]}|| j   qS r   rO   .0ir   valuer   r   
<listcomp>       z&ByteChunker.decode.<locals>.<listcomp>r   	rO   rN   writetellgetvaluerangelenseektruncater   rR   chunksr   rZ   r   r      "   



zByteChunker.decodec                 C  .   | j  }| j d | j   |r|gS g S Nr   rN   rb   re   rf   rZ   r   r   r   r         

zByteChunker.flushr   rI   rJ   r	   r"   )rR   r   r	   rS   )r	   rS   r3   r   r   r   r   rH      
    
rH   c                   @  rG   )TextChunkerz>
    Handles returning text content in fixed-size chunks.
    NrI   rJ   r	   r"   c                 C  rK   r   )rL   StringIOrN   rO   rP   r   r   r   r'      rQ   zTextChunker.__init__rR   r,   	list[str]c                   rT   )Nc                   rU   r   rV   rW   rZ   r   r   r\      r]   z&TextChunker.decode.<locals>.<listcomp>r   r^   r_   rg   r   rZ   r   r      ri   zTextChunker.decodec                 C  rj   rk   rl   rZ   r   r   r   r      rm   zTextChunker.flushr   rn   )rR   r,   r	   rr   r	   rr   r3   r   r   r   r   rp      ro   rp   c                   @  s0   e Zd ZdZddddZdddZdddZdS )TextDecoderz8
    Handles incrementally decoding bytes into text
    utf-8encodingr,   r	   r"   c                 C  s   t |dd| _d S )Nreplace)errors)codecsgetincrementaldecoderdecoder)r   rv   r   r   r   r'      r6   zTextDecoder.__init__r   r   c                 C  s   | j |S r   r{   r   r   r   r   r   r      s   zTextDecoder.decodec                 C  s   | j ddS r>   r|   r   r   r   r   r      s   zTextDecoder.flushN)ru   )rv   r,   r	   r"   )r   r   r	   r,   )r	   r,   r3   r   r   r   r   rt      s
    
rt   c                   @  s.   e Zd ZdZdddZdd	d
ZdddZd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'     s   
zLineDecoder.__init__textr,   rr   c                 C  s   d}| j rd| }d| _ |drd| _ |d d }|sg S |d |v }| }t|dkr:|s:| j|d  g S | jrQd| j|d  g|dd   }g | _|sY| g| _|S )	Nu   
  FTr^   r   r    )r   endswith
splitlinesrd   r~   appendjoinpop)r   r   ZNEWLINE_CHARSZtrailing_newlinelinesr   r   r   r     s(   
"zLineDecoder.decodec                 C  s.   | j s| jsg S d| j g}g | _ d| _|S )Nr   F)r~   r   r   )r   r   r   r   r   r   6  s   zLineDecoder.flushNr2   )r   r,   r	   rr   rs   r3   r   r   r   r   r}     s
    

&r}   )identitygzipdeflatebrr   )r   
__future__r   ry   rL   typingr$   _compatr   _exceptionsr   r   r   r!   r4   r8   r@   rH   rp   rt   r}   ZSUPPORTED_DECODERSr   r   r   r   r   <module>   s4    5''=