o
    `gV                     @   s$  U d dl Z d dlZd dlZd dlmZmZmZmZmZm	Z	m
Z
mZmZ ddlmZ ddlmZmZmZmZmZ ddlmZ dZedZG d	d
 d
eZG dd dee ZG dd dZG dd dZG dd deZG dd deZe Z ee e!d< G dd dee Z"G dd de"e Z#dS )    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 Zd ZdZdS )r   zeof stream indication.N)__name__
__module____qualname____doc__ r   r   J/var/www/html/propose/venv/lib/python3.10/site-packages/aiohttp/streams.pyr   $   s    r   c                   @   sF   e Zd ZdZdeg ee f ddfddZddd	Zdefd
dZ	dS )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	 ty   tw |dkrt|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 Zd ZdZdddZddd	Zdeeef fd
dZ	dS )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 }|dkrt|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 Zd ZdZdee fddZdedee fddZdee fdd	Z	de
fd
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  fdd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 )zYield 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 Zd ZdZdZddddededee dee	j
 d	df
d
dZd	efddZd	eeef fddZd	ee fddZefdeded	dfddZdeg df d	dfddZdEddZd	efddZd	efddZdEdd Zd!ed	dfd"d#ZdFd!ed%ed	dfd&d'ZdEd(d)ZdEd*d+Zd,ed	dfd-d.Z d	efd/d0Z!dGd2ed	efd3d4Z"dHd6ed	efd7d8Z#d	efd9d:Z$d	eeef fd;d<Z%d6ed	efd=d>Z&dHd6ed	efd?d@Z'd6ed	efdAdBZ(d6ed	efdCdDZ)dS )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 rt }|| _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&   


zStreamReader.__init__c                 C   s   | j jg}| jr|d| j  | jr|d | jdkr(|d| j| jf  | jr3|d| j  | jr>|d| j  dd	| S )	Nz%d byteseofi   zlow=%d high=%dzw=%rze=%r<%s> )
	__class__r   rL   appendrQ   rI   rJ   rR   rT   join)r$   infor   r   r   __repr__   s   


zStreamReader.__repr__c                 C   s   | j | jfS 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      zStreamReader.exceptionexc	exc_causec                 C   sX   || _ | j  | j}|d urd | _t||| | j}|d ur*d | _t||| d S d S r#   )rT   rV   clearrR   r   rS   r$   rp   rq   waiterr   r   r   r      s   
zStreamReader.set_exceptioncallbackc                 C   sB   | j rz|  W d S  ty   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   zStreamReader.on_eofc              	   C   s   d| _ | j}|d urd | _t|d  | j}|d ur!d | _t|d  | jD ]}z|  W q$ ty9   td Y q$w | j  d S )NTrw   )	rQ   rR   r   rS   rV   rx   r   rn   rr   )r$   rt   cbr   r   r   feed_eof   s    



zStreamReader.feed_eofc                 C   rl   )z&Return True if  'feed_eof' was called.rQ   r)   r   r   r   is_eof   s   zStreamReader.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   zStreamReader.at_eofc                    sH   | j rd S | jd u sJ | j | _z| jI d H  W d | _d S d | _w r#   )rQ   rS   rK   create_futurer)   r   r   r   wait_eof   s   zStreamReader.wait_eofdatac                 C   sx   t jdtdd |sdS | jr| jd | jd | jd< d| _|  jt|7  _|  jt|8  _| j| d| _	dS )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   
zStreamReader.unread_datar   sizec                 C   s   | j rJ d|sd S t|}|  j|7  _| j| |  j|7  _| j}|d ur2d | _t|d  | j| jkrC| j	j
sE| j	  d S d S d S )Nzfeed_data after feed_eof)rQ   r   rL   rO   rf   rX   rR   r   rJ   rH   _reading_pausedpause_reading)r$   r   r   data_lenrt   r   r   r   	feed_data  s   
zStreamReader.feed_datac                 C   s&   | j d u r| jrtd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	td| j r| j d nd}| j|krd S | j | j | j}|d ur2d | _t|d  d S d S )NzFCalled end_chunk_receiving without calling begin_chunk_receiving firstr   )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 js	td| jd urtd| | j  }| _z$| j |I d H  W d    n1 s0w   Y  W d | _d S W d | _d S d | _w )NzConnection 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   _waitG  s"   

zStreamReader._waitc                    s   |   I d H S r#   )	readuntilr)   r   r   r   r=   \  s   zStreamReader.readline   
	separatorc           	         s   t |}|dkrtd| jd ur| jd}d}d}|rm| jr[|r[| j}| jd ||d }| |r<|| | d nd}||7 }|t |7 }|rMd}|| jkrVtd| jr[|s"| jra	 |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   _  s<   



zStreamReader.readuntilr   r>   c                    s   | j d ur	| j 	 | jr%| js%t| ddd | _| jdkr%tjddd |s)dS |dk rEg }	 |  I d H }|s:n|| q0d	|S | jsY| jsY| 
d	I d H  | jsY| jrK| |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   




zStreamReader.readc                    sD   | j d ur	| j | js| js| dI d H  | js| jr| dS )NrD   r   )rT   rO   rQ   r   r   r)   r   r   r   rD     s   

zStreamReader.readanyc                    s   	 | j dur
| j | jr1| jd}|| jkrdS || jkr)| || j dfS td | js| jr;| ddfS | j	r@dS | 
d	I dH  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(   


zStreamReader.readchunkc                    sx   | j d ur	| j g }|dkr7| |I d H }|s(d|}t|t|| || |t|8 }|dksd|S )Nr   r-   )rT   r?   rg   r^   IncompleteReadErrorr   rf   )r$   r>   r   r   partialr   r   r   readexactly  s   



zStreamReader.readexactlyc                 C   s2   | j d ur| j | jr| j std| |S )Nz9Called while some coroutine is waiting for incoming data.)rT   rR   doner   r   rB   r   r   r   read_nowait  s   

zStreamReader.read_nowaitc                 C   s   | j d }| j}|dkr$t|| |kr$||||  }|  j|7  _n|r5| j   ||d  }d| _n| j  }t|}|  j|8  _|  j|7  _| j}|rf|d | jk rf|d |rf|d | jk sX| j| jk ru| j	j
ru| j	  |S )Nr   r   )rO   rP   r   popleftrL   rM   rN   r   rI   rH   r   resume_reading)r$   r>   first_bufferr   r   r   chunk_splitsr   r   r   r     s(   




zStreamReader._read_nowait_chunkc                 C   s^   | j   g }| jr&| |}|| |dkr#|t|8 }|dkr#n| js
|r-d|S dS )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   


zStreamReader._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 Zd ZdZd+ddZdefddZdee fdd	Z	e
fd
ededdfddZdeg df ddfddZd+ddZdefddZdefddZd+ddZd,dededdfddZdefddZd-dedefd!d"Zdefd#d$Zdeeef fd%d&Zdedefd'd(Zd-dedefd)d*ZdS ).EmptyStreamReader_read_eof_chunkr!   Nc                 C   s
   d| _ d S NFr   r)   r   r   r   r%   )  r&   zEmptyStreamReader.__init__c                 C   s   d| j j S )Nrc   )re   r   r)   r   r   r   ri   ,  rk   zEmptyStreamReader.__repr__c                 C      d S r#   r   r)   r   r   r   rn   /  r+   zEmptyStreamReader.exceptionrp   rq   c                 C   r   r#   r   )r$   rp   rq   r   r   r   r   2  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   9  s
   zEmptyStreamReader.on_eofc                 C   r   r#   r   r)   r   r   r   r|   ?  r+   zEmptyStreamReader.feed_eofc                 C      dS NTr   r)   r   r   r   r~   B  r+   zEmptyStreamReader.is_eofc                 C   r   r   r   r)   r   r   r   r   E  r+   zEmptyStreamReader.at_eofc                    s   d S r#   r   r)   r   r   r   r   H     zEmptyStreamReader.wait_eofr   r   r>   c                 C   r   r#   r   )r$   r   r>   r   r   r   r   K  r+   zEmptyStreamReader.feed_datac                       dS r,   r   r)   r   r   r   r=   N  r   zEmptyStreamReader.readliner   c                    r   r,   r   rB   r   r   r   r?   Q  r   zEmptyStreamReader.readc                    r   r,   r   r)   r   r   r   rD   V  r   zEmptyStreamReader.readanyc                    s   | j s	d| _ dS dS )NTr7   r   r   r)   r   r   r   r8   Y  s
   zEmptyStreamReader.readchunkc                    s   t d|r,   )r^   r   rB   r   r   r   r   `  s   zEmptyStreamReader.readexactlyc                 C   r   r,   r   rB   r   r   r   r   c  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 Zd ZdZdejddfddZdefddZde	fd	d
Z
de	fddZdee fddZefdededdfddZddededdfddZdddZdefddZdee fddZdS ) r   z>DataQueue is a general-purpose blocking queue with one reader.rZ   r!   Nc                 C   s&   || _ d| _d | _d | _t | _d S r   )rK   rQ   rR   rT   r`   ra   rO   )r$   rZ   r   r   r   r%   m  s
   zDataQueue.__init__c                 C   r<   r#   )r   rO   r)   r   r   r   __len__t  r&   zDataQueue.__len__c                 C   rl   r#   r}   r)   r   r   r   r~   w  ro   zDataQueue.is_eofc                 C   r   r#   r   r)   r   r   r   r   z  s   zDataQueue.at_eofc                 C   rl   r#   rm   r)   r   r   r   rn   }  ro   zDataQueue.exceptionrp   rq   c                 C   s4   d| _ || _| j }d urd | _t||| d S d S r   )rQ   rT   rR   r   rs   r   r   r   r     s   zDataQueue.set_exceptionr   r   r   c                 C   s6   | j ||f | j }d urd | _t|d  d S d S r#   )rO   rf   rR   r   )r$   r   r   rt   r   r   r   r     s
   zDataQueue.feed_datac                 C   s,   d| _ | j }d urd | _t|d  d S d S r   )rQ   rR   r   )r$   rt   r   r   r   r|     s
   zDataQueue.feed_eofc              	      s   | j s*| js*| jrJ | j | _z| jI d H  W n tjtjfy)   d | _ w | j r6| j  \}}|S | j	d ur>| j	t
r#   )rO   rQ   rR   rK   r   r^   CancelledErrorTimeoutErrorr   rT   r   )r$   r   _r   r   r   r?     s    

zDataQueue.readc                 C   r<   r#   )r   r?   r)   r   r   r   r*     r&   zDataQueue.__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   j  s&    

r   c                       s^   e Zd ZdZdededejddf fddZdd
e	deddf fddZ
de	fddZ  ZS )FlowControlDataQueuezFlowControlDataQueue 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| jkr| jjs!| 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*| js*| jrJ | j | _z| jI d H  W n tjtjfy)   d | _ w | j rL| j  \}}|  j	|8  _	| j	| j
k rJ| jjrJ| j  |S | jd urT| 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&   


zFlowControlDataQueue.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*   
 ,   CBA