File: //home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/__pycache__/streams.cpython-310.pyc
o
+wedQ � @ s U d dl Z d dlZd dlZd dlmZmZmZmZmZm Z m
Z
m Z m Z ddl
mZ ddlmZmZmZmZ ddlmZ dZe d�ZG d d
� d
e�ZG d d � d ee �ZG d
d� d�ZG 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)�BaseTimerContext� TimerNoop�
set_exception�
set_result)�internal_logger)�
EMPTY_PAYLOAD� EofStream� StreamReader� DataQueue�FlowControlDataQueue�_Tc @ s e Zd ZdZdS )r zeof stream indication.N)�__name__�
__module__� __qualname__�__doc__� r r �H/home/arjun/projects/env/lib/python3.10/site-packages/aiohttp/streams.pyr s r c @ sB e Zd Zdeg ee f ddfdd�Zd dd�Zdefd d
�ZdS ) �AsyncStreamIterator� read_func�returnNc C �
|| _ d S �N)r )�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__* s � �z AsyncStreamIterator.__anext__)r r& ) r r r r r r r$ r) r0 r r r r r # s
r c @ s6 e Zd Zd dd�Zd dd�Zdeeef fd d
�ZdS )
� ChunkTupleAsyncStreamIterator�streamr r Nc C r! r" )�_stream)r# r2 r r r r$ 5 r% z&ChunkTupleAsyncStreamIterator.__init__c C r' r" r r( r r r r) 8 r* z'ChunkTupleAsyncStreamIterator.__aiter__c � s" �| j �� I d H }|dkrt�|S )N�r, F)r3 � readchunkr- r. r r r r0 ; s
�z'ChunkTupleAsyncStreamIterator.__anext__)r2 r r N)r r1 ) r r r r$ r) r �bytes�boolr0 r r r r r1 4 s
r1 c @ sT e Zd Zdee fdd�Zdedee fdd�Zdee fdd�Zde fd d
�Z
d S ) �AsyncStreamReaderMixinr c C �
t | j�S r" )r �readliner( r r r r) C 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>H s
z5AsyncStreamReaderMixin.iter_chunked.<locals>.<lambda>)r �r# r; r r= r � iter_chunkedF s z#AsyncStreamReaderMixin.iter_chunkedc C r9 )z3Yield all available data as soon as it is received.)r �readanyr( r r r �iter_anyJ 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.
)r1 r( r r r � iter_chunksN s z"AsyncStreamReaderMixin.iter_chunksN) r r r r r6 r) �intr@ rB r1 rC r r r r r8 B s
r8 c @ s� e Zd ZdZdZddd�dededee dee j
d df
d
d �Z d e fd d
�Z
d eeef fdd�Zd ee fdd�Zded dfdd�Zdeg df d dfdd�ZdCdd�Zd efdd�Zd efd d �ZdCd d �Zd ed dfd!d"�ZdDd ed#ed dfd$d%�Z dCd&d'�Z dCd(d)�Z d*e d dfd+d,�Z d efd-d.�Z dEd0ed efd1d2�Z!dFd4ed efd5d6�Z"d efd7d8�Z#d eeef fd9d:�Z$d4ed efd;d<�Z%dFd4ed efd=d>�Z&d4ed efd?d@�Z'd4ed efdAdB�Z(dS )Gr 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():
...
r N)�timer�loop�protocol�limitrE rF r c C s� || _ || _|d | _|d u rt�� }|| _d| _d| _d | _t �
� | _ d| _ d| _
d | _d | _d | _|d u r:t� n|| _g | _d S )N� r F)� _protocol�
_low_water� _high_water�asyncio�get_event_loop�_loop�_size�_cursor�_http_chunk_splits� collections�deque�_buffer�_buffer_offset�_eof�_waiter� _eof_waiter�
_exceptionr �_timer�_eof_callbacks)r# rG rH rE rF r r r r$ g 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 rP �appendrW rK rL rX rZ �join)r# �infor r r �__repr__� s
zStreamReader.__repr__c C s | j | jfS r" )rK rL r( r r r �get_read_buffer_limits� � z#StreamReader.get_read_buffer_limitsc C � | j S r" �rZ r( r r r � exception� � zStreamReader.exception�excc C sT || _ | j�� | j}|d urd | _t||� | j}|d ur(d | _t||� d S d S r" )rZ r\ �clearrX r rY �r# rk �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)rW � Exceptionr ri r\ ra �r# ro r r r �on_eof� s �zStreamReader.on_eofc C s� d| _ | j}|d urd | _t|d � | j}|d ur!d | _t|d � | jD ]}z|� W q$ ty9 t�d� Y q$w | j�� d S )NTrq ) rW rX r rY r\ rr r ri rl )r# rn �cbr r r �feed_eof� s
�zStreamReader.feed_eofc C rg )z&Return True if 'feed_eof' was called.�rW 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.�rW rU 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" )rW rY rO �
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)rI )�
stacklevelNr )
�warnings�warn�DeprecationWarningrV rU rP �lenrQ �
appendleft� _eof_counter)r# r~ r r r � unread_data� s �
zStreamReader.unread_data�sizec C s� | j rJ d��|s d S | jt|�7 _| j�|� | jt|�7 _| j}|d ur2d | _t|d � | j| jkrC| j j
sE| j � � d S d S d S )Nzfeed_data after feed_eof) rW rP r� rU ra � total_bytesrX r rL rJ �_reading_paused�
pause_reading�r# r~ r� rn 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 whensome data was already fed)rR r� � 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 )rR r� r� ra rX r )r# �posrn r r r �end_http_chunk_receiving s
�
�z%StreamReader.end_http_chunk_receiving� func_namec � sz �| j d ur td| ��| j�� }| _ z$| j�
|I d H W d � n1 s(w Y W d | _ d S W d | _ d S d | _ w )NzH%s() called while another coroutine is already waiting for incoming data)rX r� rO r| r[ )r# r� rn r r r �_wait s �
�� �
�zStreamReader._waitc � s �| � � I d H S r" )� readuntilr( r r r r: 2 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� �
ValueErrorrZ rU rV �find�_read_nowait_chunkrL rW r� ) r# r� �seplen�chunk�
chunk_size�
not_enough�offset�icharr~ r r r r� 5 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 )
NTr� r r � zEMultiple access to StreamReader in eof state, might be infinite loop.)�
stack_infor, r< ) rZ rW rU �getattrr� r �warningrA ra rb r� � _read_nowait)r# r; �blocks�blockr r r r< Y s4 �
�
�
�
zStreamReader.readc � sD �| j d ur | j �| js | js | �d�I d H | js | jr| �d�S )NrA r� )rZ rU rW r� r� r( r r r rA � 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� Fr4 r5 ) rZ rR �poprQ r� r r� rU r� rW r� )r# r� r r r r5 � s( �
�� �zStreamReader.readchunkc � sx �| j d ur | j �g }|dkr7| �|�I d H }|s(d�|�}t�|t|�| ��|�|� |t|�8 }|dksd�|�S )Nr r, )rZ r< rb rM �IncompleteReadErrorr� ra )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.)rZ rX �doner� r� r? 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 �� }| jt|�8 _| jt|�7 _| j}|rf|d | jk rf|�d� |rf|d | jk sX| j| jk ru| j j
ru| j � � |S )Nr r� ) rU rV r� �popleftrP rQ rR r� rK rJ r� �resume_reading)r# r; � first_bufferr� 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, )r[ �assert_timeoutrU r� ra r� rb )r# r; �chunksr� r r r r� � s
�zStreamReader._read_nowait�r N�r )r� �r� ))r r r r r� r rD r r
rM �AbstractEventLoopr$ �strrd r re �
BaseExceptionri r r rt rv r7 rx r{ r} r6 r� r� r� r� r� r: r� r< rA r5 r� r� r� r� r r r r r W sN
�����
�
$* "r c @ s e Zd Zd)dd�Zdefdd�Zdee fdd�Zd eddfd
d �Z d e
g df ddfd
d�Z d)dd�Z de
fdd�Zde
fdd�Zd)dd�Zd*dededdfdd�Zdefd d �Zd+dedefd d �Zdefd!d"�Zdeee
f fd#d$�Zdedefd%d&�Zd+dedefd'd(�ZdS ),�EmptyStreamReaderr Nc C s
d| _ d S )NF��_read_eof_chunkr( r r r r$ � r% zEmptyStreamReader.__init__c C s d| j j S )Nr^ )r` r r( r r r rd � rf zEmptyStreamReader.__repr__c C � d S r" r r( r r r ri r* zEmptyStreamReader.exceptionrk c C r� r" r )r# rk r r r r r* z EmptyStreamReader.set_exceptionro c C s, z|� W d S t y t�d� Y d S w rp )rr r ri rs r r r rt s
�zEmptyStreamReader.on_eofc C r� r" r r( r r r rv
r* zEmptyStreamReader.feed_eofc C � dS �NTr r( r r r rx r* zEmptyStreamReader.is_eofc C r� r� r r( r r r r{ r* zEmptyStreamReader.at_eofc � s �d S r" r r( r r r r} � �zEmptyStreamReader.wait_eofr r~ r; c C r� r" r )r# r~ r; r r r r� r* zEmptyStreamReader.feed_datac � � �dS r+ r r( r r r r: r� zEmptyStreamReader.readliner� c � r� r+ r r? r r r r< r� zEmptyStreamReader.readc � r� r+ r r( r r r rA $ r� zEmptyStreamReader.readanyc � s �| j s d| _ dS dS )NTr4 r� r� r( r r r r5 ' s
�zEmptyStreamReader.readchunkc � s �t �d|��r+ )rM r� r? r r r r� . s � z EmptyStreamReader.readexactlyc C r� r+ r r? r r r r� 1 r* z EmptyStreamReader.read_nowaitr� r� r� )r r r r$ r� rd r r� ri r r rt rv r7 rx r{ r} r6 rD r� r: r< rA r r5 r� r� r r r r r� � s"
r� r c @ s� e Zd ZdZdejddfdd�Zdefdd�Zde fd d
�Z
de fd d �Z de e
fd
d�Zde
ddfdd�Zd dededdfdd�Zd dd�Zdefdd�Zdee fdd �ZdS ) r z>DataQueue is a general-purpose blocking queue with one reader.rF r Nc C s, || _ d| _d | _d | _d| _t�� | _d S )NFr )rO rW rX rZ rP rS rT rU )r# rF r r r r$ ; s zDataQueue.__init__c C r9 r" )r� rU r( r r r �__len__C r% zDataQueue.__len__c C rg r" rw r( r r r rx F rj zDataQueue.is_eofc C ry r" rz r( r r r r{ I s zDataQueue.at_eofc C rg r" rh r( r r r ri L rj zDataQueue.exceptionrk c C s2 d| _ || _| j}|d urd | _t||� d S d S r� )rW rZ rX r rm r r r r O s �zDataQueue.set_exceptionr r~ r� c C sD | j |7 _ | j�||f� | j}|d ur d | _t|d � d S d S r" )rP rU ra rX r r� r r r r� X s �zDataQueue.feed_datac C s, d| _ | j}|d urd | _t|d � d S d S r� )rW rX r )r# rn r r r rv a s �zDataQueue.feed_eofc � s� �| j s*| js*| jr J �| j�� | _z| jI d H W n tjtjfy) d | _� w | j r=| j �� \}}| j |8 _ |S | j
d urE| j
�t �r" ) rU rW rX rO r| rM �CancelledError� TimeoutErrorr� rP rZ r �r# r~ r� r r r r< i s" �
�
zDataQueue.readc C r9 r" )r r< r( r r r r) } r% zDataQueue.__aiter__r� r� )r r r r rM r� r$ rD r� r7 rx r{ r r� ri r r r� rv r< r r) r r r r r 8 s
r c sb e Zd ZdZdededejddf� fdd� Zdd
e d eddf� fd d
�
Z
de f� fdd� Z � Z S )r zlFlowControlDataQueue resumes and pauses an underlying stream.
It is a destination for parsed data.
rG rH rF r Nc s" t � j|d� || _|d | _d S )N)rF rI )�superr$ rJ �_limit)r# rG rH rF �r` r r r$ � s z FlowControlDataQueue.__init__r r~ r� c s8 t � �||� | j| jkr| jjs| j�� d S d S d S r" )r� r� rP r� rJ r� r� r� r� r r r� � s �z FlowControlDataQueue.feed_datac � s\ �zt � �� I d H W | j| jk r| jjr| j�� S S S | j| jk r,| jjr-| j�� w w w r" )r� r< rP r� rJ r� r� r( r� r r r< � s � � �zFlowControlDataQueue.readr� )
r r r r r rD rM r� r$ r r� r< �
__classcell__r r r� r r � s ���� r )#rM rS r� �typingr r r r r r r r r
�
base_protocolr �helpersr
r r r �logr �__all__r rr r r r1 r8 r r� r �__annotations__r r r r r r �<module> s*
, &;I