File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/http_writer.cpython-310.pyc
o
���g� � @ s� d Z ddlZddlZddlZddlmZmZmZmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ d Zd Zd
ej koTdk n Z ejdk Z e p_e Z G dd� de
�Z e dd�Z e dd�Z!e ee"ged f Z#e edged f Z$G dd� de�Z%de&de&fdd�Z'de&ddde"fdd �Z(e(Z)zddl*m+Z+ e+j)Z,es�e,Z)W dS W dS e-y� Y dS w ) z"Http related parsers and protocol.� N)�Any� Awaitable�Callable�Iterable�List�
NamedTuple�Optional�Union)� CIMultiDict� )�AbstractStreamWriter)� BaseProtocol)�ClientConnectionResetError)�ZLibCompressor)�
NO_EXTENSIONS)� StreamWriter� HttpVersion�
HttpVersion10�
HttpVersion11i )� �
r )r r � )r � � c @ s e Zd ZU eed< eed< dS )r �major�minorN)�__name__�
__module__� __qualname__�int�__annotations__� r! r! �>/usr/local/lib/python3.10/dist-packages/aiohttp/http_writer.pyr ' s
r �CIMultiDict[str]c @ sZ e Zd ZU dZee ed< dZeed< dZ eed< dZ
ee ed< d2de de
jd ed
ed df
d d
�Zed ee
j fdd��Zed e fdd��Zd3dd�Zdejfdeded dfdd�Zdee e e f d dfdd�Z d e e d dfd d �Z!d d d!�dee e e f d"ed#ed dfd$d%�Z"d&ed'd(d dfd)d*�Z#d3d+d,�Z$d4de d dfd.d/�Z%d3d0d1�Z&dS )5r N�lengthF�chunked�_eof� _compress�protocol�loop�
on_chunk_sent�on_headers_sent�returnc C s || _ || _|| _|| _d S �N)� _protocolr) �_on_chunk_sent�_on_headers_sent)�selfr( r) r* r+ r! r! r" �__init__; s
zStreamWriter.__init__c C s | j jS r- )r. � transport�r1 r! r! r" r3 G s zStreamWriter.transportc C s | j S r- )r. r4 r! r! r" r( K s zStreamWriter.protocolc C s
d| _ d S )NT)r% r4 r! r! r" �enable_chunkingO s
z StreamWriter.enable_chunking�deflate�encoding�strategyc C s t ||d�| _d S )N)r7 r8 )r r' )r1 r7 r8 r! r! r" �enable_compressionR s z StreamWriter.enable_compression�chunkc C sR t |�}| j|7 _| j|7 _| jj}|d u s |�� r"td��|�|� d S )N�!Cannot write to closing transport)�len� buffer_size� output_sizer. r3 �
is_closingr �write)r1 r: �sizer3 r! r! r" �_writeW s zStreamWriter._write�chunksc C s� d}|D ]}|t |�7 }q| j|7 _| j|7 _| jj}|d u s'|�� r+td��ts1|tk r;|� d�
|�� d S |� |� d S )Nr r; � ) r< r= r> r. r3 r? r �SKIP_WRITELINES�MIN_PAYLOAD_FOR_WRITELINESr@ �join�
writelines)r1 rC rA r: r3 r! r! r" � _writelines` s zStreamWriter._writelinesTi )�drain�LIMITrJ rK c � s �| j dur| � |�I dH t|t�r |jt|�kr |�d�}| jdur1| j�|�I dH }|s1dS | jdurTt|�}| j|krF| j| | _n|d| j� }d| _|sTdS |r�| j rk| �
t|�d�d�� d�|df� n| � |� | j
|kr�|r�d| _
| �� I dH dS dS dS 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�
�asciis
)r/ �
isinstance�
memoryview�nbytesr< �castr' �compressr$ r% rI �encoderB r= rJ )r1 r: rJ rK � chunk_lenr! r! r" r@ n s: �
�
�zStreamWriter.write� status_line�headersr# c � s4 �| j dur| � |�I dH t||�}| �|� dS )z*Write request/response status and headers.N)r0 �_serialize_headersrB )r1 rW rX �bufr! r! r" �
write_headers� s
�
zStreamWriter.write_headersc C s
d| _ dS )z&Indicate that the message is complete.TN)r& r4 r! r! r" �set_eof� s
zStreamWriter.set_eofrD c � sH �| j rd S |r| jd ur| �|�I d H | jrrg }d}|r2| j�|�I d H }r2t|�}|�|� | j�� }|t|�7 }|�|� |sFJ �| jr^|d�d��d�}| � |g|�d�R � n:t|�dkrj| � |� n.| �
|d � n&| jr�|r�t|�d�d��d�}| � ||df� n
| �
d� n|r�| �
|� | � � I d H d| _ d S ) Nr rM rN rO s
0
r s 0
T) r& r/ r' rT r< �append�flushr% rU rI rB rJ )r1 r: rC �
chunks_len�compressed_chunk� flush_chunk�
chunk_len_prer! r! r" � write_eof� s< �
zStreamWriter.write_eofc � s2 �| j }|jdur|jr|�� I dH dS dS dS )zFlush the write buffer.
The intended use is to write
await w.write(data)
await w.drain()
N)r. r3 �_paused�
_drain_helper)r1 r( r! r! r" rJ � s
��zStreamWriter.drain)NN)r, N)rD )'r r r r$ r r r r% �boolr& r' r r
�asyncio�AbstractEventLoop�_T_OnChunkSent�_T_OnHeadersSentr2 �property� Transportr3 r( r5 �zlib�Z_DEFAULT_STRATEGY�strr9 r �bytes� bytearrayrQ rB r rI r@ r[ r\ rc rJ r! r! r! r" r 4 sl
�����
�
���
� � ���
�0��
�
'r �stringr, c C s d| v sd| v r t d��| S )N�
�
zRNewline or carriage return detected in headers. Potential header injection attack.)�
ValueError)rr r! r! r" � _safe_header� s
�rv rW rX c C s2 dd� |� � D �}| d d�|� d }|�d�S )Nc s s( � | ]\}}t |�d t |� V qdS )z: N)rv )�.0�k�vr! r! r" � <genexpr>� s �& z(_py_serialize_headers.<locals>.<genexpr>rN z
zutf-8)�itemsrG rU )rW rX � headers_gen�liner! r! r" �_py_serialize_headers� s
r~ ).�__doc__rg �sysrm �typingr r r r r r r r � multidictr
�abcr �
base_protocolr
�client_exceptionsr �compression_utilsr �helpersr �__all__rF � version_info�IS_PY313_BEFORE_313_2�IS_PY_BEFORE_312_9rE r r r rp ri rj r ro rv r~ rY �aiohttp._http_writer� _http_writer�_c_serialize_headers� ImportErrorr! r! r! r" �<module> sF (
.
� �