File: //usr/local/lib/python3.10/dist-packages/aiohttp/__pycache__/streams.cpython-310.pyc
o
���g9W � @ s$ U d dl Z d dlZd dlZd dlmZmZmZmZmZm Z m
Z
m Z m Z ddl
mZ ddlmZmZmZmZmZ ddlmZ dZe d�ZG d d
� d
e�ZG d d � d ee �ZG d
d� d�Z G dd� d�Z G dd� de �Z G dd� de �Z e � Z ee e!d< G dd� dee �Z"G dd� de"e �Z#dS )� N) � Awaitable�Callable�Deque�Final�Generic�List�Optional�Tuple�TypeVar� )� BaseProtocol)�
_EXC_SENTINEL�BaseTimerContext� TimerNoop�
set_exception�
set_result)�internal_logger)�
EMPTY_PAYLOAD� EofStream� StreamReader� DataQueue�_Tc @ s e Zd ZdZdS )r zeof stream indication.N)�__name__�
__module__� __qualname__�__doc__� r r �:/usr/local/lib/python3.10/dist-packages/aiohttp/streams.pyr $ s r c @ sF e Zd ZdZdeg ee f ddfdd�Zd dd �Zdefd
d �Z dS )
�AsyncStreamIterator�� read_funcr �returnNc C �
|| _ d S �Nr )�selfr r r r �__init__, �
z AsyncStreamIterator.__init__�AsyncStreamIterator[_T]c C � | S r# r �r$ r r r � __aiter__/ � z AsyncStreamIterator.__aiter__c � s8 �z | � � I d H }W n ty t�w |dkrt�|S �N� )r r �StopAsyncIteration�r$ �rvr r r � __anext__2 s � �z AsyncStreamIterator.__anext__)r! r' )
r r r � __slots__r r r r% r* r1 r r r r r ( s
r c @ s: e Zd ZdZd dd�Zd
dd �Zdeeef fd
d �Z dS )� ChunkTupleAsyncStreamIterator��_stream�streamr r! Nc C r" r# r4 )r$ r6 r r r r% @ r& z&ChunkTupleAsyncStreamIterator.__init__c C r( r# r r) r r r r* C r+ z'ChunkTupleAsyncStreamIterator.__aiter__c � s" �| j �� I d H }|dkrt�|S )N�r- F)r5 � readchunkr. r/ r r r r1 F s
�z'ChunkTupleAsyncStreamIterator.__anext__)r6 r r! N)r! r3 )
r r r r2 r% r* r �bytes�boolr1 r r r r r3 < s
r3 c @ sX e Zd ZdZdee fdd�Zdedee fdd�Zdee fdd �Z de
fd
d �Z d S )
�AsyncStreamReaderMixinr r! c C �
t | j�S r# )r �readliner) r r r r* Q r& z AsyncStreamReaderMixin.__aiter__�nc s t � �fdd��S )z>Returns an asynchronous iterator that yields chunks of size n.c s
�� � �S r# )�readr �r> r$ r r �<lambda>V s
z5AsyncStreamReaderMixin.iter_chunked.<locals>.<lambda>)r �r$ r> r r@ r � iter_chunkedT s z#AsyncStreamReaderMixin.iter_chunkedc C r< )z3Yield all available data as soon as it is received.)r �readanyr) r r r �iter_anyX s
z AsyncStreamReaderMixin.iter_anyc C s t | �S )z�Yield chunks of data as they are received by the server.
The yielded objects are tuples
of (bytes, bool) as returned by the StreamReader.readchunk method.
)r3 r) r r r � iter_chunks\ s z"AsyncStreamReaderMixin.iter_chunksN) r r r r2 r r9 r* �intrC rE r3 rF r r r r r; M s r; c @ s� e Zd ZdZdZddd�dededee dee j
d df
d
d �Z d e fd d
�Z
d eeef fdd�Zd ee fdd�Zefdeded dfdd�Zdeg df d dfdd�ZdEdd�Zd efdd �Zd efd d �ZdEd d �Zd!ed dfd"d#�Z dFd!ed%ed dfd&d'�Z dEd(d)�Z dEd*d+�Z d,e d dfd-d.�Z d efd/d0�Z!dGd2ed efd3d4�Z"dHd6ed efd7d8�Z#d efd9d:�Z$d eeef fd;d<�Z%d6ed efd=d>�Z&dHd6ed efd?d@�Z'd6ed efdAdB�Z(d6ed efdCdD�Z)dS )Ir a* An enhancement of asyncio.StreamReader.
Supports asynchronous iteration by line, chunk or as available::
async for line in reader:
...
async for chunk in reader.iter_chunked(1024):
...
async for slice in reader.iter_any():
...
)� _protocol�
_low_water� _high_water�_loop�_size�_cursor�_http_chunk_splits�_buffer�_buffer_offset�_eof�_waiter� _eof_waiter�
_exception�_timer�_eof_callbacks� _eof_counter� total_bytesN)�timer�loop�protocol�limitrY rZ r! c C s� || _ || _|d | _|d u rt�� }|| _d| _d| _d | _t �
� | _ d| _ d| _
d | _d | _d | _|d u r:t� n|| _g | _d| _d| _d S )N� r F)rH rI rJ �asyncio�get_event_looprK rL rM rN � collections�dequerO rP rQ rR rS rT r rU rV rW rX )r$ r[ r\ rY rZ r r r r% � s&
zStreamReader.__init__c C s� | j jg}| jr|�d| j � | jr|�d� | jdkr(|�d| j| jf � | jr3|�d| j � | jr>|�d| j � dd� |� S ) Nz%d bytes�eofi zlow=%d high=%dzw=%rze=%r�<%s>� )
� __class__r rL �appendrQ rI rJ rR rT �join)r$ �infor r r �__repr__� s
zStreamReader.__repr__c C s | j | jfS r# )rI rJ r) r r r �get_read_buffer_limits� � z#StreamReader.get_read_buffer_limitsc C � | j S r# �rT r) r r r � exception� � zStreamReader.exception�exc� exc_causec C sX || _ | j�� | j}|d urd | _t|||� | j}|d ur*d | _t|||� d S d S r# )rT rV �clearrR r rS �r$ rp rq �waiterr r r r � s
�zStreamReader.set_exception�callbackc C sB | j rz|� W d S ty t�d� Y d S w | j�|� d S �N�Exception in eof callback)rQ � Exceptionr rn rV rf �r$ ru r r r �on_eof� s �zStreamReader.on_eofc C s� d| _ | j}|d urd | _t|d � | j}|d ur!d | _t|d � | jjr*| j�� | jD ]}z|� W q- tyB t �
d� Y q-w | j� � d S )NTrw ) rQ rR r rS rH �_reading_paused�resume_readingrV rx r rn rr )r$ rt �cbr r r �feed_eof� s$
�zStreamReader.feed_eofc C rl )z&Return True if 'feed_eof' was called.�rQ r) r r r �is_eof� s zStreamReader.is_eofc C � | j o| j S )z=Return True if the buffer is empty and 'feed_eof' was called.�rQ rO r) r r r �at_eof� s zStreamReader.at_eofc � sH �| j rd S | jd u s
J �| j�� | _z | jI d H W d | _d S d | _w r# )rQ rS rK �
create_futurer) r r r �wait_eof� s � zStreamReader.wait_eof�datac C sx t jdtdd� |s dS | jr | jd | jd� | jd< d| _| jt|�7 _| jt|�8 _| j�|� d| _ dS )zDrollback reading some data from stream, inserting it to buffer head.zJunread_data() is deprecated and will be removed in future releases (#3260)r] )�
stacklevelNr )
�warnings�warn�DeprecationWarningrP rO rL �lenrM �
appendleftrW )r$ r� r r r � unread_data� s �
zStreamReader.unread_datar �sizec C s� | j rJ d��|s d S t|�}| j|7 _| j�|� | j|7 _| j}|d ur2d | _t|d � | j| jkrC| j j
sE| j � � d S d S d S )Nzfeed_data after feed_eof) rQ r� rL rO rf rX rR r rJ rH r{ �
pause_reading)r$ r� r� �data_lenrt r r r � feed_data s
�zStreamReader.feed_datac C s& | j d u r| jr td��g | _ d S d S )Nz@Called begin_http_chunk_receiving when some data was already fed)rN rX � RuntimeErrorr) r r r �begin_http_chunk_receiving$ s
�
�z'StreamReader.begin_http_chunk_receivingc C sh | j d u r td��| j r| j d nd}| j|krd S | j �| j� | j}|d ur2d | _t|d � d S d S )NzFCalled end_chunk_receiving without calling begin_chunk_receiving first���r )rN r� rX rf rR r )r$ �posrt r r r �end_http_chunk_receiving, s
�
�z%StreamReader.end_http_chunk_receiving� func_namec � s� �| j js td��| jd urtd| ��| j�� }| _z$| j�
|I d H W d � n1 s0w Y W d | _d S W d | _d S d | _w )NzConnection closed.zH%s() called while another coroutine is already waiting for incoming data)rH � connectedr� rR rK r� rU )r$ r� rt r r r �_waitJ s" �
�� �
�zStreamReader._waitc � s �| � � I d H S r# )� readuntilr) r r r r= _ s �zStreamReader.readline�
� separatorc � s� �t |�}|dkr
td��| jd ur| j�d}d}d}|rm| jr[|r[| j}| jd �||�d }| �|r<|| | d nd�}||7 }|t |�7 }|rMd}|| jkrVtd��| jr[|s"| jra |S |rk| � d �I d H |s |S )
Nr z,Separator should be at least one-byte stringr- Tr r� Fz
Chunk too bigr� )
r� �
ValueErrorrT rO rP �find�_read_nowait_chunkrJ rQ r� ) r$ r� �seplen�chunk�
chunk_size�
not_enough�offset�icharr� r r r r� b s< �
�
���zStreamReader.readuntilr� r> c � s� �| j d ur | j � | jr%| js%t| dd�d | _| jdkr%tjddd� |s)dS |dk rEg } | �� I d H }|s:n|�|� q0d� |�S | jsY| jsY| �
d �I d H | jsY| jrK| � |�S )
NTrW r r � zEMultiple access to StreamReader in eof state, might be infinite loop.)�
stack_infor- r? ) rT rQ rO �getattrrW r �warningrD rf rg r� � _read_nowait)r$ r> �blocks�blockr r r r? � s4 �
�
�
�
zStreamReader.readc � sD �| j d ur | j �| js | js | �d�I d H | js | jr| �d�S )NrD r� )rT rO rQ r� r� r) r r r rD � s �
�
zStreamReader.readanyc � s� � | j dur
| j �| jr1| j�d�}|| jkrdS || jkr)| �|| j �dfS t�d� | js
| jr;| �d�dfS | j r@dS | �
d �I dH q)
a Returns a tuple of (data, end_of_http_chunk).
When chunked transfer
encoding is used, end_of_http_chunk is a boolean indicating if the end
of the data corresponds to the end of a HTTP chunk , otherwise it is
always False.
TNr �r- TzESkipping HTTP chunk end due to data consumption beyond chunk boundaryr� Fr7 r8 ) rT rN �poprM r� r r� rO r� rQ r� )r$ r� r r r r8 � s( �
�� �zStreamReader.readchunkc � sx �| j d ur | j �g }|dkr7| �|�I d H }|s(d�|�}t�|t|�| ��|�|� |t|�8 }|dksd�|�S )Nr r- )rT r? rg r^ �IncompleteReadErrorr� rf )r$ r> r� r� �partialr r r � readexactly� s �
�
zStreamReader.readexactlyc C s2 | j d ur| j �| jr| j�� std��| �|�S )Nz9Called while some coroutine is waiting for incoming data.)rT rR �doner� r� rB r r r � read_nowait� s
�
zStreamReader.read_nowaitc C s� | j d }| j}|dkr$t|�| |kr$|||| � }| j|7 _n|r5| j �� ||d � }d| _n| j �� }t|�}| j|8 _| j|7 _| j}|rf|d | jk rf|�d� |rf|d | jk sX| j| jk ru| j j
ru| j � � |S )Nr r� ) rO rP r� �popleftrL rM rN r� rI rH r{ r| )r$ r> � first_bufferr� r� r� � chunk_splitsr r r r� � s(
�
z StreamReader._read_nowait_chunkc C s^ | j �� g }| jr&| �|�}|�|� |dkr#|t|�8 }|dkr#n| js
|r-d�|�S dS )z6Read not more than n bytes, or whole buffer if n == -1r� r r- )rU �assert_timeoutrO r� rf r� rg )r$ r> �chunksr� r r r r� s
�zStreamReader._read_nowait�r! N�r )r� �r� )*r r r r r2 r rG r r r^ �AbstractEventLoopr% �strri r rj �
BaseExceptionrn r
r r rz r~ r: r� r� r� r9 r� r� r� r� r� r= r� r? rD r8 r� r� r� r� r r r r r e s\
�����
� ���
�
$* " r c @ s" e Zd ZdZd+dd�Zdefdd�Zdee fdd �Z e
fd
ed eddfd d
�Z de g df ddfdd�Z
d+dd�Zdefdd�Zdefdd�Zd+dd�Zd,dededdfd d �Zdefd d �Zd-dedefd!d"�Zdefd#d$�Zdeeef fd%d&�Zdedefd'd(�Zd-dedefd)d*�Z dS ).�EmptyStreamReader��_read_eof_chunkr! Nc C s d| _ d| _d S )NFr )r� rX r) r r r r% , s
zEmptyStreamReader.__init__c C s d| j j S )Nrc )re r r) r r r ri 0 rk zEmptyStreamReader.__repr__c C � d S r# r r) r r r rn 3 r+ zEmptyStreamReader.exceptionrp rq c C r� r# r )r$ rp rq r r r r 6 s z EmptyStreamReader.set_exceptionru c C s, z|� W d S t y t�d� Y d S w rv )rx r rn ry r r r rz = s
�zEmptyStreamReader.on_eofc C r� r# r r) r r r r~ C r+ zEmptyStreamReader.feed_eofc C � dS �NTr r) r r r r� F r+ zEmptyStreamReader.is_eofc C r� r� r r) r r r r� I r+ zEmptyStreamReader.at_eofc � s �d S r# r r) r r r r� L � �zEmptyStreamReader.wait_eofr r� r> c C r� r# r )r$ r� r> r r r r� O r+ zEmptyStreamReader.feed_datac � � �dS r, r r) r r r r= R r� zEmptyStreamReader.readliner� c � r� r, r rB r r r r? U r� zEmptyStreamReader.readc � r� r, r r) r r r rD Z r� zEmptyStreamReader.readanyc � s �| j s d| _ dS dS )NTr7 r� r� r) r r r r8 ] s
�zEmptyStreamReader.readchunkc � s �t �d|��r, )r^ r� rB r r r r� d s � z EmptyStreamReader.readexactlyc C r� r, r rB r r r r� g r+ z EmptyStreamReader.read_nowaitr� r� r� ) r r r r2 r% r� ri r r� rn r
r r rz r~ r: r� r� r� r9 rG r� r= r? rD r r8 r� r� r r r r r� ( s2
���
�
r� r c @ s� e Zd ZdZdejddfdd�Zdefdd�Zde fd d
�Z
de fd d �Z de e
fd
d�Zefde
de
ddfdd�Zd dededdfdd�Zd dd�Zdefdd�Zdee fd d �ZdS ) r z>DataQueue is a general-purpose blocking queue with one reader.rZ r! Nc C s& || _ d| _d | _d | _t�� | _d S )NF)rK rQ rR rT r` ra rO )r$ rZ r r r r% q s
zDataQueue.__init__c C r< r# )r� rO r) r r r �__len__x r& zDataQueue.__len__c C rl r# r r) r r r r� { ro zDataQueue.is_eofc C r� r# r� r) r r r r� ~ s zDataQueue.at_eofc C rl r# rm r) r r r rn � ro zDataQueue.exceptionrp rq c C s4 d| _ || _| j }d urd | _t|||� d S d S r� )rQ rT rR r rs r r r r � s �zDataQueue.set_exceptionr r� r� c C s6 | j �||f� | j }d urd | _t|d � d S d S r# )rO rf rR r )r$ r� r� rt r r r r� � s
�zDataQueue.feed_datac C s, d| _ | j }d urd | _t|d � d S d S r� )rQ rR r )r$ rt r r r r~ � s
�zDataQueue.feed_eofc � s� �| j s*| js*| jr J �| j�� | _z| jI d H W n tjtjfy) d | _� w | j r6| j �� \}}|S | j d ur>| j �t
�r# ) rO rQ rR rK r� r^ �CancelledError� TimeoutErrorr� rT r )r$ r� �_r r r r? � s �
�
zDataQueue.readc C r< r# )r r? r) r r r r* � r& zDataQueue.__aiter__r� r� )r r r r r^ r� r% rG r� r: r� r� r r� rn r
r r r� r~ r? r r* r r r r r n s& ���
�
r c s^ e Zd ZdZdededejddf� fdd� Zdd
e d eddf� fd d
�
Z
de fdd�Z � Z S )�FlowControlDataQueuez�FlowControlDataQueue resumes and pauses an underlying stream.
It is a destination for parsed data.
This class is deprecated and will be removed in version 4.0.
r[ r\ rZ r! Nc s( t � j|d� d| _|| _|d | _d S )N)rZ r r] )�superr% rL rH �_limit)r$ r[ r\ rZ �re r r r% � s z FlowControlDataQueue.__init__r r� r� c sF t � �||� | j|7 _| j| jkr | jjs!| j�� d S d S d S r# )r� r� rL r� rH r{ r� �r$ r� r� r� r r r� � s
�z FlowControlDataQueue.feed_datac � s� �| j s*| js*| jr J �| j�� | _z| jI d H W n tjtjfy) d | _� w | j rL| j �� \}}| j |8 _ | j | j
k rJ| j j rJ| j �
� |S | jd urT| j�t�r# )rO rQ rR rK r� r^ r� r� r� rL r� rH r{ r| rT r r� r r r r? � s& �
�
zFlowControlDataQueue.readr� )
r r r r r rG r^ r� r% r r� r? �
__classcell__r r r� r r� � s ���� r� )$r^ r` r� �typingr r r r r r r r r
�
base_protocolr �helpersr
r r r r �logr �__all__r rx r r r3 r; r r� r �__annotations__r r� r r r r �<module> s*
, FCA