
    gW                     n   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
e      Z G d dee         Z G d d      Z G d d      Z G d de      Z G d de      Z e       Z ee   e!d<    G d dee         Z" G d de"e         Z#y)    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                       e Zd ZdZy)r   zeof stream indication.N)__name__
__module____qualname____doc__     D/var/www/openai/venv/lib/python3.12/site-packages/aiohttp/streams.pyr   r   $   s     r   r   c                   D    e Zd ZdZdeg ee   f   ddfdZddZdefdZ	y)	AsyncStreamIterator	read_funcr$   returnNc                     || _         y Nr#   )selfr$   s     r    __init__zAsyncStreamIterator.__init__,   s	    "r   c                     | S r'   r   r(   s    r    	__aiter__zAsyncStreamIterator.__aiter__/       r   c                    K   	 | j                          d {   }|dk(  rt        |S 7 # t        $ r t        w xY wwNr   )r$   r   StopAsyncIterationr(   rvs     r    	__anext__zAsyncStreamIterator.__anext__2   sG     	%~~''B 9$$	 ( 	%$$	%s    >+ )+ >+ ;>)r%   zAsyncStreamIterator[_T])
r   r   r   	__slots__r   r   r   r)   r,   r3   r   r   r    r"   r"   (   s9    I#(2y}+<"= #$ # r   r"   c                   6    e Zd ZdZddZddZdeeef   fdZ	y)	ChunkTupleAsyncStreamIterator_streamr%   Nc                     || _         y r'   r7   )r(   streams     r    r)   z&ChunkTupleAsyncStreamIterator.__init__@   s	    r   c                     | S r'   r   r+   s    r    r,   z'ChunkTupleAsyncStreamIterator.__aiter__C   r-   r   c                 l   K   | j                   j                          d {   }|dk(  rt        |S 7 w)Nr   F)r8   	readchunkr0   r1   s     r    r3   z'ChunkTupleAsyncStreamIterator.__anext__F   s3     <<))++$$	 ,s   424)r:   r   r%   N)r%   r6   )
r   r   r   r4   r)   r,   r
   bytesboolr3   r   r   r    r6   r6   <   s&    Iud{!3 r   r6   c                   V    e Zd ZdZdee   fdZdedee   fdZdee   fdZ	de
fdZy)	AsyncStreamReaderMixinr   r%   c                 ,    t        | j                        S r'   )r"   readliner+   s    r    r,   z AsyncStreamReaderMixin.__aiter__Q   s    "4==11r   nc                 $     t         fd      S )z>Returns an asynchronous iterator that yields chunks of size n.c                  &    j                         S r'   )read)rE   r(   s   r    <lambda>z5AsyncStreamReaderMixin.iter_chunked.<locals>.<lambda>V   s    499Q<r   )r"   r(   rE   s   ``r    iter_chunkedz#AsyncStreamReaderMixin.iter_chunkedT   s    "#788r   c                 ,    t        | j                        S )z3Yield all available data as soon as it is received.)r"   readanyr+   s    r    iter_anyzAsyncStreamReaderMixin.iter_anyX   s    "4<<00r   c                     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.
        )r6   r+   s    r    iter_chunksz"AsyncStreamReaderMixin.iter_chunks\   s     -T22r   N)r   r   r   r4   r"   r?   r,   intrK   rN   r6   rP   r   r   r    rB   rB   M   sQ    I2.u5 29c 9&9%&@ 91-e4 13: 3r   rB   c                      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
Zd	efdZd	eeef   fdZd	ee   fdZefdeded	dfdZdeg df   d	dfdZd*dZd	efdZd	efdZd*dZded	dfdZd+deded	dfdZd*dZd*dZded	dfdZ d	efdZ!d,d ed	efd!Z"d-d"ed	efd#Z#d	efd$Z$d	eeef   fd%Z%d"ed	efd&Z&d-d"ed	efd'Z'd"ed	efd(Z(d"ed	efd)Z)y).r   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limitrd   re   r%   c                `   || _         || _        |dz  | _        |t        j                         }|| _        d| _        d| _        d | _        t        j                         | _        d| _        d| _        d | _        d | _        d | _        |
t#               n|| _        g | _        d| _        d| _        y )N   r   F)rS   rT   rU   asyncioget_event_looprV   rW   rX   rY   collectionsdequerZ   r[   r\   r]   r^   r_   r   r`   ra   rb   rc   )r(   rf   rg   rd   re   s        r    r)   zStreamReader.__init__   s     " 19<))+D

7;%0%6%6%8	7;;?37%*]ik8:r   c                     | j                   j                  g}| j                  r|j                  d| j                  z         | j                  r|j                  d       | j
                  dk7  r*|j                  d| j
                  | j                  fz         | j                  r|j                  d| j                  z         | j                  r|j                  d| j                  z         ddj                  |      z  S )	Nz%d byteseofi   zlow=%d high=%dzw=%rze=%r<%s> )
	__class__r   rW   appendr\   rT   rU   r]   r_   join)r(   infos     r    __repr__zStreamReader.__repr__   s    ''(::KK
TZZ/099KK??e#KK(DOOT=M=M+NNO<<KK-.??KK01&&r   c                 2    | j                   | j                  fS r'   )rT   rU   r+   s    r    get_read_buffer_limitsz#StreamReader.get_read_buffer_limits   s    !1!122r   c                     | j                   S r'   r_   r+   s    r    	exceptionzStreamReader.exception       r   exc	exc_causec                     || _         | j                  j                          | j                  }|d | _        t	        |||       | j
                  }|d | _        t	        |||       y y r'   )r_   ra   clearr]   r   r^   r(   r}   r~   waiters       r    r   zStreamReader.set_exception   si    
 !!#DL&#y1!!#D&#y1 r   callbackc                     | j                   r		  |        y | j                  j                  |       y # t        $ r t        j                  d       Y y w xY wNException in eof callback)r\   	Exceptionr   r{   ra   rs   r(   r   s     r    on_eofzStreamReader.on_eof   sL    99G
 &&x0  G))*EFGs   2 AAc                    d| _         | j                  }|d | _        t        |d        | j                  }|d | _        t        |d        | j                  j
                  r| j                  j                          | j                  D ]
  }	  |         | j                  j                          y # t        $ r t        j                  d       Y Hw xY w)NTr   )r\   r]   r   r^   rS   _reading_pausedresume_readingra   r   r   r{   r   )r(   r   cbs      r    feed_eofzStreamReader.feed_eof   s    	DLvt$!!#Dvt$>>))NN))+%%BG & 	!!#  G))*EFGs   
B..CCc                     | j                   S )z&Return True if  'feed_eof' was called.r\   r+   s    r    is_eofzStreamReader.is_eof   s    yyr   c                 8    | j                   xr | j                   S )z=Return True if the buffer is empty and 'feed_eof' was called.r\   rZ   r+   s    r    at_eofzStreamReader.at_eof   s    yy---r   c                    K   | j                   ry | j                  J | j                  j                         | _        	 | j                   d {    d | _        y 7 # d | _        w xY wwr'   )r\   r^   rV   create_futurer+   s    r    wait_eofzStreamReader.wait_eof   sa     99'''::335	$""""#D ##Ds.   ;A(A AA A(A 	A%%A(datac                 v   t        j                  dt        d       |sy| j                  r0| j                  d   | j                  d | j                  d<   d| _        | xj
                  t        |      z  c_        | xj                  t        |      z  c_        | j                  j                  |       d| _	        y)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DeprecationWarningr[   rZ   rW   lenrX   
appendleftrb   )r(   r   s     r    unread_datazStreamReader.unread_data   s    =		
 "ll1od.A.A.CDDLLO"#D

c$i
D	!%r   sizec                    | j                   rJ d       |sy t        |      }| xj                  |z  c_        | j                  j	                  |       | xj
                  |z  c_        | j                  }|d | _        t        |d        | j                  | j                  kD  r2| j                  j                  s| j                  j                          y y y )Nzfeed_data after feed_eof)r\   r   rW   rZ   rs   rc   r]   r   rU   rS   r   pause_reading)r(   r   r   data_lenr   s        r    	feed_datazStreamReader.feed_data  s    99888}t9

h
D!H$DLvt$::(((1O1ONN((* 2P(r   c                 Z    | j                   | j                  rt        d      g | _         y y )Nz@Called begin_http_chunk_receiving when some data was already fed)rY   rc   RuntimeErrorr+   s    r    begin_http_chunk_receivingz'StreamReader.begin_http_chunk_receiving$  s6    ""*"V  ')D# +r   c                    | j                   t        d      | j                   r| j                   d   nd}| j                  |k(  ry | j                   j                  | j                         | j                  }|d | _        t        |d        y y )NzFCalled end_chunk_receiving without calling begin_chunk_receiving firstr   )rY   r   rc   rs   r]   r   )r(   posr   s      r    end_http_chunk_receivingz%StreamReader.end_http_chunk_receiving,  s    ""*.  .2-D-Dd%%b)!s" &&t'7'78 DLvt$ r   	func_namec                 F  K   | j                   j                  st        d      | j                  t        d|z        | j                  j                         x}| _        	 | j                  5  | d {    d d d        d | _        y 7 # 1 sw Y   xY w# d | _        w xY ww)NzConnection closed.zH%s() called while another coroutine is already waiting for incoming data)rS   	connectedr   r]   rV   r   r`   )r(   r   r   s      r    _waitzStreamReader._waitJ  s     ~~''344 <<#46?@ 
 !%

 8 8 ::	    DL    DLsH   AB! B ,B	2B3B	7B ?B!B		BB 	BB!c                 >   K   | j                          d {   S 7 wr'   )	readuntilr+   s    r    rD   zStreamReader.readline_  s     ^^%%%%s   	separatorc                   K   t        |      }|dk(  rt        d      | j                  | j                  d}d}d}|r| j                  r|r| j                  }| j                  d   j                  ||      dz   }| j                  |r||z
  |z   dz
  nd      }||z  }|t        |      z  }|rd}|| j                  kD  rt        d      | j                  r|r| j                  r	 |S |r| j                  d	       d {    |r|S 7 	w)
Nr   z,Separator should be at least one-byte stringr   Tr   r   FzChunk too bigr   )
r   
ValueErrorr_   rZ   r[   find_read_nowait_chunkrU   r\   r   )	r(   r   seplenchunk
chunk_size
not_enoughoffseticharr   s	            r    r   zStreamReader.readuntilb  s     YQ;KLL??&//!

,,:,,Q,,Y?!C..38EFNV+a/b c$i'
!&J 0 00$_55 ,,: yy
  jj---) ,  .s   CD	%D	=D>D	D	rE   c                 2  K   | j                   | j                   	 | j                  rG| j                  s;t        | dd      dz   | _        | j                  dkD  rt        j                  dd       |sy|dk  rAg }	 | j                          d {   }|sn|j                  |       -dj                  |      S | j                  s>| j                  s2| j                  d	       d {    | j                  s| j                  s2| j                  |      S 7 7 0w)
NTrb   r   r      zEMultiple access to StreamReader in eof state, might be infinite loop.)
stack_infor   rH   )r_   r\   rZ   getattrrb   r   warningrM   rs   rt   r   _read_nowait)r(   rE   blocksblocks       r    rH   zStreamReader.read  s    ??&//! yy$+D.!$Dq$H!$$q(#++2#' q5
 F"lln,e$	 
 88F##
 ,,tyy**V$$$ ,,tyy   ## - %s+   BDDAD$D%DDDc                    K   | j                   | j                   | j                  s>| j                  s2| j                  d       d {    | j                  s| j                  s2| j	                  d      S 7 .w)NrM   r   )r_   rZ   r\   r   r   r+   s    r    rM   zStreamReader.readany  sb     ??&//!
 ,,tyy**Y''' ,,tyy   $$ (s   AA8A6A8%A8c                   K   	 | j                   | j                   | j                  r|| j                  j                  d      }|| j                  k(  ry|| j                  kD  r | j	                  || j                  z
        dfS t        j                  d       | j                  r|| j                  r| j                  d      dfS | j                  ry| j                  d	       d{    7 w)
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   Fr=   r>   )r_   rY   poprX   r   r   r   rZ   r   r\   r   )r(   r   s     r    r>   zStreamReader.readchunk  s      *oo%))--11!4$,,&&% --cDLL.@A4HH''8 )) ||//3U;; yy $**[)))1 0 *s   B!C,$A C,$C*%C,c                 T  K   | j                   | j                   g }|dkD  rs| j                  |       d {   }|s3dj                  |      }t        j                  |t        |      |z         |j                  |       |t        |      z  }|dkD  rsdj                  |      S 7 ow)Nr   r   )r_   rH   rt   rj   IncompleteReadErrorr   rs   )r(   rE   r   r   partials        r    readexactlyzStreamReader.readexactly  s     ??&//! !e))A,&E((6*11'3w<!;KLLMM% UOA !e xx 's   4B(B&AB(B(c                     | j                   | j                   | j                  r%| j                  j                         st        d      | j	                  |      S )Nz9Called while some coroutine is waiting for incoming data.)r_   r]   doner   r   rJ   s     r    read_nowaitzStreamReader.read_nowait  sQ    
 ??&//!<< 1 1 3K    ##r   c                    | j                   d   }| j                  }|dk7  r/t        |      |z
  |kD  r||||z    }| xj                  |z  c_        nC|r'| j                   j                          ||d  }d| _        n| j                   j                         }t        |      }| xj                  |z  c_        | xj
                  |z  c_        | j                  }|r8|d   | j
                  k  r&|j                  d       |r|d   | j
                  k  r&| j                  | j                  k  r0| j                  j                  r| j                  j                          |S )Nr   r   )rZ   r[   r   popleftrW   rX   rY   r   rT   rS   r   r   )r(   rE   first_bufferr   r   r   chunk_splitss          r    r   zStreamReader._read_nowait_chunk  s%   ||A$$7s<(61A5!4D1$LL  "(D"#D <<'')Dt9

h
 ..|A=Q |A= ::'DNN,J,JNN))+r   c                    | j                   j                          g }| j                  rH| j                  |      }|j	                  |       |dk7  r|t        |      z  }|dk(  rn| j                  rH|rdj                  |      S dS )z6Read not more than n bytes, or whole buffer if n == -1r   r   r   )r`   assert_timeoutrZ   r   rs   r   rt   )r(   rE   chunksr   s       r    r   zStreamReader._read_nowait  sy    ""$ll++A.EMM% BwSZ6 ll $*sxx2s2r   r%   Nr   )   
r   )*r   r   r   r   r4   r   rQ   r	   r   rj   AbstractEventLoopr)   strrv   r
   rx   BaseExceptionr{   r   r   r   r   r   r@   r   r   r   r?   r   r   r   r   r   rD   r   rH   rM   r>   r   r   r   r   r   r   r    r   r   e   s   I2 -148 
 () w001 
8'# '3c3h 38M2  $122 !2 
	2$1xD1 1d 1$0 . .	$ $ (+e +3 +t +&)%< S  T  *& &" "5 "H($C ($ ($T
%u 
% *ud{!3  *D 3  5  $S $% $C E 83c 3e 3r   r   c                      e Zd ZdZddZdefdZdee   fdZ	e
fdededdfd	Zd
eg df   ddfdZddZdefdZdefdZddZddededdfdZdefdZddedefdZdefdZdeeef   fdZdedefdZddedefdZy)EmptyStreamReader_read_eof_chunkr%   Nc                     d| _         y NFr   r+   s    r    r)   zEmptyStreamReader.__init__,  s
    $r   c                 4    d| j                   j                  z  S )Nrp   )rr   r   r+   s    r    rv   zEmptyStreamReader.__repr__/  s    ////r   c                      y r'   r   r+   s    r    r{   zEmptyStreamReader.exception2      r   r}   r~   c                      y r'   r   )r(   r}   r~   s      r    r   zEmptyStreamReader.set_exception5  s    
 	r   r   c                 \    	  |        y # t         $ r t        j                  d       Y y w xY wr   )r   r   r{   r   s     r    r   zEmptyStreamReader.on_eof<  s-    	CJ 	C%%&AB	Cs   
 ++c                      y r'   r   r+   s    r    r   zEmptyStreamReader.feed_eofB      r   c                      yNTr   r+   s    r    r   zEmptyStreamReader.is_eofE  r   r   c                      yr   r   r+   s    r    r   zEmptyStreamReader.at_eofH  r   r   c                    K   y wr'   r   r+   s    r    r   zEmptyStreamReader.wait_eofK  s	        r   rE   c                      y r'   r   )r(   r   rE   s      r    r   zEmptyStreamReader.feed_dataN  r   r   c                    K   ywr/   r   r+   s    r    rD   zEmptyStreamReader.readlineQ  	     r   c                    K   ywr/   r   rJ   s     r    rH   zEmptyStreamReader.readT  r   r   c                    K   ywr/   r   r+   s    r    rM   zEmptyStreamReader.readanyY  r   r   c                 4   K   | j                   sd| _         yyw)NTr=   r   r   r+   s    r    r>   zEmptyStreamReader.readchunk\  s     ###'D s   c                 6   K   t        j                  d|      wr/   )rj   r   rJ   s     r    r   zEmptyStreamReader.readexactlyc  s     ))#q11s   c                      yr/   r   rJ   s     r    r   zEmptyStreamReader.read_nowaitf  s    r   r   r   r   )r   r   r   r4   r)   r   rv   r	   r   r{   r   r   r   r   r   r@   r   r   r   r?   rQ   r   rD   rH   rM   r
   r>   r   r   r   r   r    r   r   (  s   $I%0# 08M2  $1 ! 
	CxD1 Cd C  e  D  C  
u ud{!3 23 25 2S % r   r   r   c                       e Zd ZdZdej
                  ddfdZdefdZde	fdZ
de	fdZdee   fd	Zefd
ededdfdZddededdfdZddZdefdZdee   fdZy)r   z>DataQueue is a general-purpose blocking queue with one reader.re   r%   Nc                 n    || _         d| _        d | _        d | _        t	        j
                         | _        y r   )rV   r\   r]   r_   rl   rm   rZ   )r(   re   s     r    r)   zDataQueue.__init__p  s.    
	7;37.9.?.?.Ar   c                 ,    t        | j                        S r'   )r   rZ   r+   s    r    __len__zDataQueue.__len__w  s    4<<  r   c                     | j                   S r'   r   r+   s    r    r   zDataQueue.is_eofz  s    yyr   c                 8    | j                   xr | j                   S r'   r   r+   s    r    r   zDataQueue.at_eof}  s    yy---r   c                     | j                   S r'   rz   r+   s    r    r{   zDataQueue.exception  r|   r   r}   r~   c                 f    d| _         || _        | j                  x}d | _        t        |||       y y r   )r\   r_   r]   r   r   s       r    r   zDataQueue.set_exception  s9    
 	ll"F/DL&#y1 0r   r   r   c                     | j                   j                  ||f       | j                  x}d | _        t        |d        y y r'   )rZ   rs   r]   r   )r(   r   r   r   s       r    r   zDataQueue.feed_data  s=    T4L)ll"F/DLvt$ 0r   c                 V    d| _         | j                  x}d | _        t        |d        y y r   )r\   r]   r   )r(   r   s     r    r   zDataQueue.feed_eof  s.    	ll"F/DLvt$ 0r   c                   K   | j                   sN| j                  sB| j                  rJ | j                  j	                         | _        	 | j                   d {    | j                   r| j                   j                         \  }}|S | j                  | j                  t        7 M# t
        j                  t
        j                  f$ r	 d | _         w xY wwr'   )rZ   r\   r]   rV   r   rj   CancelledErrorTimeoutErrorr   r_   r   )r(   r   _s      r    rH   zDataQueue.read  s     ||DII||##::335DLll"" <<ll**,GD!K??&//! #**G,@,@A #s1   AC	B( B&B( A	C&B( (,CCc                 ,    t        | j                        S r'   )r"   rH   r+   s    r    r,   zDataQueue.__aiter__  s    "499--r   r   r   )r   r   r   r   rj   r   r)   rQ   r   r@   r   r   r	   r   r{   r   r   r   r   r   rH   r"   r,   r   r   r    r   r   m  s    HBW66 B4 B! ! . .8M2  $1	2	2 !	2 
		2%b % %D %%B  ..r2 .r   r   c                   n     e Zd ZdZdededej                  ddf fdZdde	d	eddf fd
Z
de	fdZ x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.
    rf   rg   re   r%   Nc                V    t         |   |       d| _        || _        |dz  | _        y )N)re   r   ri   )superr)   rW   rS   _limit)r(   rf   rg   re   rr   s       r    r)   zFlowControlDataQueue.__init__  s.     	d#
!air   r   r   c                     t         |   ||       | xj                  |z  c_        | j                  | j                  kD  r2| j                  j
                  s| j                  j                          y y y r'   )r  r   rW   r  rS   r   r   )r(   r   r   rr   s      r    r   zFlowControlDataQueue.feed_data  sU    $%

d
::#DNN,J,JNN((* -K#r   c                 n  K   | j                   sN| j                  sB| j                  rJ | j                  j	                         | _        	 | j                   d {    | j                   r}| j                   j                         \  }}| xj                  |z  c_	        | j                  | j                  k  r0| j                  j                  r| j                  j                          |S | j                  | j                  t        7 # t
        j                  t
        j                  f$ r	 d | _         w xY wwr'   )rZ   r\   r]   rV   r   rj   r  r  r   rW   r  rS   r   r   r_   r   )r(   r   r   s      r    rH   zFlowControlDataQueue.read  s     ||DII||##::335DLll"" <<--/JD$JJ$JzzDKK'DNN,J,J--/K??&//! #**G,@,@A #s1   AD5	D DD B'D5D ,D22D5r   )r   r   r   r   r   rQ   rj   r   r)   r   r   rH   __classcell__)rr   s   @r    r	  r	    sY     $ -0 ;B;T;T 	 +b + +D +B r   r	  )$rj   rl   r   typingr   r   r   r   r   r   r	   r
   r   base_protocolr   helpersr   r   r   r   r   logr   __all__r   r   r   r"   r6   rB   r   r   r   __annotations__r   r	  r   r   r    <module>r     s      
 
 
 (  ! T]!	 !'"+ ( "3 30@3) @3F? ?D &7%8u\" 8>. >.B(9R= (r   