
    gN                        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Zd dl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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"m#Z# dd	l$m%Z%m&Z&m'Z'm(Z(m)Z) dd
l*m+Z+m,Z,m-Z-m.Z. ddl/m0Z0 ddl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8 ddl9m:Z: ejv                  dk\  rd dlm<Z< nd dlm=Z=  e=dd      Z<dZ>erddl?m@Z@  G d deA      ZB G d deA      ZCdeeD   deeeD   eeDeDf   f   fdZE	 d,deeDeDf   deDdeeD   fdZF G d  d!      ZG G d" d      ZH e8eHe4j                  #       G d$ d%e5             ZJ G d& d'      ZKee5eDeDf   ZL G d( d)e5      ZM G d* d+      ZNy)-    N)deque)TracebackType)TYPE_CHECKINGAnyDequeDictIteratorListMappingOptionalSequenceTupleTypeUnioncast)	parse_qslunquote	urlencode)CIMultiDictCIMultiDictProxy   )ZLibCompressorZLibDecompressor)CONTENT_DISPOSITIONCONTENT_ENCODINGCONTENT_LENGTHCONTENT_TRANSFER_ENCODINGCONTENT_TYPE)CHARTOKENparse_mimetypereify)HeadersParser)JsonPayloadLookupErrorOrderPayloadStringPayloadget_payloadpayload_type)StreamReader)      )Self)TypeVarr.   BodyPartReader)bound)MultipartReaderMultipartWriterr0   BadContentDispositionHeaderBadContentDispositionParamparse_content_dispositioncontent_disposition_filename)ClientResponsec                       e Zd Zy)r4   N__name__
__module____qualname__     F/var/www/openai/venv/lib/python3.12/site-packages/aiohttp/multipart.pyr4   r4   I       r?   r4   c                       e Zd Zy)r5   Nr:   r>   r?   r@   r5   r5   M   rA   r?   r5   headerreturnc                 R   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      }	d|	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        |	              ||
      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 Hw xY w)NstringrD   c                 @    t        |       xr t        t        |       k\  S N)boolr    setrF   s    r@   is_tokenz+parse_content_disposition.<locals>.is_tokenT   s    F|4V 44r?   c                 .    | d   | d   cxk(  xr dk(  S c S )Nr   "r>   rK   s    r@   	is_quotedz,parse_content_disposition.<locals>.is_quotedW   s     ayF2J-#----r?   c                 @     |       xr | j                  d      dk(  S )N'   )count)rF   rL   s    r@   
is_rfc5987z-parse_content_disposition.<locals>.is_rfc5987Z   s!    :FLL$5$::r?   c                 $    | j                  d      S )N*)endswithrK   s    r@   is_extended_paramz4parse_content_disposition.<locals>.is_extended_param]   s    s##r?   c                     | j                  d      dz   }|sy| j                  d      r| |d n| |d  }|j                         S )NrW   r   FrN   )findrX   isdigit)rF   pos	substrings      r@   is_continuous_paramz6parse_content_disposition.<locals>.is_continuous_param`   sH    kk#"&,ooc&:F3rNst	  ""r?    )charstextra   c                8    t        j                  d| dd|       S )Nz\\([z])z\1)resub)rb   ra   s     r@   unescapez+parse_content_disposition.<locals>.unescapeg   s    vvugR(%66r?   ;r   =r   rN   rR   rS   utf-8strictTFz\/)strrI   joinmaprd   escaper   splitwarningswarnr4   poplowerstriplstripr5   r   UnicodeDecodeError)rC   rP   rU   rY   r_   rf   disptypepartsparamsitemkeyvalueencoding_failed_valuerL   s                   @r@   r6   r6   Q   s   5 5 5.# .$ .;3 ;4 ;$# $$ $#C #D # -/GGC		44H,I 7s 7c 7c 7 Rx||C(HuH1&9:RxF
yy|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y | >>V##5 & 8>?s   ;K; ;'L&%L&ry   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rH   )
startswith).0r{   r|   name_sufs      r@   	<genexpr>z/content_disposition_filename.<locals>.<genexpr>   s(      
+9ZS%S^^H=US%L>s   &	&rW   r   rN   r`   rR   rS   ri   rj   )	sorteditems	enumeratero   rX   rk   appendrl   r   )ry   r   rx   fnparamsnumr{   r|   r~   tailr}   r   s             @r@   r7   r7      s    t|H	V	h	d| 
+1<<>
 
 "+8!4C#uiiQ'GAt}}S!CRys3xU# "5 %<!&S!!4Ha*7H5(H55r?   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.
    rD   Nc                      || _         || _        y rH   )respstream)selfr   r   s      r@   __init__z!MultipartResponseWrapper.__init__   s    
 	r?   c                     | S rH   r>   r   s    r@   	__aiter__z"MultipartResponseWrapper.__aiter__       r?   )r2   r0   c                 R   K   | j                          d {   }|t        |S 7 wrH   nextStopAsyncIterationr   parts     r@   	__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    r@   r   zMultipartResponseWrapper.at_eof   s    yy  ''))r?   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   rz   s     r@   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    r@   r   z MultipartResponseWrapper.release   s     
 ii!!!s   (&()r   r8   r   r2   rD   N)rD   r   rD   N)r;   r<   r=   __doc__r   r   r   r   rI   r   r   r   r   r>   r?   r@   r   r      si     " 
		2	3* *	%;<	="r?   r   c                      e Zd ZdZdZddddeddd	ed
ede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)+r0   z&Multipart reader for single body part.i    mixedNsubtypedefault_charsetboundaryheadersCIMultiDictProxy[str]r   r   r   rD   c                ^   || _         || _        t        |      dz   | _        || _        || _        d| _        |dk(  | _        | j                  rd n| j                   j                  t        d       }|t        |      nd | _        d| _        t               | _        d | _        d| _        i | _        y )NrS   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)r   r   r   r   r   r   lengths          r@   r   zBodyPartReader.__init__	  s     ! ]Q. /$3++1A1A.RV1W&,&8s6{d%*W,0&(r?   r   c                     | S rH   r>   r   s    r@   r   zBodyPartReader.__aiter__"  r   r?   c                 R   K   | j                          d {   }|t        |S 7 wrH   r   r   s     r@   r   zBodyPartReader.__anext__%  s)     YY[ <$$ !r   c                 H   K   | j                          d {   }|sy |S 7 	wrH   )readr   s     r@   r   zBodyPartReader.next+  s$     YY[  !s   " 
"Fdecoder   c                   K   | j                   ryt               }| j                   s?|j                  | j                  | j                         d{          | j                   s?|r| j                  |      S |S 7 *w)zReads body part data.

        decode: Decodes data following by encoding
                method from Content-Encoding header. If it missed
                data remains untouched
        r?   N)r   	bytearrayextend
read_chunk
chunk_sizer   )r   r   datas      r@   r   zBodyPartReader.read1  sa      <<{,,KKdoodoo>>? ,,;;t$$ ?s   AA<A:A<%A<sizec                 @  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                   r.| j                  j!                          d{   }d|k(  sJ d       |S 7 7 7 7 w)	zWReads body part content chunk of the specified size.

        size: chunk size
        r?   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   rs   rl   ro   r   r   r   r   r   r   readline)	r   r   chunkr}   stripped_chunk	remainderover_chunk_size
over_chunkclrfs	            r@   r   zBodyPartReader.read_chunkA  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<<//11D4EDEK =<$ #P 2sN   .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       r@   r   z&BodyPartReader._read_chunk_from_lengthp  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(| j                  j                  |       d {   | _        d}t	        |      | j                   k  r|| j                  j                  |       d {   z  }| xj
                  t        | j                  j                               z  c_        | j
                  dk  sJ 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        ||kD  r| j                  d | | _        |t	        | j                        | }|sd	| _        | j                  }|| _        |S 7 7 # 1 sw Y   ^xY ww)
Nz<Chunk size must be greater or equal than boundary length + 2r?   r,   zReading after EOFr   r   ignore)categoryT)r   r   r   r   r   r   r   r   unread_datar   r[   maxrp   catch_warningsfilterwarningsDeprecationWarningr   )r   r   first_chunkr   windowre   idxresults           r@   r   z&BodyPartReader._read_chunk_from_streamz  s1     D&&&	JI	J&&&$.%)]]%7%7%==D %j4---4==--d333ET]]%9%9%;!<<$$q(=*==(   %j4--- u:MM%%eDEl3%4LE+++!!E)&++c"C++c3q#d.>.>*?#c(*J#KLC!8((*'';MN))&,7 + cz#'#3#3DS#9 3t//037E#!! E  > 4$ +*sF   AI+IAI+
IA7I+CI+:IAI+I+I($I+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.r?   Nr      --T)	r   r   popleftr   r   r   r   rstripr   )r   linesliner   last_boundary	next_lines         r@   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    r@   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   Nri   default)r   get_charsetr   r   r}   r   s      r@   rb   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   Nri   r   )	r   r   r   r   rk   r   jsonloadsr   r   s      r@   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   Nri   r   z'data cannot be decoded with %s encoding)keep_blank_valuesr}   )r   r   r   r   rv   
ValueErrorr   )r   r}   r   real_encodingdecoded_datas        r@   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    r@   r   zBodyPartReader.at_eof      ||r?   r   c                     t         | j                  v r| j                  |      }| j                  s#t        | j                  v r| j                  |      S |S )zDecodes data.

        Decoding is done according the specified Content-Encoding
        or Content-Transfer-Encoding headers value.
        )r   r   _decode_content_transferr   r   _decode_content)r   r   s     r@   r   zBodyPartReader.decode  sK     %4006D!!&6$,,&F''--r?   c                     | j                   j                  t        d      j                         }|dk(  r|S |dv rt	        |d      j                  |      S t        d|       )Nr`   identity>   gzipdeflateT)r}   suppress_deflate_headerunknown content encoding: )r   r   r   rs   r   decompress_syncRuntimeErrorr   r   r}   s      r@   r  zBodyPartReader._decode_content  sl    <<##$4b9??Az!K**#!(, od#$
 7zBCCr?   c                     | j                   j                  t        d      j                         }|dk(  rt	        j
                  |      S |dk(  rt        j                  |      S |dv r|S t        d|       )Nr`   r   quoted-printable)binary8bit7bit#unknown content transfer encoding: )	r   r   r   rs   r   	b64decodebinasciia2b_qpr  r  s      r@   r
  z'BodyPartReader._decode_content_transfer  st    <<##$=rBHHJx##D))++??4((33K!DXJOPPr?   r   c                     | j                   j                  t        d      }t        |      }|j                  j                  d| j
                  xs |      S )z>Returns charset parameter from Content-Type header or default.r`   charset)r   r   r   r!   
parametersr   )r   r   ctypemimetypes       r@   r   zBodyPartReader.get_charset  sF      r2!%(""&&y$2G2G2R7SSr?   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   r6   r   r   r   r7   r   r~   ry   s      r@   r   zBodyPartReader.name  s/     .dll.>.>?R.ST	6+FF;;r?   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      r@   r'  zBodyPartReader.filename&  s/     .dll.>.>?R.ST	6+FJ??r?   r   )%r;   r<   r=   r   r   bytesr+   rk   r   r   r.   r   r   r   rI   r   r   r   r   r   r   r   rb   r   r   r   r
   r   r  r   r   r  r
  r   r"   r   r'  r>   r?   r@   r0   r0     s   0J )-)) )) 	) ) "#) 
)2   HUO  ,1 D U   ,6 -S -% -^# % *# *% *X :3 7; %hsm %s % 7; Ghsm GxSRUX?W G 7; 
hsm 
tE#s(O?T 
( 5 U 
DE 
De 
D
QU 
Qu 
QT3 T3 T <hsm < < @(3- @ @r?   )orderc                   b     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eddfdZ	 xZ
S )BodyPartReaderPayloadr   r|   argskwargsrD   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|   r,  r-  ry   	__class__s        r@   r   zBodyPartReaderPayload.__init__4  si    000!#::!"ZZF6N>>%!&F:(D((FvF r?   r}   errorsc                     t        d      )NzUnable to decode.)	TypeErrorr   r}   r3  s      r@   r   zBodyPartReaderPayload.decode@  s    +,,r?   writerc                    K   | j                   }|j                  d       d {   }|rF|j                  |j                  |             d {    |j                  d       d {   }|rEy y 7 M7 %7 w)Ni   )r   )r   r   writer   )r   r7  fieldr   s       r@   r9  zBodyPartReaderPayload.writeC  sk     &&E&22,,u||E2333***66E  336s9   "A8A2)A8A4A8(A6)A80A84A86A8ri   rj   )r;   r<   r=   r0   __annotations__r   r   rk   r   r9  __classcell__r2  s   @r@   r+  r+  0  s\    
Gn 
GS 
GC 
GD 
G-s -c - -7# 7$ 7r?   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)r2   zMultipart body reader.Nmultipart_reader_clsr   r   rD   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      r@   r   zMultipartReader.__init__V  s    '(=>~~""k1V3VV1T^^66669NN  !3!3!55==?/3NR$&r?   r   c                     | S rH   r>   r   s    r@   r   zMultipartReader.__aiter__g  r   r?   c                 R   K   | j                          d {   }|t        |S 7 wrH   r   r   s     r@   r   zMultipartReader.__anext__j  r   r   responser8   c                 `    | j                  | | |j                  |j                              }|S )zConstructs reader instance from HTTP response.

        :param response: :class:`~aiohttp.client.ClientResponse` instance
        )response_wrapper_clsr   r   )clsrL  objs      r@   from_responsezMultipartReader.from_responser  s3     &&c(**H,<,<=
 
r?   c                     | j                   S )z@Returns True if the final boundary was reached, false otherwise.r  r   s    r@   r   zMultipartReader.at_eof  r  r?   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_partrI  _read_until_first_boundary_read_boundaryfetch_next_partrH  rD  r   
isinstancer0   r6   r   r   r   r   r   r  rt   r   r   )r   r   r~   ry   r  s        r@   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     r@   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     r@   rZ  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
        r`   rB  r   )r   r   r!   rE  r@  r   part_reader_clsr   rD  r   r   )r   r   r!  r"  s       r@   r_  z MultipartReader._get_part_reader  s     L"-!%(==K'((0!tDz'4==99,,WdmmDD''.. $ 5 5 (  r?   c                 p    | j                   j                  d   }t        |      dkD  rt        d|z        |S )Nr   F   z&boundary %r is too long (70 chars max))rD  r   r   r  )r   r   s     r@   rF  zMultipartReader._get_boundary  s8    >>,,Z8x=2EPQQr?   c                    K   | j                   r| j                   j                         S | j                  j                          d {   S 7 wrH   )r   rr   r   r   r   s    r@   	_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)NTr?   z#Could not find starting boundary %rr   )re  r  r   r   r   r   r   s     r@   rX  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   TrS   zInvalid boundary z, expected )re  r   r   r   r   r   r   r  )r   r   epiloguer   s       r@   rY  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   dg}	 | j                   j                          d {   }|j                         }|j                  |       |snGt	               }|j                  |      \  }}|S 7 Iw)Nr?   )r   r   rt   r   r#   parse_headers)r   linesr   parserr   raw_headerss         r@   r^  zMultipartReader._read_headers  so     --0022EKKME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)rH  r   r   r   r   r   s    r@   rW  z(MultipartReader._maybe_release_last_part  sa     ??&??))+oo--///LL 7 78"DO	 '/s   ABB=Br   )rD   r   )#r;   r<   r=   r   r   rN  r@  r   r   r<  r0   ra  r   rk   r+   r   r.   r   r   r   classmethodrQ  rI   r   r   r   rZ  r_  rF  r(  re  rX  rY  r^  rW  r>   r?   r@   r2   r2   K  s:     4 ?C(4(9#:;B$O'S 1 'L 'T '"  	%)>9:	; " 
"  !	%)>9:	;!F!.	 .0	1.( 
 .0	14s . .
Y.
#r?   r2   c                       e Zd ZU dZded<   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dfd!Z' xZ(S )'r3   zMultipart body writer.Nr   r   r   rD   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hexrG  r   UnicodeEncodeErrorr  _boundary_valuer0  r   _partsr   )r   r   r   r!  r2  s       r@   r   zMultipartWriter.__init__  s    '389I9I	S%__W5DN WI[1E1E0FGE2#%$3 " 	SGHdR	Ss   A0 0Bc                     | S rH   r>   r   s    r@   	__enter__zMultipartWriter.__enter__0  r   r?   exc_typeexc_valexc_tbc                      y rH   r>   )r   r}  r~  r  s       r@   __exit__zMultipartWriter.__exit__3  s     	r?   c                 ,    t        | j                        S rH   )iterrz  r   s    r@   __iter__zMultipartWriter.__iter__;  s    DKK  r?   c                 ,    t        | j                        S rH   )r   rz  r   s    r@   __len__zMultipartWriter.__len__>  s    4;;r?   c                      y)NTr>   r   s    r@   __bool__zMultipartWriter.__bool__A  s    r?   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.
        rs  z*boundary value contains invalid characters   \s   \\   "s   \"rO   )	r   rd   match_valid_tchar_regexr   search_invalid_qdtext_char_regexr  replace)r   r|   quoted_value_contents      r@   ry  zMultipartWriter._boundary_valueG  s    & 88D++U3<<((99T44e<IJJ  %}}UG<3;;D&I)0099C??r?   c                 8    | j                   j                  d      S )Nrs  )r   r   r   s    r@   r   zMultipartWriter.boundaryg  s    ~~$$W--r?   rP  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   r[  r'   r   updateappend_payloadr)   r%   r5  )r   rP  r   payloads       r@   r   zMultipartWriter.appendk  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   r`   )r  r  r  r  r  )r`   r   r  r  r  r  )r   r   r   r   r   keysr   r   rz  r1  r   rs   r  r   rk   r   )r   r  r}   te_encodingr   r   s         r@   r  zMultipartWriter.append_payloadz  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{;<r?   c                 R    |
t               }| j                  t        ||            S )zHelper to append JSON part.r  )r   r  r$   )r   rP  r   s      r@   append_jsonzMultipartWriter.append_json  s(     ?!mG"";sG#DEEr?   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   rt  )	r[  r   r   r   listr   r   r  r(   )r   rP  r   r   s       r@   append_formzMultipartWriter.append_form  sn     #'2333?!mGc7#syy{#CD)""g4W
 	
r?   c                 .   d}| j                   D ]e  \  }}}|s|s|j                   y|t        dt        | j                        z   dz   |j                  z   t        |j
                        z   dz         z  }g |dt        | j                        z   dz   z  }|S )zSize of the payload.r   NrS   r   )rz  r   r   r   r   _binary_headers)r   totalr   r}   r  s        r@   r   zMultipartWriter.size  s     +/;;'D(K;$))*;Sdnn%& )) d**+	,
  E	 ,7 	S((1,,r?   r}   r3  c                 T     dj                   fd j                  D              S )Nr`   c              3      K   | ]H  \  }}}d j                   z   dz   |j                  j                        z   |j                         z    J yw)rC  
N)r   r  r   )r   r   _e_ter}   r3  r   s       r@   r   z)MultipartWriter.decode.<locals>.<genexpr>  sd      
 "-b# mm ""))(F;< kkm	
 "-s   AA)rl   rz  r6  s   ```r@   r   zMultipartWriter.decode  s'    ww 
 "&
 
 	
r?   r7  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   Ns   --
)rz  r   r   r   r9  r   r  MultipartPayloadWriterenable_compressionenable_encoding	write_eof)r   r7  r  r   r}   r  ws          r@   r9  zMultipartWriter.write  sH    +/;;'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''') ,7, ,,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)r   N)rD   r3   rH   r;  )T))r;   r<   r=   r   r<  rk   r   r   r|  r   BaseExceptionr   r  r	   _Partr  r   r  rI   r  rd   compiler  r  propertyry  r   r   r   r'   r   r  r  r   r   r   r  r   r   r9  r=  r>  s   @r@   r3   r3     s    L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 
 
C# Ct Ct Cr?   r3   c                   r    e Zd ZdeddfdZdeddfdZdej                  fdede	ddfd	Z
dd
ZdeddfdZy)r  r7  rD   Nc                 <    || _         d | _        d | _        d | _        y rH   )_writer	_encoding	_compress_encoding_buffer)r   r7  s     r@   r   zMultipartPayloadWriter.__init__  s    (,3759r?   r}   c                 V    |dk(  r|| _         t               | _        y |dk(  rd| _         y y )Nr   r  )r  r   r  )r   r}   s     r@   r  z&MultipartPayloadWriter.enable_encoding  s0    x%DN$-KD!++/DN ,r?   r  strategyc                 *    t        |d|      | _        y )NT)r}   r  r  )r   r  )r   r}   r  s      r@   r  z)MultipartPayloadWriter.enable_compression  s     ($(
r?   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  flushr9  r  r  r  r   	b64encoderg  s     r@   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  r9  r  b2a_qp)r   r   bufdivmod	enc_chunkb64chunks          r@   r9  zMultipartPayloadWriter.write  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   )r;   r<   r=   r   r   rk   r  zlibZ_DEFAULT_STRATEGYr   r  r  r(  r9  r>   r?   r@   r  r    sk    :s :t :0 0 0 (9P9P

36
	
	R, ,4 ,r?   r  )r'  )Or   r  r   rd   sysru  rp   r  collectionsr   typesr   typingr   r   r   r   r	   r
   r   r   r   r   r   r   r   urllib.parser   r   r   	multidictr   r   compression_utilsr   r   hdrsr   r   r   r   r   helpersr   r    r!   r"   httpr#   r  r$   r%   r&   r'   r(   r)   r*   streamsr+   version_infor.   r/   __all__client_reqrepr8   RuntimeWarningr4   r5   rk   r6   r7   r   r0   	try_firstr+  r2   r  r3   r  r>   r?   r@   <module>r     s      	 
         7 6 3 ?  8 7    "w6!12D -	. 		 	`$SM`$
8C=$sCx.()`$H ,6CH%(c]B," ,"^i@ i@X	 nEOO47G 7 574H# H#V 	gsC \Cg \C~7, 7,r?   