
    pi"                        d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl	m
Z
mZ d dlmZ d dlmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZmZ d dlmZm Z  dd	l!m"Z" dd
l#m$Z$m%Z%m&Z& ddl'm(Z(m)Z)m*Z*m+Z+m,Z, ddl-m.Z.m/Z/m0Z0m1Z1 ddl2m3Z3 ddl4m5Z5 ddl6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z= ddl>m?Z? ej                  dk\  rd dlmAZA nd dlmBZB  eBdd      ZAdZCerddlDmEZE  G d deF      ZG G d deF      ZHdeeI   deeeI   eeIeIf   f   fdZJ	 d/d e
eIeIf   d!eIdeeI   fd"ZK G d# d$      ZL G d% d      ZM e=eMe9j                  &       G d' d(e:             ZO G d) d*      ZPee:eIeIf   ZQ G d+ d,e:      ZR G d- d.      ZSy)0    N)deque)MappingSequence)TracebackType)TYPE_CHECKINGAnyDequeDictIteratorListOptionalTupleTypeUnioncast)	parse_qslunquote	urlencode)CIMultiDictCIMultiDictProxy   )AbstractStreamWriter)DEFAULT_MAX_DECOMPRESS_SIZEZLibCompressorZLibDecompressor)CONTENT_DISPOSITIONCONTENT_ENCODINGCONTENT_LENGTHCONTENT_TRANSFER_ENCODINGCONTENT_TYPE)CHARTOKENparse_mimetypereify)HeadersParser)internal_logger)JsonPayloadLookupErrorOrderPayloadStringPayloadget_payloadpayload_type)StreamReader)      )Self)TypeVarr1   BodyPartReader)bound)MultipartReaderMultipartWriterr3   BadContentDispositionHeaderBadContentDispositionParamparse_content_dispositioncontent_disposition_filename)ClientResponsec                       e Zd Zy)r7   N__name__
__module____qualname__     S/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/aiohttp/multipart.pyr7   r7   M       rB   r7   c                       e Zd Zy)r8   Nr=   rA   rB   rC   r8   r8   Q   rD   rB   r8   headerreturnc                    dt         dt        fddt         dt        fd}dt         dt        ffd}dt         dt        fd}dt         dt        fd}dj                  t        t        j
                  t                    d	d
t         dt         dt         fd}| sd i fS | j                  d      ^}} |      s"t        j                  t        |              d i fS i }|r*|j                  d      }	|	st        j                  t        |              5d|	vr"t        j                  t        |              d i fS |	j                  dd      \  }
}|
j                         j                         }
|j                         }|
|v r"t        j                  t        |              d i fS  |
      st        j                  t        |	              ||
      r> ||      r ||dd       }n |      st        j                  t        |	             1 ||
      rT ||      r|j                  dd      \  }}}|xs d}n t        j                  t        |	             ~	 t!        ||d      }nd} ||      rd} ||dd j                  d            }nL |      rd}nA|r?| d|d    } ||      r-|j                  d        ||dd j                  d            }d}|r"t        j                  t        |              d i fS |||
<   |r*|j                         |fS # t"        $ r" t        j                  t        |	             Y iw xY w)NstringrG   c                 @    t        |       xr t        t        |       k\  S N)boolr"   setrI   s    rC   is_tokenz+parse_content_disposition.<locals>.is_tokenX   s    F|4V 44rB   c                 .    | d   | d   cxk(  xr dk(  S c S )Nr   "rA   rN   s    rC   	is_quotedz,parse_content_disposition.<locals>.is_quoted[   s     ayF2J-#----rB   c                 @     |       xr | j                  d      dk(  S )N'   )count)rI   rO   s    rC   
is_rfc5987z-parse_content_disposition.<locals>.is_rfc5987^   s!    :FLL$5$::rB   c                 $    | j                  d      S )N*)endswithrN   s    rC   is_extended_paramz4parse_content_disposition.<locals>.is_extended_parama   s    s##rB   c                     | j                  d      dz   }|sy| j                  d      r| |d n| |d  }|j                         S )NrZ   r   FrQ   )findr[   isdigit)rI   pos	substrings      rC   is_continuous_paramz6parse_content_disposition.<locals>.is_continuous_paramd   sH    kk#"&,ooc&:F3rNst	  ""rB    )charstextrd   c                8    t        j                  d| dd|       S )Nz\\([z])z\1)resub)re   rd   s     rC   unescapez+parse_content_disposition.<locals>.unescapek   s    vvugR(%66rB   ;r   =r   rQ   rU   rV   utf-8strictTFz\/)strrL   joinmaprg   escaper!   splitwarningswarnr7   poplowerstriplstripr8   r   UnicodeDecodeError)rF   rS   rX   r\   rb   ri   disptypepartsparamsitemkeyvalueencoding_failed_valuerO   s                   @rC   r9   r9   U   s   5 5 5.# .$ .;3 ;4 ;$# $$ $#C #D # -/GGC		44H,I 7s 7c 7c 7 Rx||C(HuH1&9:RxF
yy|MM5f=>d?MM5f=>8OZZQ'
Uiik!&=MM5f=>8O}MM4T:; % q-e_8>?s#% %*[[a%8"!U#.w8>?x: F q!3!3E!:;% "7!E!H:.V$IIaL$VAb\%8%8%?@E"F9&ABRxsA D >>V##5 & 8>?s   L 'MMr|   namec                   
 d|z  
| sy 
| v r| 
   S || v r| |   S g }t        
fd| j                         D              }t        |      D ]T  \  }\  }}|j                  dd      \  }}|j	                  d      r|d d }|t        |      k(  r|j                  |       T n |sy dj                  |      }d|v r)|j                  dd      \  }	}}|	xs d	}	t        ||	d
      S |S )Nz%s*c              3   P   K   | ]  \  }}|j                        s||f  y wrK   )
startswith).0r~   r   name_sufs      rC   	<genexpr>z/content_disposition_filename.<locals>.<genexpr>   s)      
'S%S^^H=US%L
s   &	&rZ   r   rQ   rc   rU   rV   rl   rm   )	sorteditems	enumeraterr   r[   rn   appendro   r   )r|   r   r{   fnparamsnumr~   r   r   tailr   r   s             @rC   r:   r:      s    t|H	V	h	d| 
+1<<>
 
 "+8!4 	C#uiiQ'GAt}}S!CRys3xU#	 %<!&S!!4Ha*7H5(H55rB   c                   j    e Zd ZdZ	 	 	 	 	 	 ddZddZded   fdZdefdZ	de
ed      fd	Zdd
Zy)MultipartResponseWrapperzzWrapper around the MultipartReader.

    It takes care about
    underlying connection and close it when it needs in.
    rG   Nc                      || _         || _        y rK   )respstream)selfr   r   s      rC   __init__z!MultipartResponseWrapper.__init__   s    
 	rB   c                     | S rK   rA   r   s    rC   	__aiter__z"MultipartResponseWrapper.__aiter__       rB   )r5   r3   c                 R   K   | j                          d {   }|t        |S 7 wrK   nextStopAsyncIterationr   parts     rC   	__anext__z"MultipartResponseWrapper.__anext__   +      YY[ <$$ !   '%'c                 J    | j                   j                  j                         S )z2Returns True when all response data had been read.)r   contentat_eofr   s    rC   r   zMultipartResponseWrapper.at_eof   s    yy  ''))rB   c                    K   | j                   j                          d{   }| j                   j                         r| j                          d{    |S 7 87 w)z#Emits next multipart reader object.N)r   r   r   releaser   r}   s     rC   r   zMultipartResponseWrapper.next   sJ      [[%%'';;,,.   ( s!   AA1AAAAc                 T   K   | j                   j                          d{    y7 w)z_Release the connection gracefully.

        All remaining content is read to the void.
        N)r   r   r   s    rC   r   z MultipartResponseWrapper.release  s     
 ii!!!s   (&()r   r;   r   r5   rG   N)rG   r   rG   N)r>   r?   r@   __doc__r   r   r   r   rL   r   r   r   r   rA   rB   rC   r   r      si     " 
		2	3* *	%;<	="rB   r   c                      e Zd ZdZdZddeddeddd	ed
ede	e   de
ddfdZdedefdZdefdZde	e   fdZdddedefdZefde
defdZde
defdZde
defdZdefdZd+dZddde	e   defdZddde	e   de	eeef      fd Zddde	e   deeeef      fd!Zdefd"Zd#edefd$Zd#edefd%Z d#edefd&Z!d'edefd(Z"e#de	e   fd)       Z$e#de	e   fd*       Z%y),r3   z&Multipart reader for single body part.i    mixedN)subtypedefault_charsetmax_decompress_sizeboundaryheadersCIMultiDictProxy[str]r   r   r   r   rG   c                l   || _         || _        t        |      dz   | _        || _        || _        d| _        |dk(  | _        | j                  rd n| j                   j                  t        d       }|t        |      nd | _        d| _        t               | _        d | _        d| _        i | _        || _        y )NrV   F	form-datar   )r   	_boundarylen_boundary_len_content_default_charset_at_eof_is_form_datagetr   int_length_read_bytesr   _unread_prev_chunk_content_eof_cache_max_decompress_size)r   r   r   r   r   r   r   lengths           rC   r   zBodyPartReader.__init__  s     ! ]Q. /$3++1A1A.RV1W&,&8s6{d%*W,0&($7!rB   r   c                     | S rK   rA   r   s    rC   r   zBodyPartReader.__aiter__,  r   rB   c                 R   K   | j                          d {   }|t        |S 7 wrK   r   r   s     rC   r   zBodyPartReader.__anext__/  s)     YY[ <$$ !r   c                 H   K   | j                          d {   }|sy |S 7 	wrK   )readr   s     rC   r   zBodyPartReader.next5  s$     YY[  !s   " 
"Fdecoder   c                  K   | j                   ryt               }| j                   s?|j                  | j                  | j                         d{          | j                   s?|r| j                  |       d{   S |S 7 27 w)zReads body part data.

        decode: Decodes data following by encoding
                method from Content-Encoding header. If it missed
                data remains untouched
        rB   N)r   	bytearrayextend
read_chunk
chunk_sizer   )r   r   datas      rC   r   zBodyPartReader.read;  sh      <<{,,KKdoodoo>>? ,,T*** ?*s*   ABBB%B;B<BBsizec                 D  K   | j                   ry| j                  r| j                  |       d{   }n| j                  |       d{   }| j                  j                  t              }|r)|j                         dk(  rdj                  |j                               }t        |      dz  }|dk7  r| j                         sd|z
  }d}| j                  r,| j                  d| }| j                  t        |      d | _        t        |      |k7  r2|| j                  j                  dt        |      z
         d{   z  }|sd| _         |dj                  |j                               z  }||z  }t        |      dz  }|dk7  r| j                         s| xj                  t        |      z  c_        | j                  | j                  k(  rd| _         | j                   r0| j                  j!                          d{   dk7  rt#        d      |S 7 7 7 7 w)	zWReads body part content chunk of the specified size.

        size: chunk size
        rB   Nbase64   r   T   
z3Reader did not read all the data or it is malformed)r   r   _read_chunk_from_length_read_chunk_from_streamr   r   r   rv   ro   rr   r   r   r   r   r   r   readline
ValueError)r   r   chunkr   stripped_chunk	remainderover_chunk_size
over_chunks           rC   r   zBodyPartReader.read_chunkK  s    
 <<<<66t<<E66t<<E <<##$=>(H4 XXekkm4NN+a/Iq."#i- 
##!%!1!12B?!CJ'+'7'7J8I'JD$z?o5(:(:1s:;N(O"OOJ!#'DL#((:+;+;+=">>#/!3	! q.$ 	CJ&t||+DL<<$--"8"8"::gERSSE =<$ #P ;sN   .H HH 
HC4H ?H AH A'H HH H H H c                   K   | j                   J d       t        || j                   | j                  z
        }| j                  j	                  |       d {   }| j                  j                         rd| _        |S 7 'w)Nz(Content-Length required for chunked readT)r   minr   r   r   r   r   )r   r   r   r   s       rC   r   z&BodyPartReader._read_chunk_from_lengthw  sq      ||'S)SS't||d.>.>>?
mm((44==!DL 5s   ABA?(Bc           
        K   || j                   k\  sJ d       | j                  d u }|r+d| j                  j                  |       d {   z   | _        d}t	        |      | j                   k  r|| j                  j                  |       d {   z  }| xj
                  t        | j                  j                               z  c_        | j
                  dkD  rt        d      | j
                  rnt	        |      | j                   k  rt	        |      |kD  r#| j                  j                  ||d         |d | }| j                  J | j                  |z   }d| j                  z   }|r|j                  |      }n;|j                  |t        dt	        | j                        t	        |      z
              }|dk\  rt        j                         5  t        j                  dt                | j                  j                  ||d         d d d        | j                  d | | _        |t	        | j                        | }|sd	| _        | j                  |rdndd  }|| _        |S 7 7 # 1 sw Y   `xY ww)
Nz<Chunk size must be greater or equal than boundary length + 2r   rB   rV   zReading after EOFr   ignore)categoryT)r   r   r   r   r   r   r   r   r   unread_datar   r^   maxrs   catch_warningsfilterwarningsDeprecationWarningr   )r   r   first_chunkr   windowrh   idxresults           rC   r   z&BodyPartReader._read_chunk_from_stream  sC     D&&&	JI	J&&&$.&t}}/A/A$/G)GGD %j4---4==--d333ET]]%9%9%;!<<  1$ !455   %j4--- u:MM%%eDEl3%4LE+++!!E)&++c"C++c3q#d.>.>*?#c(*J#KLC!8((* 8'';MN))&,78  $//5D3t//037E#!!{!";< E *H 4&8 8sF   AI4I"	AI4I%A;I4
CI4:I(	AI4%I4(I1-I4c                 h  K   | j                   ry| j                  r| j                  j                         }n"| j                  j	                          d{   }|j                  | j                        r[|j                  d      }| j                  }| j                  dz   }||k(  s||k(  r#d| _         | j                  j                  |       y|S | j                  j	                          d{   }|j                  | j                        r|dd }| j                  j                  |       |S 7 7 Cw)z Reads body part by line by line.rB   Nr      --T)	r   r   popleftr   r   r   r   rstripr   )r   linesliner   last_boundary	next_lines         rC   r   zBodyPartReader.readline  s     <<<<<<'')D//11D??4>>* KK(E~~H NNU2M E]$:###D)  #mm4466I##DNN3CRyLL	*' 2 7s&   AD2D.BD2,D0-AD20D2c                    K   | j                   ry| j                   s1| j                  | j                         d{    | j                   s0yy7 w)z0Like read(), but reads all the data to the void.N)r   r   r   r   s    rC   r   zBodyPartReader.release  s9     <<,,//$//222 ,,2s   8AAAA)r   r   c                   K   | j                  d       d{   }|xs | j                  d      }|j                  |      S 7 +w)z;Like read(), but assumes that body part contains text data.Tr   Nrl   default)r   get_charsetr   r   r   r   s      rC   re   zBodyPartReader.text  sG     YYdY++ @t///@{{8$$	 ,s   AA,Ac                   K   | j                  d       d{   }|sy|xs | j                  d      }t        t        t        t
        f   t        j                  |j                  |                  S 7 \w)z<Like read(), but assumes that body parts contains JSON data.Tr   Nrl   r   )	r   r   r   r
   rn   r   jsonloadsr   r   s      rC   r  zBodyPartReader.json  sc     YYdY++@t///@DcNDJJt{{8/D$EFF	 ,s   A7A5AA7c                
  K   | j                  d       d{   }|sg S ||}n| j                  d      }	 |j                         j                  |      }t        |d|      S 7 M# t        $ r t        d|z        w xY ww)zFLike read(), but assumes that body parts contain form urlencoded data.Tr   Nrl   r   z'data cannot be decoded with %s encoding)keep_blank_valuesr   )r   r   r   r   ry   r   r   )r   r   r   real_encodingdecoded_datas        rC   formzBodyPartReader.form  s     YYdY++I$M ,,W,=M	X;;=//>L ""
 	
 , " 	XFVWW	Xs&   BA&BA( B(B  Bc                     | j                   S )z<Returns True if the boundary was reached or False otherwise.r   r   s    rC   r   zBodyPartReader.at_eof      ||rB   r   c                    K   t         | j                  v r| j                  |      }| j                  s+t        | j                  v r| j                  |       d{   S |S 7 w)zDecodes data.

        Decoding is done according the specified Content-Encoding
        or Content-Transfer-Encoding headers value.
        N)r   r   _decode_content_transferr   r   _decode_content)r   r   s     rC   r   zBodyPartReader.decode  sW      %4006D!!&6$,,&F--d333 4s   AA!AA!c                    K   | j                   j                  t        d      j                         }|dk(  r|S |dv r0t	        |d      j                  || j                         d {   S t        d|       7 w)Nrc   identity>   gzipdeflateT)r   suppress_deflate_header)
max_lengthunknown content encoding: )r   r   r   rv   r   
decompressr   RuntimeErrorr   r   r   s      rC   r  zBodyPartReader._decode_content  s     <<##$4b9??Az!K**)!(, j$*C*CjDE E
 7zBCCEs   A%A<'A:(A<c                     | j                   j                  t        d      j                         }|dk(  rt	        j
                  |      S |dk(  rt        j                  |      S |dv r|S t        d|       )Nrc   r   quoted-printable)binary8bit7bit#unknown content transfer encoding: )	r   r   r   rv   r   	b64decodebinasciia2b_qpr  r  s      rC   r  z'BodyPartReader._decode_content_transfer  st    <<##$=rBHHJx##D))++??4((33K!DXJOPPrB   r   c                     | j                   j                  t        d      }t        |      }|j                  j                  d| j
                  xs |      S )z>Returns charset parameter from Content-Type header or default.rc   charset)r   r   r    r#   
parametersr   )r   r   ctypemimetypes       rC   r   zBodyPartReader.get_charset  sF      r2!%(""&&y$2G2G2R7SSrB   c                 p    t        | j                  j                  t                    \  }}t	        |d      S )z|Returns name specified in Content-Disposition header.

        If the header is missing or malformed, returns None.
        r   r9   r   r   r   r:   r   r   r|   s      rC   r   zBodyPartReader.name%  s/     .dll.>.>?R.ST	6+FF;;rB   c                 p    t        | j                  j                  t                    \  }}t	        |d      S )zReturns filename specified in Content-Disposition header.

        Returns None if the header is missing or malformed.
        filenamer(  r)  s      rC   r+  zBodyPartReader.filename.  s/     .dll.>.>?R.ST	6+FJ??rB   r   )&r>   r?   r@   r   r   r   bytesr.   rn   r   r   r   r1   r   r   r   rL   r   r   r   r   r   r   re   r
   r   r  r   r   r  r   r   r  r  r   r$   r   r+  rA   rB   rC   r3   r3     s   0J )-#>88 )8 	8 8 "#8 !8 
86   HUO  ,1 D U   ,6 *S *% *X# % +# +% +Z :3 7; %hsm %s % 7; Ghsm GxSRUX?W G 7; 
hsm 
tE#s(O?T 
(  5 
D% 
DE 
D
QU 
Qu 
QT3 T3 T <hsm < < @(3- @ @rB   )orderc                   x     e Zd ZU eed<   dedededdf fdZdded	edefd
Zdded	ede	fdZ
deddfdZ xZS )BodyPartReaderPayloadr   r   argskwargsrG   Nc                     t        |   |g|i | i }|j                  |j                  |d<   |j                  |j                  |d<   |r | j                  di | y y )Nr   r+  )
attachmentT)superr   r   r+  set_content_disposition)r   r   r0  r1  r|   	__class__s        rC   r   zBodyPartReaderPayload.__init__=  si    000!#::!"ZZF6N>>%!&F:(D((FvF rB   r   errorsc                     t        d      )NzUnable to decode.	TypeErrorr   r   r7  s      rC   r   zBodyPartReaderPayload.decodeI  s    +,,rB   c                     K   t        d      w)aR  Raises TypeError as body parts should be consumed via write().

        This is intentional: BodyPartReader payloads are designed for streaming
        large data (potentially gigabytes) and must be consumed only once via
        the write() method to avoid memory exhaustion. They cannot be buffered
        in memory for reuse.
        z:Unable to read body part as bytes. Use write() to consume.r9  r;  s      rC   as_byteszBodyPartReaderPayload.as_bytesL  s      TUUs   writerc                   K   | j                   }|j                  d       d {   }|rN|j                  |j                  |       d {          d {    |j                  d       d {   }|rMy y 7 U7 17 '7 w)Ni   )r   )r   r   writer   )r   r>  fieldr   s       rC   r@  zBodyPartReaderPayload.writeV  sr     &&E&22,,U\\%%88999***66E  3896sK   "BA:%B
A<BA>B0B 1B8B<B>B Brl   rm   )r>   r?   r@   r3   __annotations__r   r   rn   r   r,  r=  r   r@  __classcell__r6  s   @rC   r/  r/  8  s|    
Gn 
GS 
GC 
GD 
G-s -c - -Vs Vc VQV V7"6 74 7rB   r/  c                   0   e Zd ZU dZeZdZeed       e	d<   e
Zdeeef   deddfdZdedefd	Zdeed e
f      fd
Zedddefd       ZdefdZdeed e
f      fdZddZded e
f   fdZddded e
f   fdZdefdZdefdZddZddZ ddZ!ddZ"y)r5   zMultipart body reader.Nmultipart_reader_clsr   r   rG   c                 z   t        |t                 | _        | j                  j                  dk(  sJ d       d| j                  j                  vrt        d|t           z        || _        d| j                         z   j                         | _	        || _
        d | _        d | _        d| _        d| _        g | _        y )N	multipartz!multipart/* content type expectedr   z$boundary missed for Content-Type: %s--FT)r#   r    	_mimetypetyper$  r   r   _get_boundaryencoder   r   r   
_last_partr   _at_bofr   )r   r   r   s      rC   r   zMultipartReader.__init__i  s    '(=>~~""k1V3VV1T^^66669NN  !3!3!55==?/3NR$&rB   r   c                     | S rK   rA   r   s    rC   r   zMultipartReader.__aiter__z  r   rB   c                 R   K   | j                          d {   }|t        |S 7 wrK   r   r   s     rC   r   zMultipartReader.__anext__}  r   r   responser;   c                 `    | j                  | | |j                  |j                              }|S )zConstructs reader instance from HTTP response.

        :param response: :class:`~aiohttp.client.ClientResponse` instance
        )response_wrapper_clsr   r   )clsrS  objs      rC   from_responsezMultipartReader.from_response  s3     &&c(**H,<,<=
 
rB   c                     | j                   S )z@Returns True if the final boundary was reached, false otherwise.r
  r   s    rC   r   zMultipartReader.at_eof  r  rB   c                 .  K   | j                   ry| j                          d{    | j                  r | j                          d{    d| _        n| j	                          d{    | j                   ry| j                          d{   }| j                  | j                  j                  dk(  rt        |t              rt        |j                  j                  t                    \  }}|j                  d      dk(  rm|j                  d       d{   }t!        |      dkD  rt#        d      |j%                         j'                         | _        | j                          d{   }|| _        | j                  S 7 b7 A7 $7 7 w7 %w)	z#Emits the next multipart body part.NFr   r   	_charset_       zInvalid default charset)r   _maybe_release_last_partrP  _read_until_first_boundary_read_boundaryfetch_next_partrO  rK  r   
isinstancer3   r9   r   r   r   r   r   r  rw   r   r   )r   r   r   r|   r#  s        rC   r   zMultipartReader.next  sM    
 <<++---<<11333 DL%%'''<<))++ OO#&&+5401$,,2B2BCV2WXIAvzz&![0 !% 33w<"$&'@AA(/(>(>(@%!11335 	.3 ( , 4 4sk   !FF#FFF'F($FFBFFAF-F.FFFFFFc                    K   | j                   sA| j                          d{   }|y|j                          d{    | j                   s@yy7 .7 w)z=Reads all the body parts to the void till the final boundary.N)r   r   r   r   s     rC   r   zMultipartReader.release  sD     ,,$D|,,.  	 ,,$ !s%    AAAAAAAc                 `   K   | j                          d{   }| j                  |      S 7 w)z"Returns the next body part reader.N)_read_headers_get_part_reader)r   r   s     rC   ra  zMultipartReader.fetch_next_part  s/      **,,$$W-- -s   .,.r   c                 z   |j                  t        d      }t        |      }|j                  dk(  rD| j                   t        |       || j
                        S | j	                  || j
                        S | j                  | j                  || j
                  | j                  j                  | j                        S )zDispatches the response by the `Content-Type` header.

        Returns a suitable reader instance.

        :param dict headers: Response headers
        rc   rI  )r   r   )r   r    r#   rL  rG  r   part_reader_clsr   rK  r   r   )r   r   r%  r&  s       rC   rf  z MultipartReader._get_part_reader  s     L"-!%(==K'((0!tDz'4==99,,WdmmDD''.. $ 5 5 (  rB   c                 p    | j                   j                  d   }t        |      dkD  rt        d|z        |S )Nr   F   z&boundary %r is too long (70 chars max))rK  r$  r   r   )r   r   s     rC   rM  zMultipartReader._get_boundary  s8    >>,,Z8x=2EPQQrB   c                    K   | j                   r| j                   j                         S | j                  j                          d {   S 7 wrK   )r   ru   r   r   r   s    rC   	_readlinezMultipartReader._readline  s9     <<<<##%%]]++----s   AAAAc                    K   	 | j                          d {   }|dk(  rt        d| j                  z        |j                         }|| j                  k(  ry || j                  dz   k(  rd| _        y p7 \w)NTrB   z#Could not find starting boundary %rr   )rl  r   r   r   r   r   r   s     rC   r_  z*MultipartReader._read_until_first_boundary  sw     ..**E| 9T^^L  LLNE&$..500# *s   A6A4AA6c                   K   | j                          d {   j                         }|| j                  k(  ry || j                  dz   k(  ryd| _        | j                          d {   }| j                          d {   }|d d dk(  r| j                  j                  |       y | j                  j                  ||g       y t        d|d| j                        7 7 {7 ew)Nr   TrV   zInvalid boundary z, expected )rl  r   r   r   r   r   r   r   )r   r   epiloguer   s       rC   r`  zMultipartReader._read_boundary  s     ~~''//1DNN"dnnu,,DL!^^--H"nn..I !}%##I.
 ##Y$9:0	T^^DVWXX) (
 ..s5   C%CAC%%C!&C%=C#>A"C%!C%#C%c                    K   g }	 | j                   j                          d {   }|j                  d      }|j                  |       |snHt	               }|j                  |      \  }}|S 7 Jw)Nr   )r   r   r   r   r%   parse_headers)r   linesr   parserr   raw_headerss         rC   re  zMultipartReader._read_headers  sp     --0022ELL)ELL  %33E: 3s   !A0A.AA0c                   K   | j                   s| j                   j                         s"| j                   j                          d{    | j                  j	                  | j                   j                         d| _         yy7 <w)z8Ensures that the last read body part is read completely.N)rO  r   r   r   r   r   s    rC   r^  z(MultipartReader._maybe_release_last_part   sa     ??&??))+oo--///LL 7 78"DO	 '/s   ABB=Br   )rG   r   )#r>   r?   r@   r   r   rU  rG  r   r   rC  r3   rh  r   rn   r.   r   r1   r   r   r   classmethodrX  rL   r   r   r   ra  rf  rM  r,  rl  r_  r`  re  r^  rA   rB   rC   r5   r5   ^  s:     4 ?C(4(9#:;B$O'S 1 'L 'T '"  	%)>9:	; " 
"  !	%)>9:	;!F!.	 .0	1.( 
 .0	14s . .
Y.
#rB   r5   c                   0    e Zd ZU dZded<   dZd%dedee   ddf fdZd&d	Z	d
ee
e      dee   dee   ddfdZdee   fdZdefdZdefdZ ej,                  d      Z ej,                  d      Zedefd       Zedefd       Zd'dedeeeef      defdZdedefdZ 	 d'dedeeeef      defdZ!	 d'de"e#e$eef      eeef   f   deeeef      defdZ%edee   fd       Z&d(dededefdZ'd(dedede(fd Z)	 d)d!e*d"eddfd#Z+d*d$Z, xZ-S )+r6   zMultipart body writer.Nr   Tr   r   rG   c                    ||nt        j                         j                  }	 |j                  d      | _        d| d| j                   }t        | %  d |       g | _
        |dk(  | _        y # t
        $ r t        d      d w xY w)Nasciiz(boundary should contain ASCII only charsz
multipart/z; boundary=)content_typer   )uuiduuid4hexrN  r   UnicodeEncodeErrorr   _boundary_valuer4  r   _partsr   )r   r   r   r%  r6  s       rC   r   zMultipartWriter.__init__3  s    '389I9I	S%__W5DN WI[1E1E0FGE2#%$3 " 	SGHdR	Ss   A0 0Bc                     | S rK   rA   r   s    rC   	__enter__zMultipartWriter.__enter__E  r   rB   exc_typeexc_valexc_tbc                      y rK   rA   )r   r  r  r  s       rC   __exit__zMultipartWriter.__exit__H  s     	rB   c                 ,    t        | j                        S rK   )iterr  r   s    rC   __iter__zMultipartWriter.__iter__P  s    DKK  rB   c                 ,    t        | j                        S rK   )r   r  r   s    rC   __len__zMultipartWriter.__len__S  s    4;;rB   c                      y)NTrA   r   s    rC   __bool__zMultipartWriter.__bool__V  s    rB   s   \A[!#$%&'*+\-.^_`|~\w]+\Zs   [\x00-\x08\x0A-\x1F\x7F]c                 H   | j                   }t        j                  | j                  |      r|j	                  d      S t        j
                  | j                  |      rt        d      |j                  dd      }|j                  dd      }d|j	                  d      z   dz   S )zzWrap boundary parameter value in quotes, if necessary.

        Reads self.boundary and returns a unicode string.
        rz  z*boundary value contains invalid characters   \s   \\   "s   \"rR   )	r   rg   match_valid_tchar_regexr   search_invalid_qdtext_char_regexr   replace)r   r   quoted_value_contents      rC   r  zMultipartWriter._boundary_value\  s    & 88D++U3<<((99T44e<IJJ  %}}UG<3;;D&I)0099C??rB   c                 8    | j                   j                  d      S )Nrz  )r   r   r   s    rC   r   zMultipartWriter.boundary|  s    ~~$$W--rB   rW  r   c                    |
t               }t        |t              r,|j                  j	                  |       | j                  |      S 	 t        ||      }| j                  |      S # t        $ r t        d|z        w xY w)Nr   zCannot create payload from %r)	r   rb  r*   r   updateappend_payloadr,   r(   r:  )r   rW  r   payloads       rC   r   zMultipartWriter.append  s    ?!mGc7#KKw'&&s++4%c7; **733  G ?# EFFGs   
A( (B r  c                    d}d}| j                   rlt        t        t        h|j                  j                         z  rJ t        |j                  vrdt        | j                         }|j                  d|       n|j                  j                  t        d      j                         }|r|dvrt        d|       |dk(  rd}|j                  j                  t        d      j                         }|d	vrt        d
|       |dk(  rd}|j                  }| |s|st        |      |j                  t        <   | j                  j                  |||f       |S )z)Adds a new body part to multipart writer.Nzsection-r   )r   rc   )r  r  r  r  r  )rc   r   r  r  r  r  )r   r   r   r   r   keysr   r   r  r5  r   rv   r  r   rn   r   )r   r  r   te_encodingr   r   s         rC   r  zMultipartWriter.append_payload  sU   "&%) &~7PQ//&&()) #'//9!#dkk"2!34//$/G **+;R@FFHHH,KK"%?z#JKK:% "//--.GLRRTK"NN"%H#VWWh&" <<D[25d)/GX{;<rB   c                 R    |
t               }| j                  t        ||            S )zHelper to append JSON part.r  )r   r  r'   )r   rW  r   s      rC   append_jsonzMultipartWriter.append_json  s(     ?!mG"";sG#DEErB   c                     t        |t        t        f      sJ |
t               }t        |t              rt	        |j                               }t        |d      }| j                  t        ||d            S )z&Helper to append form urlencoded part.T)doseqz!application/x-www-form-urlencoded)r   r{  )	rb  r   r   r   listr   r   r  r+   )r   rW  r   r   s       rC   append_formzMultipartWriter.append_form  sn     #'2333?!mGc7#syy{#CD)""g4W
 	
rB   c                    d}| j                   D ]]  \  }}}|j                  }|s|s| y|t        dt        | j                        z   dz   |z   t        |j
                        z   dz         z  }_ |dt        | j                        z   dz   z  }|S )zSize of the payload.r   NrV   r   )r  r   r   r   r   _binary_headers)r   totalr   r   r  	part_sizes         rC   r   zMultipartWriter.size  s     +/;; 	'D(K		I;)*;Sdnn%&  d**+	,
  E	 	S((1,,rB   r   r7  c                 T     dj                   fd j                  D              S )zReturn string representation of the multipart data.

        WARNING: This method may do blocking I/O if parts contain file payloads.
        It should not be called in the event loop. Use as_bytes().decode() instead.
        rc   c              3      K   | ]H  \  }}}d j                   z   dz   |j                  j                        z   |j                         z    J yw)rJ  z
N)r   r  r   )r   r   _e_ter   r7  r   s       rC   r   z)MultipartWriter.decode.<locals>.<genexpr>  sb      
 b# mm ""))(F;< kkm	
s   AA)ro   r  r;  s   ```rC   r   zMultipartWriter.decode  s)     ww 
 "&
 
 	
rB   c                   K   g }| j                   D ]~  \  }}}|j                  d| j                  z   dz          |j                  |j                         |j	                  ||       d{   }|j                  |       |j                  d        |j                  d| j                  z   dz          dj                  |      S 7 Zw)zReturn bytes representation of the multipart data.

        This method is async-safe and calls as_bytes on underlying payloads.
        r   r   N   --
rB   )r  r   r   r  r=  ro   )r   r   r7  r{   r   r  r  
part_bytess           rC   r=  zMultipartWriter.as_bytes  s     
   "[[ 	"MD"cLL/'9: LL--.  $}}Xv>>JLL$ LL!	" 	UT^^+i78xx ?s   A'C)C*ACr>  close_boundaryc                   K   | j                   D ]#  \  }}}| j                  r+t        |j                  v sJ d|j                  t           v sJ |j	                  d| j
                  z   dz          d{    |j	                  |j                         d{    |s|rct        |      }|r|j                  |       |r|j                  |       |j	                  |       d{    |j                          d{    n|j	                  |       d{    |j	                  d       d{    & |r*|j	                  d| j
                  z   dz          d{    yy7 7 7 7 m7 U7 >7 w)zWrite body.zname=r   r   Nr  )r  r   r   r   r@  r   r  MultipartPayloadWriterenable_compressionenable_encoding	write_eof)r   r>  r  r   r   r  ws          rC   r@  zMultipartWriter.write  sO     ,0;; 	('D(K!!*dll:::$,,/B"CCCC,,ut~~5?@@@,,t33444;*62((2%%k2jjm##kkm##jj(((,,w''')	(, ,,ut~~5	ABBB ! A4 $#(' Cs   A0E12E#3"E1E%AE1#E'$E1;E)<E1E+E1.E-/-E1E/E1%E1'E1)E1+E1-E1/E1c                 >  K   | j                   ryd| _         t        | j                        D ]<  \  }\  }}}|j                  r|j                  r$	 |j                          d{    > y7 # t        $ r#}t        j                  d||d       Y d}~id}~ww xY ww)a  
        Close all part payloads that need explicit closing.

        IMPORTANT: This method must not await anything that might not finish
        immediately, as it may be called during cleanup/cancellation. Schedule
        any long-running operations without awaiting them.
        NTz%Failed to close multipart part %d: %s)exc_info)		_consumedr   r  	autocloseconsumedclose	Exceptionr&   error)r   r   r   r   excs        rC   r  zMultipartWriter.close/  s      >> "+4;;!7 	C$1>>$--**,&&	 '  #))?cTX sH   A BBA.$A,%A.)B,A..	B7BBBB)r   N)rG   r6   rK   rB  )Tr   ).r>   r?   r@   r   rC  
_autoclosern   r   r   r  r   BaseExceptionr   r  r   _Partr  r   r  rL   r  rg   compiler  r  propertyr  r   r   r   r*   r   r  r  r   r   r   r  r   r   r,  r=  r   r@  r  rD  rE  s   @rC   r6   r6   ,  s?    LJ4 4# 4RV 4$4./ -( '	
 
!(5/ !   $  $$AB!+,H!I@ @ @> .# . .4# 4c1B(C 4w 4$g $' $N @DFF!)'#s(*;!<F	F 04
8E#s(O,gc3h.??@
 '#s(+,
 
	
* hsm  (
s 
c 
 
s c QV 8 DHC*C<@C	C:rB   r6   c                   b    e Zd ZdeddfdZdeddfdZ	 ddedee   ddfdZ	dd	Z
d
eddfdZy)r  r>  rG   Nc                 <    || _         d | _        d | _        d | _        y rK   )_writer	_encoding	_compress_encoding_buffer)r   r>  s     rC   r   zMultipartPayloadWriter.__init__J  s    (,3759rB   r   c                 V    |dk(  r|| _         t               | _        y |dk(  rd| _         y y )Nr   r  )r  r   r  )r   r   s     rC   r  z&MultipartPayloadWriter.enable_encodingP  s0    x%DN$-KD!++/DN ,rB   strategyc                 *    t        |d|      | _        y )NT)r   r  r  )r   r  )r   r   r  s      rC   r  z)MultipartPayloadWriter.enable_compressionW  s     ($(
rB   c                 ^  K   | j                   <| j                   j                         }|r d | _         | j                  |       d {    | j                  dk(  rN| j                  rA| j
                  j                  t        j                  | j                               d {    y y y 7 b7 	w)Nr   )r  flushr@  r  r  r  r   	b64encodern  s     rC   r  z MultipartPayloadWriter.write_eof`  s     >>%NN((*E!%jj'''>>X%$$ll(()9)9$:O:O)PQQQ % & ( Rs%   AB-B)AB-!B+"B-+B-r   c                   K   | j                   (|r&| j                   j                  |       d {   }|sy | j                  dk(  r| j                  }|J |j	                  |       |rjt        t        |      d      \  }}|d |dz   ||dz  d  c}| _        |r9t        j                  |      }| j                  j                  |       d {    y y y | j                  dk(  r7| j                  j                  t        j                  |             d {    y | j                  j                  |       d {    y 7 7 t7 .7 w)Nr   r/   r  )r  compressr  r  r   divmodr   r   r  r  r@  r   b2a_qp)r   r   bufdivmod	enc_chunkb64chunks          rC   r@  zMultipartPayloadWriter.writek  s'    >>%"nn55e<<>>X%''C?"?JJu!#c(A.S47	#'NCa	N0	40%//	:H,,,,X666   ^^11,,$$X__U%;<<<,,$$U+++% = 7<+sG   -ED>BEEAEE#E8E9EEEE)r  Nr   )r>   r?   r@   r   r   rn   r  r   r   r  r  r,  r@  rA   rB   rC   r  r  I  sj    :3 : :0 0 0 DH

3;C=
	
	R, ,4 ,rB   r  )r+  )Tr   r   r  rg   sysr|  rs   collectionsr   collections.abcr   r   typesr   typingr   r   r	   r
   r   r   r   r   r   r   r   urllib.parser   r   r   	multidictr   r   abcr   compression_utilsr   r   r   hdrsr   r   r   r   r    helpersr!   r"   r#   r$   httpr%   logr&   r  r'   r(   r)   r*   r+   r,   r-   streamsr.   version_infor1   r2   __all__client_reqrepr;   RuntimeWarningr7   r8   rn   r9   r:   r   r3   	try_firstr/  r5   r  r6   r  rA   rB   rC   <module>r     s      	 
    -     7 6 3 % 
  8 7      "w6!12D -	. 		 	d$SMd$
8C=$sCx.()d$P ,6CH%(c]B," ,"^i@ i@X	 nEOO4"7G "7 5"7JH# H#V 	gsC Zg Zz7, 7,rB   