File: //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/http_writer.cpython-37.pyc
B
�k�aX � @ s* d Z ddlZddlZddlmZmZmZmZmZm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ d ZG d
d � d e�Zedd�Zedd�Zeeeged f Zeed ged f ZG d
d� de
�Zeed�dd�Zed ed�dd�Z e Z y ddl m Z e j Z e�s e Z W n e!k
�r$ Y nX dS )z"Http related parsers and protocol.� N)�Any� Awaitable�Callable�
NamedTuple�Optional�Union)� CIMultiDict� )�AbstractStreamWriter)� BaseProtocol)�
NO_EXTENSIONS)� StreamWriter� HttpVersion�
HttpVersion10�
HttpVersion11c @ s e Zd ZU eed< eed< dS )r �major�minorN)�__name__�
__module__� __qualname__�int�__annotations__� r r �F/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/http_writer.pyr s
r zCIMultiDict[str]c @ s� e Zd Zd"eejeedd�dd�Ze e
ej d�dd��Z e ed�dd ��Z
dd�d
d �Zd ejfeedd
�dd�Zedd�dd�Zddd�eeedd�dd�Zeddd�dd �Zd#edd�d d �Zdd�d d!�ZdS )$r
N)�protocol�loop�
on_chunk_sent�on_headers_sent�returnc C sN || _ |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�_on_headers_sent)�selfr r r r r r r �__init__ s zStreamWriter.__init__)r c C s | j S )N)r! )r* r r r r 5 s zStreamWriter.transportc C s | j S )N)r )r* r r r r 9 s zStreamWriter.protocolc C s
d| _ d S )NT)r# )r* r r r �enable_chunking= s z StreamWriter.enable_chunking�deflate)�encoding�strategyr c C s, |dkrdt j nt j}t j||d�| _d S )N�gzip� )Zwbitsr/ )�zlib� MAX_WBITS� compressobjr' )r* r. r/ Z zlib_moder r r �enable_compression@ 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* r6 �sizer r r �_writeF s zStreamWriter._writeTi )�drain�LIMIT)r6 r< r= 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 z%x
�asciis
)r( �
isinstance�
memoryview�nbytesr7 �castr' �compressr" r# �encoder; r$ r< )r* r6 r< r= � chunk_lenZ
chunk_len_prer r r r9 O s2
zStreamWriter.writezCIMultiDict[str])� status_line�headersr c � s2 | j dk r| � |�I dH t||�}| �|� dS )z*Write request/response status and headers.N)r) �_serialize_headersr; )r* rG rH �bufr r r �
write_headersz 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 )Nz%x
r? s
0
s 0
T) r& r( r' rD �flushr# r7 rE r; r< r! )r* r6 rF r r r � write_eof� 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_helper)r* r r r r< � s zStreamWriter.drain)NN)rL )r r r r �asyncioZAbstractEventLoop�_T_OnChunkSent�_T_OnHeadersSentr+ �propertyr Z Transportr r r, r2 �Z_DEFAULT_STRATEGY�strr r5 �bytesr; �boolr9 rK rN r< r r r r r
s"
* r
)�stringr c C s d| ksd| krt d��| S )N�
�
zRNewline or carriage return detected in headers. Potential header injection attack.)�
ValueError)rW r r r � _safe_header� s r[ )rG rH r c C s2 dd� |� � D �}| d d�|� d }|�d�S )Nc s s& | ] \}}t |�d t |� V qdS )z: N)r[ )�.0�k�vr r r � <genexpr>� s z(_py_serialize_headers.<locals>.<genexpr>z
z
zutf-8)�items�joinrE )rG rH Z headers_gen�liner r r �_py_serialize_headers� s rc )"�__doc__rO r2 �typingr r r r r r Z multidictr �abcr
Z
base_protocolr Zhelpersr �__all__r r r rU rP rQ r
rT r[ rc rI Zaiohttp._http_writerZ _http_writerZ_c_serialize_headers� ImportErrorr r r r �<module> s2