File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/compression_utils.cpython-310.pyc
o
���g1 � @ s� d dl Z d dlZd dlmZ d dlmZmZ zzd dlZW n
e y* d dlZY nw dZ
W n e y9 dZ
Y nw dZ ddee de
d efd
d �ZG d d
� d
�ZG dd� de�ZG dd� de�ZG dd� d�ZdS )� N)�Executor)�Optional�castTFi �encoding�suppress_deflate_header�returnc C s$ | dkr dt j S |rt j S t jS )N�gzip� )�zlib� MAX_WBITS)r r � r �D/usr/local/lib/python3.10/dist-packages/aiohttp/compression_utils.py�encoding_to_mode s
r c @ s0 e Zd Zdefdedee dee fdd�ZdS )�ZlibBaseHandlerN�mode�executor�max_sync_chunk_sizec C s || _ || _|| _d S �N)�_mode� _executor�_max_sync_chunk_size)�selfr r r r r r
�__init__ s
zZlibBaseHandler.__init__)�__name__�
__module__� __qualname__�MAX_SYNC_CHUNK_SIZE�intr r r r r r r
r s ����r c s� e Zd Zddddejdefdee dedee dee de dee
d ee f� fd
d �
Z d e d
e fdd�Z
d e d
e fdd�Zejfde d
e fdd�Z� ZS )�ZLibCompressorNFr r �level�wbits�strategyr r c sb t � j|d u r t||�n|||d� |d u r tj| j|d�| _n
tj| j||d�| _t�� | _ d S )N�r r r )r r! )r r! r )
�superr r r
� compressobjr � _compressor�asyncio�Lock�_compress_lock)r r r r r r! r r �� __class__r r
r * s
�� �zZLibCompressor.__init__�datar c C � | j �|�S r )r% �compress�r r+ r r r
�
compress_syncE � z ZLibCompressor.compress_syncc � s� �| j 4 I dH �5 | jdur.t|�| jkr.t�� �| j| jj|�I dH W d �I dH S | � |�W d �I dH S 1 I dH sCw Y dS )aC Compress the data and returned the compressed bytes.
Note that flush() must be called after the last call to compress()
If the data size is large than the max_sync_chunk_size, the compression
will be done in the executor. Otherwise, the compression will be done
in the event loop.
N)
r( r �lenr& �get_running_loop�run_in_executorr r% r- r/ r. r r r
r- H s �
�� 0�zZLibCompressor.compressr c C r, r )r% �flush)r r r r r
r4 _ r0 zZLibCompressor.flush)r r r r
�Z_DEFAULT_STRATEGYr r �str�boolr r r �bytesr/ r- �Z_FINISHr4 �
__classcell__r r r) r
r ) s4 �������� r c
s� e Zd Zdddefdee dedee dee f� fdd�
Z dd
e
d ed e
fd
d�Z dd
e
d ed e
fdd�Z dded e
fdd�Z
ed efdd��Zed e
fdd��Zed e
fdd��Z� ZS )�ZLibDecompressorNFr r r r c s, t � jt||�||d� tj| jd�| _d S )Nr" )r )r# r r r
�
decompressobjr �
_decompressor)r r r r r r) r r
r d s �zZLibDecompressor.__init__r r+ �
max_lengthr c C s | j �||�S r )r= �
decompress�r r+ r> r r r
�decompress_syncr s z ZLibDecompressor.decompress_syncc � sF �| j dur t|�| j kr t�� �| j| jj||�I dH S | �||�S )z�Decompress the data and return the decompressed bytes.
If the data size is large than the max_sync_chunk_size, the decompression
will be done in the executor. Otherwise, the decompression will be done
in the event loop.
N) r r1 r& r2 r3 r r= r? rA r@ r r r
r? u s �
� zZLibDecompressor.decompress�lengthc C s |dkr
| j �|�S | j �� S )Nr )r= r4 )r rB r r r
r4 � s
���zZLibDecompressor.flushc C � | j jS r )r= �eof�r r r r
rD � � zZLibDecompressor.eofc C rC r )r= �unconsumed_tailrE r r r
rG � rF z ZLibDecompressor.unconsumed_tailc C rC r )r= � unused_datarE r r r
rH � rF z ZLibDecompressor.unused_data)r )r r r r r r6 r7 r r r r8 rA r? r4 �propertyrD rG rH r: r r r) r
r; c s. �����r; c @ s6 e Zd Zd
dd�Zdedefdd�Zdefdd �ZdS ) �BrotliDecompressorr Nc C s t std��t�� | _d S )NzIThe brotli decompression is not available. Please install `Brotli` module)�
HAS_BROTLI� RuntimeError�brotli� Decompressor�_objrE r r r
r � s
�zBrotliDecompressor.__init__r+ c C s0 t | jd�rtt| j�|��S tt| j�|��S )Nr? )�hasattrrO r r8 r? �processr. r r r
rA � s z"BrotliDecompressor.decompress_syncc C s t | jd�rtt| j�� �S dS )Nr4 � )rP rO r r8 r4 rE r r r
r4 � s zBrotliDecompressor.flush)r N)r r r r r8 rA r4 r r r r
rJ � s
rJ )NF)r& r
�concurrent.futuresr �typingr r �
brotlicffirM � ImportErrorrK r r6 r7 r r r r r; rJ r r r r
�<module> s8 � ����
�
:6