
    pir\                     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__     Q/opt/services/ai/voice_agent/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ed	ef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_low_water_chunks_high_water_chunks_loop_size_cursor_http_chunk_splits_buffer_buffer_offset_eof_waiter_eof_waiter
_exception_timer_eof_callbacks_eof_countertotal_bytestotal_compressed_bytesN)timerloopprotocollimitrg   rh   r%   c                   || _         || _        |dz  | _        |t        j                         }t        d|dz        | _        t        d| j                  dz        | _        || _        d| _	        d| _
        d | _        t        j                         | _        d| _        d| _        d | _        d | _        d | _        |
t)               n|| _        g | _        d| _        d| _        d | _        y )N         r   F)rS   rT   rU   asyncioget_event_loopmaxrW   rV   rX   rY   rZ   r[   collectionsdequer\   r]   r^   r_   r`   ra   r   rb   rc   rd   re   rf   )r(   ri   rj   rg   rh   s        r    r)   zStreamReader.__init__   s     " 19<))+D"%a!"4 "%Q(?(?1(D!E

8<%0%6%6%8	7;;?37%*]ik8:59#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   rY   appendr^   rT   rU   r_   ra   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'   ra   r+   s    r    	exceptionzStreamReader.exception       r   exc	exc_causec                     || _         | j                  j                          | j                  }|d | _        t	        |||       | j
                  }|d | _        t	        |||       y y r'   )ra   rc   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   rc   ry   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_readingrc   r   r   r   r   )r(   r   cbs      r    feed_eofzStreamReader.feed_eof   s    	DLvt$!!#Dvt$>>))NN))+%% 	GBG	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^   r\   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`   rX   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(c                 J    | j                   | j                  S | j                   S r'   )rf   re   r+   s    r    total_raw_byteszStreamReader.total_raw_bytes  s&    &&.###***r   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)rl   )
stacklevelNr   )
warningswarnDeprecationWarningr]   r\   rY   lenrZ   
appendleftrd   )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   rY   r\   ry   re   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                 ~    | j                   1| j                  rt        d      t        j                         | _         y y )Nz@Called begin_http_chunk_receiving when some data was already fed)r[   re   RuntimeErrorrr   rs   r+   s    r    begin_http_chunk_receivingz'StreamReader.begin_http_chunk_receiving3  s?    ""*"V  '2&7&7&9D# +r   c                    | j                   t        d      | j                   r| j                   d   nd}| j                  |k(  ry | j                   j                  | j                         t	        | j                         | j
                  kD  r0| j                  j                  s| j                  j                          | j                  }|d | _	        t        |d        y y )NzFCalled end_chunk_receiving without calling begin_chunk_receiving firstr   )r[   r   re   ry   r   rW   rS   r   r   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 ''(4+B+BBNN22NN((* 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_   rX   r   rb   )r(   r   r   s      r    _waitzStreamReader._waitb  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.readlinew  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   
ValueErrorra   r\   r]   find_read_nowait_chunkrU   r^   r   )	r(   r   seplenchunk
chunk_size
not_enoughoffseticharr   s	            r    r   zStreamReader.readuntilz  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)
NTrd   r   r      zEMultiple access to StreamReader in eof state, might be infinite loop.)
stack_infor   rH   )ra   r^   r\   getattrrd   r   warningrM   ry   rz   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   )ra   r\   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                         }|| 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.
        TNr   TzESkipping HTTP chunk end due to data consumption beyond chunk boundaryr   Fr=   r>   )ra   r[   popleftrZ   r   r   r   r\   r   r^   r   )r(   r   s     r    r>   zStreamReader.readchunk  s      *oo%))--557$,,&&% --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   )ra   rH   rz   ro   IncompleteReadErrorr   ry   )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.)ra   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                  }|r7|d   | j
                  k  r%|j                          |r|d   | j
                  k  r%| j                  j                  ra| j                  | j                  k  rH| j                  "t        | j                        | j                  k  r| j                  j                          |S )Nr   r   )r\   r]   r   r   rY   rZ   r[   rS   r   rT   rV   r   )r(   rE   first_bufferr   r   r   chunk_splitss          r    r   zStreamReader._read_nowait_chunk  sK   ||A$$7s<(61A5!4D1$LL  "(D"#D <<'')Dt9

h
 ..|A=  " |A= NN**

T__,''/t../$2H2HH 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   )rb   assert_timeoutr\   r   ry   r   rz   )r(   rE   chunksr   s       r    r   zStreamReader._read_nowait7  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   ro   AbstractEventLoopr)   strr|   r
   r~   BaseExceptionr   r   r   r   r   r   r@   r   r   r   propertyr   r?   r   r   r   r   r   rD   r   rH   rM   r>   r   r   r   r   r   r   r    r   r   e   s   I8 -148 : :  :
 () : w001 : 
 :D'# '3c3h 38M2  $122 !2 
	2$1xD1 1d 1$0 . .	$ + + +
 $ (+e +3 +t +&:%%N S  T  *& &" "5 "H($C ($ ($T
%u 
% *ud{!3  *D 3  5  $S $% $!C !E !F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| _         d| _        y )NFr   )r   re   r+   s    r    r)   zEmptyStreamReader.__init__K  s    $r   c                 4    d| j                   j                  z  S )Nrv   )rx   r   r+   s    r    r|   zEmptyStreamReader.__repr__O  s    ////r   c                      y r'   r   r+   s    r    r   zEmptyStreamReader.exceptionR      r   r   r   c                      y r'   r   )r(   r   r   s      r    r   zEmptyStreamReader.set_exceptionU  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/   )ro   r   rJ   s     r    r   zEmptyStreamReader.readexactly  s     ))#q11s   c                      yr/   r   rJ   s     r    r   zEmptyStreamReader.read_nowait  s    r   r   r   r   )r   r   r   r4   r)   r   r|   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   G  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.rh   r%   Nc                 n    || _         d| _        d | _        d | _        t	        j
                         | _        y )NF)rX   r^   r_   ra   rr   rs   r\   )r(   rh   s     r    r)   zDataQueue.__init__  s.    
	7;37.9.?.?.Ar   c                 ,    t        | j                        S r'   )r   r\   r+   s    r    __len__zDataQueue.__len__  s    4<<  r   c                     | j                   S r'   r   r+   s    r    r   zDataQueue.is_eof  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'   r   r+   s    r    r   zDataQueue.exception  r   r   r   r   c                 f    d| _         || _        | j                  x}d | _        t        |||       y y r   )r^   ra   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'   )r\   ry   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'   )r\   r^   r_   rX   r   ro   CancelledErrorTimeoutErrorr   ra   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   ro   r   r)   rQ   r  r@   r   r   r	   r   r   r   r   r   r   r   rH   r"   r,   r   r   r    r   r     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.
    ri   rj   rh   r%   Nc                V    t         |   |       d| _        || _        |dz  | _        y )N)rh   r   rl   )superr)   rY   rS   _limit)r(   ri   rj   rh   rx   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   rY   r  rS   r   r   )r(   r   r   rx   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'   )r\   r^   r_   rX   r   ro   r  r  r   rY   r  rS   r   r   ra   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   ro   r   r)   r   r   rH   __classcell__)rx   s   @r    r  r    sY     $ -0 ;B;T;T 	 +b + +D +B r   r  )$ro   rr   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) _3D@ @F &7%8u\" 8>. >.B(9R= (r   