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_chunkingZdeflate)�encoding�strategyr c C s, |dkrdt j nt j}t j||d�| _d S )NZgzip� )Zwbitsr. )�zlibZ MAX_WBITSZ 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* r2 �sizer r r �_writeF s zStreamWriter._writeTi )�drain�LIMIT)r2 r8 r9 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�nbytesr3 �castr' �compressr" r# �encoder7 r$ r8 )r* r2 r8 r9 � chunk_lenZ
chunk_len_prer r r r5 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_headersr7 )r* rC rD Zbufr 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' r@ �flushr# r3 rA r7 r8 r! )r* r2 rB 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 r8 � s zStreamWriter.drain)NN)rG )r r r r �asyncioZAbstractEventLoop�_T_OnChunkSent�_T_OnHeadersSentr+ �propertyr Z Transportr r r, r0 ZZ_DEFAULT_STRATEGY�strr r1 �bytesr7 �boolr5 rF rI r8 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)rQ r r r � _safe_header� s rU )rC rD r c C s2 dd� |� � D �}| d d�|� d }|�d�S )Nc s s& | ] \}}t |�d t |� V qdS )z: N)rU )�.0�k�vr r r � <genexpr>� s z(_py_serialize_headers.<locals>.<genexpr>z
z
zutf-8)�items�joinrA )rC rD Z headers_gen�liner r r �_py_serialize_headers� s r] )"�__doc__rJ r0 �typingr r r r r r Z multidictr �abcr
Z
base_protocolr Zhelpersr �__all__r r r rO rK rL r
rN rU r] rE Zaiohttp._http_writerZ _http_writerZ_c_serialize_headers� ImportErrorr r r r �<module> s2