U
'�7`� � @ s d Z ddlZddlZddlZddlmZmZmZmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ d Ze�d
d d g�Zedd�Zedd�Zeeeged f ZG d
d� de
�Zeded�dd�ZeZ z ddl m Z e j Z es�e Z W n e k
r� Y nX dS )z"Http related parsers and protocol.� N)�Any� Awaitable�Callable�Optional�Union)� CIMultiDict� )�AbstractStreamWriter)� BaseProtocol)�
NO_EXTENSIONS)� StreamWriter� HttpVersion�
HttpVersion10�
HttpVersion11r
�major�minorc @ s� e Zd Zd"eejedd�dd�Zee ej
d�dd��Z eed�dd ��Z dd�d
d �Z
d#edd
�dd�Zedd�dd�Zddd�eeedd�dd�Zeddd�dd �Zd$edd�d d �Zdd�d d!�ZdS )%r N)�protocol�loop�
on_chunk_sent�returnc C sH || _ |j| _|| _d | _d| _d| _d| _d| _d | _ d | _
|| _ d S )NFr ) � _protocol� transport�
_transportr �length�chunked� buffer_size� output_size�_eof� _compressZ
_drain_waiter�_on_chunk_sent)�selfr r r � r! �F/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/http_writer.py�__init__ s zStreamWriter.__init__)r c C s | j S �N)r �r r! r! r" r . s zStreamWriter.transportc C s | j S r$ )r r% r! r! r" r 2 s zStreamWriter.protocolc C s
d| _ d S )NT)r r% r! r! r" �enable_chunking6 s z StreamWriter.enable_chunking�deflate)�encodingr c C s* |dkrdt j nt j}t j|d�| _d S )NZgzip� )Zwbits)�zlibZ MAX_WBITSZ compressobjr )r r( Z zlib_moder! r! r" �enable_compression9 s z StreamWriter.enable_compression)�chunkr c C sP t |�}| j|7 _| j|7 _| jd ks8| j�� r@td��| j�|� d S )Nz!Cannot write to closing transport)�lenr r r Z
is_closing�ConnectionResetError�write)r r, �sizer! r! r" �_write= s zStreamWriter._writeTi )�drain�LIMIT)r, r2 r3 r c � s� | j dk r| � |�I dH t|t�r<|jt|�kr<|�d�}| jdk rZ| j�|�}|sZdS | jdk r�t|�}| j|kr�| j| | _n |d| j� }d| _|s�dS |r�| j r�dt|� �
d�}|| d }| � |� | j |kr�|r�d| _ | �
� I dH dS )z�Writes chunk of data to a stream.
write_eof() indicates end of stream.
writer can't be used after write_eof() method being called.
write() return drain future.
N�cr �%x
�ascii�
)r �
isinstance�
memoryview�nbytesr- �castr �compressr r �encoder1 r r2 )r r, r2 r3 � chunk_lenZ
chunk_len_prer! r! r" r/ F s2
zStreamWriter.write�CIMultiDict[str]�� status_line�headersr c � s t ||�}| �|� dS )z*Write request/response status and headers.N)�_serialize_headersr1 )r rA rB Zbufr! r! r" �
write_headersq s
zStreamWriter.write_headers� c � s� | j r
d S |r(| jd k r(| �|�I d H | jrv|r>| j�|�}|| j�� }|r�| jr�dt|� �d�}|| d }n.| jr�|r�dt|� �d�}|| d }nd}|r�| �|� | � � I d H d| _ d | _
d S )Nr5 r6 s
0
s 0
T) r r r r< �flushr r- r= r1 r2 r )r r, r> r! r! r" � write_eofy s*
zStreamWriter.write_eofc � s | j jdk r | j �� I dH dS )zFlush the write buffer.
The intended use is to write
await w.write(data)
await w.drain()
N)r r Z
_drain_helperr% r! r! r" r2 � s zStreamWriter.drain)N)r' )rE )�__name__�
__module__� __qualname__r
�asyncioZAbstractEventLoop�_T_OnChunkSentr# �propertyr Z Transportr r r& �strr+ �bytesr1 �bool�intr/ rD rG r2 r! r! r! r" r s8 ��
� � , � r r? r@ c C s. | d d� dd� |�� D �� }|�d�d S )N�
� c S s g | ]\}}|d | d �qS )z: rR r! )�.0�k�vr! r! r" �
<listcomp>� s z)_py_serialize_headers.<locals>.<listcomp>zutf-8r7 )�join�itemsr= )rA rB �liner! r! r" �_py_serialize_headers� s ���r[ )!�__doc__rK � collectionsr* �typingr r r r r Z multidictr �abcr Z
base_protocolr
Zhelpersr �__all__�
namedtupler
r r rO rL r rN r[ rC Zaiohttp._http_writerZ _http_writerZ_c_serialize_headers� ImportErrorr! r! r! r" �<module> s0