
    3Cfy                        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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: dZ;erddl<m=Z=  G d de>      Z? G d de>      Z@deeA   deeeA   eeAeAf   f   fdZB	 d'deeAeAf   deAdeeA   fdZC G d d      ZD G d d      ZE e8eEe4j                         G d d e5             ZG G d! d"      ZHee5eAeAf   ZI G d# d$e5      ZJ G d% d&      ZKy)(    N)deque)TracebackType)TYPE_CHECKINGAnyAsyncIterator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)MultipartReaderMultipartWriterBodyPartReaderBadContentDispositionHeaderBadContentDispositionParamparse_content_dispositioncontent_disposition_filename)ClientResponsec                       e Zd Zy)r0   N__name__
__module____qualname__     Q/var/www/highfloat_scraper/venv/lib/python3.12/site-packages/aiohttp/multipart.pyr0   r0   B       r;   r0   c                       e Zd Zy)r1   Nr6   r:   r;   r<   r1   r1   F   r=   r;   r1   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stringr@   c                 @    t        |       xr t        t        |       k\  S N)boolr!   setrB   s    r<   is_tokenz+parse_content_disposition.<locals>.is_tokenM   s    F|4V 44r;   c                 .    | d   | d   cxk(  xr dk(  S c S )Nr   "r:   rG   s    r<   	is_quotedz,parse_content_disposition.<locals>.is_quotedP   s     ayF2J-#----r;   c                 @     |       xr | j                  d      dk(  S )N'   )count)rB   rH   s    r<   
is_rfc5987z-parse_content_disposition.<locals>.is_rfc5987S   s!    :FLL$5$::r;   c                 $    | j                  d      S )N*)endswithrG   s    r<   is_extended_paramz4parse_content_disposition.<locals>.is_extended_paramV   s    s##r;   c                     | j                  d      dz   }|sy| j                  d      r| |d n| |d  }|j                         S )NrS   r   FrJ   )findrT   isdigit)rB   pos	substrings      r<   is_continuous_paramz6parse_content_disposition.<locals>.is_continuous_paramY   sH    kk#"&,ooc&:F3rNst	  ""r;    )charstextr]   c                8    t        j                  d| dd|       S )Nz\\([z])z\1)resub)r^   r]   s     r<   unescapez+parse_content_disposition.<locals>.unescape`   s    vvugR(%66r;   ;r   =r   rJ   rN   rO   utf-8strictTFz\/)strrE   joinmapr`   escaper    splitwarningswarnr0   poplowerstriplstripr1   r   UnicodeDecodeError)r?   rL   rQ   rU   r[   rb   disptypepartsparamsitemkeyvalueencoding_failed_valuerH   s                   @r<   r2   r2   J   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&ru   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rD   )
startswith).0rw   rx   name_sufs      r<   	<genexpr>z/content_disposition_filename.<locals>.<genexpr>   s)      
'S%S^^H=US%L
s   &	&rS   r   rJ   r\   rN   rO   re   rf   )	sorteditems	enumeraterk   rT   rg   appendrh   r   )ru   r}   rt   fnparamsnumrw   rx   rz   tailry   r   s             @r<   r3   r3      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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.
    r@   Nc                      || _         || _        y rD   )respstream)selfr   r   s      r<   __init__z!MultipartResponseWrapper.__init__   s    
 	r;   c                     | S rD   r:   r   s    r<   	__aiter__z"MultipartResponseWrapper.__aiter__       r;   )r-   r/   c                 R   K   | j                          d {   }|t        |S 7 wrD   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   rv   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   r4   r   r-   r@   N)r@   r   r@   N)r7   r8   r9   __doc__r   r   r   r   rE   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    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)*r/   z&Multipart reader for single body part.i    mixedNsubtypedefault_charsetboundaryheadersCIMultiDictProxy[str]r   r   r   r@   c                8   || _         || _        || _        || _        d| _        |dk(  | _        | j
                  rd n| j                   j                  t        d       }|t        |      nd | _	        d| _
        t               | _        d | _        d| _        i | _        y )NF	form-datar   )r   	_boundary_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     ! /$3++1A1A.RV1W&,&8s6{d%*W,0&(r;   c                     | S rD   r:   r   s    r<   r   zBodyPartReader.__aiter__  r   r;   c                 R   K   | j                          d {   }|t        |S 7 wrD   r   r   s     r<   r   zBodyPartReader.__anext__  s)     YY[ <$$ !r   c                 H   K   | j                          d {   }|sy |S 7 	wrD   )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.read)  sa      <<{,,KKdoodoo>>? ,,;;t$$ ?s   AA<A:A<%A<sizec                   K   | j                   ry| j                  r| j                  |       d{   }n| j                  |       d{   }| xj                  t        |      z  c_        | j                  | j                  k(  rd| _         | j                   r.| j                  j                          d{   }d|k(  sJ d       |S 7 7 7 w)zWReads body part content chunk of the specified size.

        size: chunk size
        r;   NT   
z3reader did not read all the data or it is malformed)r   r   _read_chunk_from_length_read_chunk_from_streamr   lenr   readline)r   r   chunkclrfs       r<   r   zBodyPartReader.read_chunk9  s     
 <<<<66t<<E66t<<ECJ&t||+DL<<//11D4EDE =< 2s4   .CC	C
CA+C6C7CCC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_lengthO  sq      ||'S)SS't||d.>.>>?
mm((44==!DL 5s   ABA?(Bc           
      
  K   |t        | j                        dz   k\  sJ d       | j                  d u }|r(| j                  j	                  |       d {   | _        | j                  j	                  |       d {   }| xj
                  t        | j                  j                               z  c_        | j
                  dk  sJ 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 y# 1 sw Y   ^xY ww)
NrO   z<Chunk size must be greater or equal than boundary length + 2   zReading after EOFr   r   ignore)categoryT)r   r   r   r   r   r   r   r   rW   maxrl   catch_warningsfilterwarningsDeprecationWarningunread_datar   )r   r   first_chunkr   windowra   idxresults           r<   r   z&BodyPartReader._read_chunk_from_streamY  s     C'!++	JI	J+&&$.%)]]%7%7%==Dmm((..S!5!5!788  1$9&99$+++!!E)&++c"C++c3q#d.>.>*?#c(*J#KLC!8((* 8'';MN))&,78 cz#'#3#3DS#9 3t//037E#!! 1  >.8 8s>   AHG1'H;G4<C$H :G7AH4H7H <H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)ry   ry   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   Nre   default)r   get_charsetr   r   ry   r   s      r<   r^   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   Nre   r   )	r   r   r   r	   rg   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   Nre   r   z'data cannot be decoded with %s encoding)keep_blank_valuesry   )r   r   r   r   rr   
ValueErrorr   )r   ry   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)ry   suppress_deflate_headerunknown content encoding: )r   r   r   ro   r   decompress_syncRuntimeErrorr   r   ry   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j                  |            )Nr\   base64quoted-printable)binary8bit7bitz%unknown content transfer encoding: {})
r   r   r   ro   r  	b64decodebinasciia2b_qpr	  formatr
  s      r<   r   z'BodyPartReader._decode_content_transfer  s|    <<##$=rBHHJx##D))++??4((33K:AA(K r;   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}   r2   r   r   r   r3   r   rz   ru   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   )%r7   r8   r9   r   r   bytesr,   rg   r   r   r   r   r   r   rE   r   r   r   r   r   r   r   r^   r	   r   r   r   r   r   r   r   r  r   r   r#   r}   r  r:   r;   r<   r/   r/      s   0J )-)) )) 	) ) "#) 
)0=)9:  HUO  ,1 D U   ,6 S % ,# %  #  %  D :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U u T3 T3 T <hsm < < @(3- @ @r;   r/   )orderc                   @     e Zd Zdedededdf fdZdeddfdZ xZS )	BodyPartReaderPayloadrx   argskwargsr@   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   rx   r#  r$  ru   	__class__s        r<   r   zBodyPartReaderPayload.__init__	  si    000!#::!"ZZF6N>>%!&F:(D((FvF 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   r*  fieldr   s       r<   r,  zBodyPartReaderPayload.write  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)r7   r8   r9   r/   r   r   r,  __classcell__r)  s   @r<   r"  r"    s<    
Gn 
GS 
GC 
GD 
G7# 7$ 7r;   r"  c                      e Zd ZdZeZdZeZde	e
e
f   deddfdZded   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)r-   zMultipart body reader.Nr   r   r@   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: %sz--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__(  s    '(=>~~""k1V3VV1T^^66669NN  !3!3!55==?/3NR$&r;   r/   c                     | S rD   r:   r   s    r<   r   zMultipartReader.__aiter__9  s	     r;   c                 R   K   | j                          d {   }|t        |S 7 wrD   r   r   s     r<   r   zMultipartReader.__anext__>  r   r   responser4   c                 `    | j                  | | |j                  |j                              }|S )zConstructs reader instance from HTTP response.

        :param response: :class:`~aiohttp.client.ClientResponse` instance
        )response_wrapper_clsr   r   )clsr;  objs      r<   from_responsezMultipartReader.from_responseF  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_eofT  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_partr8  _read_until_first_boundary_read_boundaryfetch_next_partr7  r3  r   
isinstancer/   r2   r   r   r   r   r   r	  rp   r   r   )r   r   rz   ru   r  s        r<   r   zMultipartReader.nextX  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<   rI  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\   r2  r   )r   r   r"   r4  multipart_reader_clsr   part_reader_clsr   r3  r   r   )r   r   r  r  s       r<   rN  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))r3  r  r   r   )r   r   s     r<   r5  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rD   )r   rn   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   )rU  r   r   r   r   r   r   s     r<   rG  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   TrO   zInvalid boundary z, expected )rU  r   r   r   r   r   r   r   )r   r   epiloguer   s       r<   rH  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   rp   r   r$   parse_headers)r   linesr   parserr   raw_headerss         r<   rM  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)r7  r   r   r   r   r   s    r<   rF  z(MultipartReader._maybe_release_last_part  sa     ??&??))+oo--///LL 7 78"DO	 '/s   ABB=Br   )r@   r   )!r7   r8   r9   r   r   r=  rP  r/   rQ  r   rg   r,   r   r   r   r   r   r   classmethodr@  rE   r   r   r   rI  rN  r5  r  rU  rG  rH  rM  rF  r:   r;   r<   r-   r-     s)     4  $O'S 1 'L 'T '"	'	(
	%)>9:	; " 
"  !	%)>9:	;!F!.	 .0	1.( 
 .0	14s . .
Y.
#r;   r-   c                       e Zd Z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dfdZ% xZ&S )"r.   zMultipart body writer.Nr   r   r@   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hexr6  r   UnicodeEncodeErrorr   _boundary_valuer'  r   _partsr   )r   r   r   r  r)  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 rD   r:   r   s    r<   	__enter__zMultipartWriter.__enter__  r   r;   exc_typeexc_valexc_tbc                      y rD   r:   )r   rm  rn  ro  s       r<   __exit__zMultipartWriter.__exit__  s     	r;   c                 ,    t        | j                        S rD   )iterrj  r   s    r<   __iter__zMultipartWriter.__iter__  s    DKK  r;   c                 ,    t        | j                        S rD   )r   rj  r   s    r<   __len__zMultipartWriter.__len__  s    4;;r;   c                      y)NTr:   r   s    r<   __bool__zMultipartWriter.__bool__  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.
        rc  z*boundary value contains invalid characters   \s   \\   "s   \"rK   )	r   r`   match_valid_tchar_regexr   search_invalid_qdtext_char_regexr   replace)r   rx   quoted_value_contents      r<   ri  zMultipartWriter._boundary_value  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 )Nrc  )r   r   r   s    r<   r   zMultipartWriter.boundary9  s    ~~$$W--r;   r?  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   rJ  r(   r   updateappend_payloadr*   r&   	TypeError)r   r?  r   payloads       r<   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}   r\   )r  r  r  r  r  )r\   r  r  r  z#unknown content transfer encoding: r  )r   r   r   r   r   keysr   r   rj  r(  r   ro   r	  r   rg   r   )r   r  ry   te_encodingr}   r   s         r<   r  zMultipartWriter.append_payloadL  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   r?  r   s      r<   append_jsonzMultipartWriter.append_jsonr  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   rd  )	rJ  r   r   r   listr   r   r  r)   )r   r?  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   NrO      )rj  r   r   r   r   _binary_headers)r   totalr   ry   r  s        r<   r   zMultipartWriter.size  s     +/;; 	'D(K;$))*;Sdnn%& )) d**+	,
  E		 	S((1,,r;   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   Ns   --
)rj  r   r   r   r,  r   r  MultipartPayloadWriterenable_compressionenable_encoding	write_eof)r   r*  r  r   ry   r  ws          r<   r,  zMultipartWriter.write  sM    +/;; 	('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)r   N)r@   r.   rD   )T)'r7   r8   r9   r   rg   r   r   rl  r   BaseExceptionr   rq  r
   _Partrt  r   rv  rE   rx  r`   compiler}  r  propertyri  r   r   r   r(   r   r  r  r   r   r   r  r   r,  r.  r/  s   @r<   r.   r.     s    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  &C# Ct Ct Cr;   r.   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  r*  r@   Nc                 <    || _         d | _        d | _        d | _        y rD   )_writer	_encoding	_compress_encoding_buffer)r   r*  s     r<   r   zMultipartPayloadWriter.__init__  s    (,3759r;   ry   c                 V    |dk(  r|| _         t               | _        y |dk(  rd| _         y y )Nr  r  )r  r   r  )r   ry   s     r<   r  z&MultipartPayloadWriter.enable_encoding  s0    x%DN$-KD!++/DN ,r;   r  strategyc                 *    t        |d|      | _        y )NT)ry   r  r  )r   r  )r   ry   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  flushr,  r  r  r  r  	b64encoderW  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  r,  r  b2a_qp)r   r   bufdivmod	enc_chunkb64chunks          r<   r,  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   )r7   r8   r9   r   r   rg   r  zlibZ_DEFAULT_STRATEGYr   r  r  r  r,  r:   r;   r<   r  r    sk    :s :t :0 0 0 (9P9P

36
	
	R, ,4 ,r;   r  )r  )Lr  r  r   r`   re  rl   r  collectionsr   typesr   typingr   r   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,   __all__client_reqrepr4   RuntimeWarningr0   r1   rg   r2   r3   r   r/   	try_firstr"  r-   r  r.   r  r:   r;   r<   <module>r     sp      	          7 6 3 ?  8 7    " -	. 		 	`$SM`$
8C=$sCx.()`$H ,6CH%(c]B," ,"^G@ G@T nEOO47G 7 57*J# J#Z 	gsC PCg PCf7, 7,r;   