U
    {f&                     @  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kred 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ddZddddZdS )	ContentDecoderbytesdatareturnc                 C  s
   t  d S NNotImplementedErrorselfr	    r   N/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/httpx/_decoders.pydecode   s    zContentDecoder.decoder
   c                 C  s
   t  d S r   r   r   r   r   r   flush   s    zContentDecoder.flushN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   c                   @  s.   e Zd ZdZdddddZddddZd	S )
IdentityDecoderz 
    Handle unencoded data.
    r   r   c                 C  s   |S r   r   r   r   r   r   r      s    zIdentityDecoder.decoder   c                 C  s   dS N    r   r   r   r   r   r   !   s    zIdentityDecoder.flushN)r   r   r   __doc__r   r   r   r   r   r   r      s   r   c                   @  s<   e Zd ZdZddddZddddd	Zddd
dZdS )DeflateDecoderzZ
    Handle 'deflate' decoding.

    See: https://stackoverflow.com/questions/1838699
    Noner   c                 C  s   d| _ t | _d S )NT)first_attemptzlibdecompressobjdecompressorr   r   r   r   __init__,   s    zDeflateDecoder.__init__r   r   c              
   C  sx   | j }d| _ z| j|W S  tjk
rr } z8|rTttj | _| | W Y S tt	||W 5 d }~X Y nX d S 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  sD   z| j  W S  tjk
r> } ztt||W 5 d }~X Y nX d S r   r"   r   r    r&   r   r(   r   r)   r   r   r   r   ;   s    zDeflateDecoder.flushNr   r   r   r   r#   r   r   r   r   r   r   r   %   s   r   c                   @  s<   e Zd ZdZddddZddddd	Zddd
dZdS )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   s    zGZipDecoder.__init__r   r   c              
   C  sF   z| j |W S  tjk
r@ } ztt||W 5 d }~X Y nX d S r   )r"   r%   r    r&   r   r(   r   r	   r)   r   r   r   r   L   s    zGZipDecoder.decodec              
   C  sD   z| j  W S  tjk
r> } ztt||W 5 d }~X Y nX d S r   r*   r+   r   r   r   r   R   s    zGZipDecoder.flushNr,   r   r   r   r   r-   B   s   r-   c                   @  s<   e Zd ZdZddddZddddd	Zddd
dZd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'
    r   r   c                 C  sL   t d krtdd t  | _d| _|  t| jdr>| jj| _n
| 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  sR   |sdS d| _ z| |W S  tjk
rL } ztt||W 5 d }~X Y nX d S Nr   T)r2   r4   r   r&   r   r(   r/   r   r   r   r   u   s    zBrotliDecoder.decodec              
   C  s^   | j s
dS zt| jdr"| j  W dS  tjk
rX } ztt||W 5 d }~X Y nX d S )Nr   finish)r2   r3   r"   r7   r   r&   r   r(   r+   r   r   r   r   ~   s    
zBrotliDecoder.flushNr,   r   r   r   r   r0   Y   s   		r0   c                   @  s>   e Zd ZdZdddddZdddd	d
ZddddZdS )MultiDecoderzE
    Handle the case where multiple encodings have been applied.
    ztyping.Sequence[ContentDecoder]r   )childrenr
   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reversedr9   )r   r9   r   r   r   r#      s    zMultiDecoder.__init__r   r   c                 C  s   | j D ]}||}q|S r   )r9   r   r   r	   childr   r   r   r      s    
zMultiDecoder.decoder   c                 C  s&   d}| j D ]}|||  }q
|S r   )r9   r   r   r<   r   r   r   r      s    
zMultiDecoder.flushNr,   r   r   r   r   r8      s   r8   c                   @  s@   e Zd ZdZddddddZdd	d
ddZd	dddZdS )ByteChunkerz>
    Handles returning byte content in fixed-size chunks.
    N
int | Noner   
chunk_sizer
   c                 C  s   t  | _|| _d S r   )ioBytesIO_buffer_chunk_sizer   rA   r   r   r   r#      s    
zByteChunker.__init__r   zlist[bytes]contentr
   c                   s    j d kr|r|gS g S  j|  j  j kr j  fddtdt j D }t|d  j kr jd  j  |S  jd  j|d   j  |d d S ng S d S )Nc                   s   g | ]}|| j   qS r   rE   .0ir   valuer   r   
<listcomp>   s   z&ByteChunker.decode.<locals>.<listcomp>r   	rE   rD   writetellgetvaluerangelenseektruncater   rH   chunksr   rM   r   r      s"    



zByteChunker.decoder   c                 C  s.   | j  }| j d | j   |r*|gS g S Nr   rD   rT   rW   rX   rM   r   r   r   r      s    

zByteChunker.flush)Nr,   r   r   r   r   r>      s   r>   c                   @  s@   e Zd ZdZddddddZdd	d
ddZd	dddZdS )TextChunkerz>
    Handles returning text content in fixed-size chunks.
    Nr?   r   r@   c                 C  s   t  | _|| _d S r   )rB   StringIOrD   rE   rF   r   r   r   r#      s    
zTextChunker.__init__r(   	list[str]rG   c                   s    j d kr|r|gS g S  j|  j  j kr j  fddtdt j D }t|d  j kr jd  j  |S  jd  j|d   j  |d d S ng S d S )Nc                   s   g | ]}|| j   qS r   rI   rJ   rM   r   r   rO      s   z&TextChunker.decode.<locals>.<listcomp>r   rP   rQ   rY   r   rM   r   r      s"    



zTextChunker.decoder   c                 C  s.   | j  }| j d | j   |r*|gS g S r[   r\   rM   r   r   r   r      s    

zTextChunker.flush)Nr,   r   r   r   r   r]      s   r]   c                   @  s@   e Zd ZdZddddddZddd	d
dZddddZdS )TextDecoderz8
    Handles incrementally decoding bytes into text
    utf-8r(   r   )encodingr
   c                 C  s   t |dd| _d S )Nreplace)errors)codecsgetincrementaldecoderdecoder)r   rb   r   r   r   r#      s    zTextDecoder.__init__r   r   c                 C  s   | j |S r   rg   r   r   r   r   r   r      s    zTextDecoder.decoder   c                 C  s   | j ddS r6   rh   r   r   r   r   r      s    zTextDecoder.flushN)ra   r,   r   r   r   r   r`      s   r`   c                   @  s<   e Zd ZdZddddZdddd	d
Zd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__r(   r_   )textr
   c                 C  s   d}| j rd| }d| _ |dr4d| _ |d d }|s<g S |d |k}| }t|dkrt|st| j|d  g S | jrd| j|d  g|dd   }g | _|s| g| _|S )	Nu   
  FTrP   r   r    )rk   endswith
splitlinesrV   rj   appendjoinpop)r   rl   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 )Nrn   F)rj   rk   rr   )r   rt   r   r   r   r   6  s    zLineDecoder.flushNr,   r   r   r   r   ri     s   &ri   )identitygzipdeflatebrrx   )r   
__future__r   re   rB   typingr    Z_compatr   _exceptionsr   r   r   r   r-   r0   r8   r>   r]   r`   ri   ZSUPPORTED_DECODERSrs   r   r   r   r   <module>   s0   5''=