
    tni"                     R   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	 d0d!e
eIeIf         d"eIdeeI         fd#ZK G d$ d%          ZL G d& d          ZM e=eMe9jN        '           G d( d)e:                      ZO G d* d+          ZPee:eIeIf         ZQ G d, d-e:          ZR G d. d/          ZSdS )1    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dS )r7   N__name__
__module____qualname__     N/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/aiohttp/multipart.pyr7   r7   M           DrB   r7   c                       e Zd ZdS )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                    t          t          j        t                              d	d
t           dt           dt           fd}| sd i fS |                     d          ^}} |          s%t          j	        t          |                      d i fS i }|r|                    d          }	|	s"t          j	        t          |                      <d|	vr%t          j	        t          |                      d i fS |	                    dd          \  }
}|
                                                                }
|                                }|
|v r%t          j	        t          |                      d i fS  |
          s#t          j	        t          |	                      ||
          rP ||          r ||dd                   }n |          s#t          j	        t          |	                     hn_ ||
          r ||          r|                    dd          \  }}}|pd}n#t          j	        t          |	                     	 t!          ||d          }n# t"          $ r% t          j	        t          |	                     Y w xY wd} ||          r)d} ||dd                             d                    }ne |          rd}nW|rU| d|d          } ||          r=|                    d            ||dd                             d                    }d}|r%t          j	        t          |                      d i fS |||
<   ||                                |fS )NstringrG   c                 P    t          |           o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    odk    nc S )Nr   "rA   rN   s    rC   	is_quotedz,parse_content_disposition.<locals>.is_quoted[   s,    ayF2J----#-----rB   c                 L     |           o|                      d          dk    S )N'   )count)rI   rO   s    rC   
is_rfc5987z-parse_content_disposition.<locals>.is_rfc5987^   s)    x:FLL$5$5$::rB   c                 ,    |                      d          S )N*)endswithrN   s    rC   is_extended_paramz4parse_content_disposition.<locals>.is_extended_parama   s    s###rB   c                     |                      d          dz   }|sdS |                     d          r
| |d         n	| |d          }|                                S )NrZ   r   FrQ   )findr[   isdigit)rI   pos	substrings      rC   is_continuous_paramz6parse_content_disposition.<locals>.is_continuous_paramd   sa    kk#" 	5&,ooc&:&:LF3r6NNstt	  """rB    )charstextrd   c                6    t          j        d| dd|           S )Nz\\([z])z\1)resub)re   rd   s     rC   unescapez+parse_content_disposition.<locals>.unescapek   s"    v(u(((%6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 5 5 5.# .$ . . . .;3 ;4 ; ; ; ; ; ;$# $$ $ $ $ $#C #D # # # # -/GGC	44H4H,I,I 7 7 7s 7c 7c 7 7 7 7  Rx||C((Hu8H 1&99:::RxF
 @yy|| 	M5f==>>>d??M5f==>>>8OZZQ''
Uiikk!!&==M5f==>>>8Ox}} +	 M4T::;;;  %% '	 y  qt--Xe__ 8>>??? s##  	 z%   %*[[a%8%8"!U#.w8>>???x::%   8>>???
 Fy # qt!3!3E!:!:;;% 	# # "..E!H..9V$$ #IIaLLL$HVAbD\%8%8%?%?@@E"F  9&AABBBRxsA  @D >>V##s   <K +K=<K=filenamer|   namec                 2  
 d|z  
| sd S 
| v r| 
         S || v r| |         S g }t          
fd|                                 D                       }t          |          D ]i\  }\  }}|                    dd          \  }}|                    d          r
|d d         }|t          |          k    r|                    |           i |sd S d                    |          }d|v r/|                    dd          \  }	}}|	pd	}	t          ||	d
          S |S )Nz%s*c              3   P   K   | ] \  }}|                               ||fV  !d S rK   )
startswith).0r~   r   name_sufs      rC   	<genexpr>z/content_disposition_filename.<locals>.<genexpr>   sP       
 
'S%S^^H=U=U
%L
 
 
 
 
 
rB   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:      si    t|H t	V		h	d| 
 
 
 
+1<<>>
 
 
 
 
 "+8!4!4 	 	C#uiiQ''GAt}}S!! !CRCys3xxU#### 	4%<<!&S!!4!4Ha*7H5(H555rB   c                   ~    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dS )MultipartResponseWrapperzzWrapper around the MultipartReader.

    It takes care about
    underlying connection and close it when it needs in.
    respr;   streamr5   rG   Nc                 "    || _         || _        d S rK   )r   r   )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                 P   K   |                                   d {V }|t          |S rK   nextStopAsyncIterationr   parts     rC   	__anext__z"MultipartResponseWrapper.__anext__   7       YY[[      <$$rB   c                 >    | j         j                                        S )z2Returns True when all response data had been read.)r   contentat_eofr   s    rC   r   zMultipartResponseWrapper.at_eof   s    y '')))rB   c                    K   | j                                          d{V }| j                                         r|                                  d{V  |S )z#Emits next multipart reader object.N)r   r   r   releaser   r}   s     rC   r   zMultipartResponseWrapper.next   sh       [%%''''''''; 	!,,..       rB   c                 H   K   | j                                          d{V  dS )z_Release the connection gracefully.

        All remaining content is read to the void.
        N)r   r   r   s    rC   r   z MultipartResponseWrapper.release  s4      
 i!!!!!!!!!!!rB   )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      s          " 
	      	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%dS ),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                ~   || _         || _        t          |          dz   | _        || _        || _        d| _        |dk    | _        | j        rd n| j                             t          d           }|t          |          nd | _        d| _        t                      | _        d | _        d| _        i | _        || _        d S )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+W1A1A.RV1W1W&,&8s6{{{d%*WW,0&($7!!!rB   r   c                     | S rK   rA   r   s    rC   r   zBodyPartReader.__aiter__,  r   rB   c                 P   K   |                                   d {V }|t          |S rK   r   r   s     rC   r   zBodyPartReader.__anext__/  s5      YY[[      <$$rB   c                 F   K   |                                   d {V }|sd S |S rK   )readr   s     rC   r   zBodyPartReader.next5  s7      YY[[       	4rB   Fdecoder   c                   K   | j         rdS t                      }| j         s:|                    |                     | j                   d{V            | j         :|r|                     |           d{V S |S )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;  s       < 	3{{, 	@KKdoodo>>>>>>>>??? , 	@ 	+T*********rB   sizec                 ^  K   | j         rdS | j        r|                     |           d{V }n|                     |           d{V }| j                            t                    }|rT|                                dk    r;d                    |	                                          }t          |          dz  }|dk    r|                                 sd|z
  }d}| j        r0| j        d|         }| j        t          |          d         | _        t          |          |k    r3|| j                            dt          |          z
             d{V z  }|sd| _         |d                    |	                                          z  }||z  }t          |          dz  }|dk    r|                                 | xj        t          |          z  c_        | j        | j        k    rd| _         | j         r2| j                                         d{V dk    rt#          d          |S )	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  sR     
 < 	3< 	=66t<<<<<<<<EE66t<<<<<<<<E <##$=>> 	4((H44 XXekkmm44NN++a/Iq..."#i- 
# K!%!12B?2B!CJ'+'7J8I8I'JD$z??o55(:(:1s:;N(O(O"O"O"O"O"O"OOJ! (#'DL#((:+;+;+=+=">">>#//!3	! q...$ 	CJJ&t|++DL< 	T$-"8"8":":::::::gEERSSSrB   c                    K   | j         
J d            t          || j         | j        z
            }| j                            |           d {V }| j                                        rd| _        |S )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  s       |'')S'''t|d.>>??
m((44444444=!! 	 DLrB   c           
        K   || j         k    s
J d            | j        d u }|r(d| j                            |           d {V z   | _        d}t	          |          | j         k     r|| j                            |           d {V z  }| xj        t          | j                                                  z  c_        | j        dk    rt          d          | j        rnt	          |          | j         k     t	          |          |k    r,| j        	                    ||d                     |d |         }| j        J | j        |z   }d| j
        z   }|r|                    |          }nF|                    |t          dt	          | j                  t	          |          z
                      }|dk    rt          j                    5  t          j        dt                      | j        	                    ||d                     d d d            n# 1 swxY w Y   | j        d |         | _        |t	          | j                  |         }|sd	| _        | j        |rdndd          }|| _        |S )
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  s      D&&&&I '&&&$. 	H&t}/A/A$/G/G)G)G)G)G)G)GGD %jj4---4=--d333333333ET]%9%9%;%;!<!<< 1$$ !4555   %jj4--- u::M%%eDEEl333%4%LE+++!E)& 	M++c""CC++c3q#d.>*?*?#c((*J#K#KLLC!88(** 8 8';MNNNN))&,7778 8 8 8 8 8 8 8 8 8 8 8 8 8 8  $/5D3t/00367E $#!{"9!!";";< s   	>HHHc                 6  K   | j         rdS | j        r| j                                        }n| j                                         d{V }|                    | j                  rV|                    d          }| j        }| j        dz   }||k    s||k    r#d| _         | j                            |           dS n]| j                                         d{V }|                    | j                  r
|dd         }| j                            |           |S )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  s8     < 	3< 	2<''))DD//11111111D??4>** 	+ KK((E~H NU2M  E]$:$:###D)))s %;
 #m4466666666I##DN33 !CRCyL	***rB   c                 |   K   | j         rdS | j         s)|                     | j                   d{V  | j         'dS dS )z0Like read(), but reads all the data to the void.N)r   r   r   r   s    rC   r   zBodyPartReader.release  se      < 	F, 	3//$/222222222 , 	3 	3 	3 	3 	3rB   )r   r   c                   K   |                      d           d{V }|p|                     d          }|                    |          S )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  sZ      YYdY++++++++ @t///@@{{8$$$rB   c                  K   |                      d           d{V }|sdS |p|                     d          }t          t          t          t
          f         t          j        |                    |                              S )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  s      YYdY++++++++ 	4@t///@@DcNDJt{{8/D/D$E$EFFFrB   c                2  K   |                      d           d{V }|sg S ||}n|                     d          }	 |                                                    |          }n # t          $ r t          d|z            w xY wt          |d|          S )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$MM ,,W,==M	X;;==//>>LL! 	X 	X 	XFVWWW	X ""
 
 
 	
s   'A' '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        s)t          | j        v r|                     |           d{V S |S )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  sk       %440066D! 	4&6$,&F&F--d333333333rB   c                   K   | j                             t          d                                          }|dk    r|S |dv r1t	          |d                              || j                   d {V S t          d|           )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9??AAz!!K***)!(,   j$*CjDDE E E E E E E
 BBBCCCrB   c                     | j                             t          d                                          }|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  s    <##$=rBBHHJJx#D)))+++?4(((333KOXOOPPPrB   r   c                     | j                             t          d          }t          |          }|j                            d| j        p|          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  sG      r22!%(("&&y$2G2R7SSSrB   c                     t          | 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%  s6     .dl.>.>?R.S.STT	6+FF;;;rB   c                     t          | j                            t                              \  }}t	          |d          S )zReturns filename specified in Content-Disposition header.

        Returns None if the header is missing or malformed.
        r   r(  r)  s      rC   r   zBodyPartReader.filename.  s6     .dl.>.>?R.S.ST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0J )-#>8 8 88 )8 	8 8 "#8 !8 
8 8 8 86         HUO     ,1   D U      ,6 * *S *% * * * *X# %    +# +% + + + +Z    :3 3 3 3 7; % % %hsm %s % % % % 7; G G Ghsm GxSRUX?W G G G G 7; 
 
 
hsm 
tE#s(O?T 
 
 
 
(     5    
D% 
DE 
D 
D 
D 
D
QU 
Qu 
Q 
Q 
Q 
QT3 T3 T T T T <hsm < < < U< @(3- @ @ @ U@ @ @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                      j        |g|R i | i }|j        
|j        |d<   |j        
|j        |d<   |r | j        di | d S d S )Nr   r   
attachmentT)r2  T)superr   r   r   set_content_disposition)r   r   r/  r0  r|   	__class__s        rC   r   zBodyPartReaderPayload.__init__=  s    0000000!#:!"ZF6N>%!&F: 	G(D(FFvFFFFF	G 	GrB   rl   rm   r   errorsc                      t          d          )NzUnable to decode.	TypeErrorr   r   r6  s      rC   r   zBodyPartReaderPayload.decodeI  s    +,,,rB   c                 $   K   t          d          )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.r8  r:  s      rC   as_byteszBodyPartReaderPayload.as_bytesL  s       TUUUrB   writerc                    K   | j         }|                    d           d {V }|rT|                    |                    |           d {V            d {V  |                    d           d {V }|Rd S d S )Ni   )r   )r   r   writer   )r   r=  fieldr   s       rC   r?  zBodyPartReaderPayload.writeV  s      &&E&22222222 	7,,U\\%%8%8888888999999999***66666666E  	7 	7 	7 	7 	7rB   rl   rm   )r>   r?   r@   r3   __annotations__r   r   rn   r   r+  r<  r   r?  __classcell__r5  s   @rC   r.  r.  8  s         
Gn 
GS 
GC 
GD 
G 
G 
G 
G 
G 
G- -s -c - - - - -V Vs Vc VQV V V V V7"6 74 7 7 7 7 7 7 7 7rB   r.  c                   n   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"dS )r5   zMultipart body reader.Nmultipart_reader_clsr   r   rG   c                    t          |t                             | _        | j        j        dk    s
J d            d| j        j        vrt          d|t                   z            || _        d|                                 z                                   | _	        || _
        d | _        d | _        d| _        d| _        g | _        d S )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113V111T^66669NN   !3!3!5!55==??/3NR$&rB   r   c                     | S rK   rA   r   s    rC   r   zMultipartReader.__aiter__z  r   rB   c                 P   K   |                                   d {V }|t          |S rK   r   r   s     rC   r   zMultipartReader.__anext__}  r   rB   responser;   c                 Z    |                      | | |j        |j                            }|S )zConstructs reader instance from HTTP response.

        :param response: :class:`~aiohttp.client.ClientResponse` instance
        )response_wrapper_clsr   r   )clsrR  objs      rC   from_responsezMultipartReader.from_response  s7     &&cc(*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dS |                                  d{V  | j        r"|                                  d{V  d| _        n|                                  d{V  | j         rdS |                                  d{V }| j        | j        j        dk    rt          |t                    rt          |j                            t                              \  }}|                    d          dk    r|                    d           d{V }t!          |          dk    rt#          d          |                                                                | _        |                                  d{V }|| _        | j        S )	z#Emits the next multipart body part.NFr   r   	_charset_       zInvalid default charset)r   _maybe_release_last_partrO  _read_until_first_boundary_read_boundaryfetch_next_partrN  rJ  r   
isinstancer3   r9   r   r   r   r   r   r  rw   r   r   )r   r   r   r|   r#  s        rC   r   zMultipartReader.next  s     
 < 	4++---------< 	(11333333333 DLL%%'''''''''< 	4))++++++++ O#&+55400 6 2$,2B2BCV2W2WXXIAvzz&!![00 !% 3 3333333w<<"$$&'@AAA(/(>(>(@(@%!1133333333rB   c                    K   | j         sA|                                  d{V }|dS |                                 d{V  | j         ?dS dS )z=Reads all the body parts to the void till the final boundary.N)r   r   r   r   s     rC   r   zMultipartReader.release  sy      , 	!$$$$$$D|,,..       	 , 	! 	! 	! 	! 	!rB   c                 d   K   |                                   d{V }|                     |          S )z"Returns the next body part reader.N)_read_headers_get_part_reader)r   r   s     rC   r`  zMultipartReader.fetch_next_part  s?       **,,,,,,,,$$W---rB   r   c                 R   |                     t          d          }t          |          }|j        dk    r@| j         t          |           || j                  S |                     || j                  S |                     | 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   rH  N)r   r   )r   r    r#   rK  rF  r   part_reader_clsr   rJ  r   r   )r   r   r%  r&  s       rC   re  z MultipartReader._get_part_reader  s     L"--!%((=K''(0!tDzz'4=999,,WdmDDD''. $ 5 (   rB   c                 t    | j         j        d         }t          |          dk    rt          d|z            |S )Nr   F   z&boundary %r is too long (70 chars max))rJ  r$  r   r   )r   r   s     rC   rL  zMultipartReader._get_boundary  s;    >,Z8x==2EPQQQrB   c                    K   | j         r| j                                         S | j                                         d {V S rK   )r   ru   r   r   r   s    rC   	_readlinezMultipartReader._readline  sJ      < 	&<##%%%]++---------rB   c                    K   	 |                                   d {V }|dk    rt          d| j        z            |                                }|| j        k    rd S || j        dz   k    r	d| _        d S p)NTrB   z#Could not find starting boundary %rr   )rk  r   r   r   r   r   r   s     rC   r^  z*MultipartReader._read_until_first_boundary  s      	..********E|| 9T^L   LLNNE&&$.5000#	rB   c                   K   |                                   d {V                                 }|| j        k    rd S || j        dz   k    rd| _        |                                   d {V }|                                   d {V }|d d         dk    r| j                            |           d S | j                            ||g           d S t          d|d| j                  )Nr   TrV   zInvalid boundary z, expected )rk  r   r   r   r   r   r   r   )r   r   epiloguer   s       rC   r_  zMultipartReader._read_boundary  s     ~~''''''''//11DN""Ddnu,,,DL!^^--------H"nn........I !}%%##I.....
 ##Y$9:::::WWWT^WWXXXrB   c                    K   g }	 | j                                          d {V }|                    d          }|                    |           |snMt	                      }|                    |          \  }}|S )NTr   )r   r   r   r   r%   parse_headers)r   linesr   parserr   raw_headerss         rC   rd  zMultipartReader._read_headers  s      	-0022222222ELL))ELL 	 %33E::rB   c                    K   | j         e| j                                         s| j                                          d{V  | j                            | j         j                   d| _         dS dS )z8Ensures that the last read body part is read completely.N)rN  r   r   r   r   r   s    rC   r]  z(MultipartReader._maybe_release_last_part   sx      ?&?))++ 0o--/////////L 7888"DOOO	 '&rB   r   )rG   r   )#r>   r?   r@   r   r   rT  rF  r   r   rB  r3   rg  r   rn   r.   r   r1   r   r   r   classmethodrW  rL   r   r   r   r`  re  rL  r+  rk  r^  r_  rd  r]  rA   rB   rC   r5   r5   ^  s8           4 ?C(4(9#:;BBB$O'S 1 'L 'T ' ' ' '"     	%)>9:	;    " 
"   [    !	%)>9:	;! ! ! !F! ! ! !.	 .0	1. . . .( 
 .0	1   4s    . . . . .
   Y Y Y Y.
 
 
 
# # # # # #rB   r5   c                       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   r   rG   c                 8   ||nt          j                    j        }	 |                    d          | _        n# t
          $ r t          d          d w xY wd| d| j         }t                      	                    d |           g | _
        |dk    | _        d S )Nasciiz(boundary should contain ASCII only charsz
multipart/z; boundary=)content_typer   )uuiduuid4hexrM  r   UnicodeEncodeErrorr   _boundary_valuer3  r   _partsr   )r   r   r   r%  r5  s       rC   r   zMultipartWriter.__init__3  s    '3889I	S%__W55DNN! 	S 	S 	SGHHdR	SGWGG1EGGE222#%$3s	   : Ac                     | S rK   rA   r   s    rC   	__enter__zMultipartWriter.__enter__E  r   rB   exc_typeexc_valexc_tbc                     d S 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    DK   rB   c                 *    t          | j                  S rK   )r   r  r   s    rC   __len__zMultipartWriter.__len__S  s    4;rB   c                     dS )NTrA   r   s    rC   __bool__zMultipartWriter.__bool__V  s    trB   s   \A[!#$%&'*+\-.^_`|~\w]+\Zs   [\x00-\x08\x0A-\x1F\x7F]c                 N   | j         }t          j        | j        |          r|                    d          S t          j        | j        |          rt          d          |                    dd          }|                    dd          }d|                    d          z   dz   S )zzWrap boundary parameter value in quotes, if necessary.

        Reads self.boundary and returns a unicode string.
        ry  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    & 8D+U33 	)<<(((9T4e<< 	KIJJJ  %}}UG<<3;;D&II)00999C??rB   c                 6    | j                             d          S )Nry  )r   r   r   s    rC   r   zMultipartWriter.boundary|  s    ~$$W---rB   rV  r   c                 8   |t                      }t          |t                    r/|j                            |           |                     |          S 	 t          ||          }|                     |          S # t          $ r t          d|z            w xY w)Nr   zCannot create payload from %r)	r   ra  r*   r   updateappend_payloadr,   r(   r9  )r   rV  r   payloads       rC   r   zMultipartWriter.append  s    ?!mmGc7## 		4Kw'''&&s+++4%c7;;; **7333  G G G ?# EFFFGs   A< <Br  c                    d}d}| j         rmt          t          t          h|j                                        z  rJ t          |j        vr.dt          | j                   }|	                    d|           n|j        
                    t          d                                          }|r|dvrt          d|           |dk    rd}|j        
                    t          d                                          }|d	vrt          d
|           |dk    rd}|j        }| |s|st          |          |j        t          <   | 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  r4  r   rv   r  r   rn   r   )r   r  r   te_encodingr   r   s         rC   r  zMultipartWriter.append_payload  s   "&%) 	< &~7PQ/&&(()  
 #'/994#dk"2"244//$/GGG **+;R@@FFHHH LH,KKK"#J#J#JKKK:%% "/--.GLLRRTTK"NNN"#V#V#VWWWh&&" <D[25d))/GX{;<<<rB   c                 j    |t                      }|                     t          ||                    S )zHelper to append JSON part.Nr  )r   r  r'   )r   rV  r   s      rC   append_jsonzMultipartWriter.append_json  s3     ?!mmG"";sG#D#D#DEEErB   c                 6   t          |t          t          f          sJ |t                      }t          |t                    r!t	          |                                          }t          |d          }|                     t          ||d                    S )z&Helper to append form urlencoded part.NT)doseqz!application/x-www-form-urlencoded)r   rz  )	ra  r   r   r   listr   r   r  r+   )r   rV  r   r   s       rC   append_formzMultipartWriter.append_form  s     #'233333?!mmGc7## 	$syy{{##CD)))""g4W  
 
 	
rB   c                 
   d}| j         D ][\  }}}|j        }|s|s| dS |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 ; )*;ttSdn%%&  d*++	,
   EE 	S(((1,,rB   rl   rm   r   r6  c                 X     d                      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   | ]F\  }}}d j         z   dz   |j                                      z   |                                z   V  GdS )rI  z
N)r   r  r   )r   r   _e_ter   r6  r   s       rC   r   z)MultipartWriter.decode.<locals>.<genexpr>  s       
 
 b# m "))(F;;< kkmm	
 
 
 
 
 
rB   )ro   r  r:  s   ```rC   r   zMultipartWriter.decode  sQ     ww 
 
 
 
 
 
 "&
 
 
 
 
 	
rB   c                   K   g }| j         D ]\  }}}|                    d| j        z   dz              |                    |j                   |                    ||           d{V }|                    |           |                    d           |                    d| j        z   dz              d                    |          S )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   r6  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88xxrB   r=  close_boundaryc                   K   | j         D ]'\  }}}| j        r&t          |j        v sJ d|j        t                   v sJ |                    d| j        z   dz              d{V  |                    |j                   d{V  |s|rst          |          }|r|                    |           |r|	                    |           |                    |           d{V  |
                                 d{V  n|                    |           d{V  |                    d           d{V  )|r(|                    d| j        z   dz              d{V  dS dS )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  s      ,0; 	( 	('D(K! D*dl::::$,/B"CCCCC,,ut~5?@@@@@@@@@,,t3444444444 	); 	)*622 3((222 3%%k222jjmm#######kkmm########jj(((((((((,,w'''''''''' 	C,,ut~5	ABBBBBBBBBBB	C 	CrB   c                   K   | j         rdS d| _         t          | j                  D ]b\  }\  }}}|j        sR|j        sK	 |                                 d{V  3# t          $ r"}t          j        d||d           Y d}~Zd}~ww xY wcdS )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       > 	F "+4;!7!7 	 	C$1> $- **,,&&&&&&&&    #)?cTX        		 	s   A
B$BB)r   N)rG   r6   rK   rA  )Tr   ).r>   r?   r@   r   rB  
_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  rC  rD  s   @rC   r6   r6   ,  s          LLLJ4 4 4# 4RV 4 4 4 4 4 4$   4./ -( '	
 
   !(5/ ! ! ! !         $     $$ABB!+,H!I!I@ @ @ @ X@> .# . . . X.4 4# 4c1B(C 4w 4 4 4 4$g $' $ $ $ $N @DF FF!)'#s(*;!<F	F F F F 04
 
8E#s(O,gc3h.??@
 '#s(+,
 
	
 
 
 
* hsm    X(
 
s 
c 
 
 
 
 
 s c QV    8 DHC C*C<@C	C C C C:       rB   r6   c                   j    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dS )r  r=  rG   Nc                 >    || _         d | _        d | _        d | _        d S rK   )_writer	_encoding	_compress_encoding_buffer)r   r=  s     rC   r   zMultipartPayloadWriter.__init__J  s%    (,3759rB   r   c                 h    |dk    r|| _         t                      | _        d S |dk    r	d| _         d S d S )Nr   r  )r  r   r  )r   r   s     rC   r  z&MultipartPayloadWriter.enable_encodingP  sE    x%DN$-KKD!!!+++/DNNN ,+rB   r  strategyc                 4    t          |d|          | _        d S )NT)r   r  r  )r   r  )r   r   r  s      rC   r  z)MultipartPayloadWriter.enable_compressionW  s'     ($(
 
 
rB   c                 ,  K   | j         =| j                                         }|r"d | _         |                     |           d {V  | j        dk    r@| j        r;| j                            t          j        | j                             d {V  d S d S d S )Nr   )r  flushr?  r  r  r  r   	b64encoderm  s     rC   r  z MultipartPayloadWriter.write_eof`  s      >%N((**E (!%jj'''''''''>X%%$ Rl(()9$:O)P)PQQQQQQQQQQQ &%R RrB   r   c                 t  K   | j         &|r$| j                             |           d {V }|sd S | j        dk    r| j        }|J |                    |           |rxt          t          |          d          \  }}|d |dz           ||dz  d          c}| _        |r8t          j        |          }| j	        
                    |           d {V  d S d S d S | j        dk    r4| j	        
                    t          j        |                     d {V  d S | j	        
                    |           d {V  d S )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     >% "n55e<<<<<<<< F>X%%'C???JJu 7!#c((A..S47	#'	NCa		N0	40 7%/	::H,,,X666666666667 77 7 ^111,$$X_U%;%;<<<<<<<<<<<,$$U+++++++++++rB   )r  Nr   )r>   r?   r@   r   r   rn   r  r   r   r  r  r+  r?  rA   rB   rC   r  r  I  s        :3 : : : : :0 0 0 0 0 0 DH
 

3;C=
	
 
 
 
	R 	R 	R 	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 6 6 6 6 6 6 6 6 3 3 3 3 3 3 3 3 % % % % % %         
              8 7 7 7 7 7 7 7 7 7 7 7                                    " ! ! ! ! !w76!1222D  .------	 	 	 	 	. 	 	 		 	 	 	 	 	 	 	d$SMd$
8C=$sCx.()d$ d$ d$ d$P ,6 CH%(c]   B," ," ," ," ," ," ," ,"^i@ i@ i@ i@ i@ i@ i@ i@X	 nEO444"7 "7 "7 "7 "7G "7 "7 54"7JH# H# H# H# H# H# H# H#V 	gsC Z Z Z Z Zg Z Z Zz7, 7, 7, 7, 7, 7, 7, 7, 7, 7,rB   