Edit File: 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