o
    n~b                     @   s~   d dl Z d dlmZmZmZmZ d dlmZ d dlm	Z	m
Z
 ddlmZmZmZ ddlmZ ddlmZ d	ZG d
d dZdS )    N)AnyIterableListOptional)	urlencode)	MultiDictMultiDictProxy   )hdrs	multipartpayload)guess_filename)Payload)FormDatac                   @   s   e Zd ZdZ			ddee dedee ddfd	d
Z	e
defddZdddddededee dee dee ddfddZdeddfddZdejfddZdejfddZdefddZdS )r   ztHelper class for form body generation.

    Supports multipart/form-data and application/x-www-form-urlencoded.
     TNfieldsquote_fieldscharsetreturnc                 C   sd   t d| _g | _d| _d| _|| _|| _t|t	r!t
| }n
t|t
tfs+|f}| j|  d S )N	form-dataF)r   MultipartWriter_writer_fields_is_multipart_is_processed_quote_fields_charset
isinstancedictlistitemstuple
add_fields)selfr   r   r   r   r   ;/usr/local/lib/python3.10/dist-packages/aiohttp/formdata.py__init__   s   
zFormData.__init__c                 C   s   | j S N)r   r#   r   r   r$   is_multipart'   s   zFormData.is_multipart)content_typefilenamecontent_transfer_encodingnamevaluer)   r*   r+   c                C   s  t |tjr
d| _nt |tttfr|d u r|d u r|}td|i}|d ur1t |ts1t	d| |d u r@t |tjr@t
||}|d urK||d< d| _i }|d urdt |ts\t	d| ||tj< d| _|d ur{t |tsst	d| ||tj< d| _| j|||f d S )NTr,   z,filename must be an instance of str. Got: %sr*   z0content_type must be an instance of str. Got: %sz=content_transfer_encoding must be an instance of str. Got: %s)r   ioIOBaser   bytes	bytearray
memoryviewr   str	TypeErrorr   r
   CONTENT_TYPEZCONTENT_TRANSFER_ENCODINGr   append)r#   r,   r-   r)   r*   r+   type_optionsheadersr   r   r$   	add_field+   sB   





zFormData.add_fieldc                 G   s   t |}|rO|d}t|tjrt|d}| || n.t|ttfr,|	|
  nt|t tfrDt|dkrD|\}}| || ntd||sd S d S )Nr   unknown   zxOnly io.IOBase, multidict and (name, file) pairs allowed, use .add_field() for passing more complex parameters, got {!r})r   popr   r.   r/   r   r9   r   r   extendr    r!   lenr4   format)r#   r   Zto_addZreckfpr   r   r$   r"   Y   s    

zFormData.add_fieldsc                 C   sn   g }| j D ]\}}}||d |f q| jd ur| jnd}|dkr%d}nd| }tjt|d|d |dS )Nr,   zutf-8z!application/x-www-form-urlencodedz-application/x-www-form-urlencoded; charset=%sT)doseqencoding)r)   )r   r6   r   r   BytesPayloadr   encode)r#   datar7   _r-   r   r)   r   r   r$   _gen_form_urlencodedq   s   zFormData._gen_form_urlencodedc                 C   s   | j rtd| jD ]_\}}}ztj|v r#tj||tj || jd}n	tj||| jd}W n tyE } zt	dt
|||f |d}~ww |rc|j	d	d| ji| |jdus[J |jtjd | j| q
d| _ | jS )
zAEncode a list of fields using the multipart/form-data MIME formatz$Form data has been processed already)r)   r8   rC   )r8   rC   z8Can not serialize value type: %r
 headers: %r
 value: %rNr   r   T)r   )r   RuntimeErrorr   r
   r5   r   get_payloadr   	Exceptionr4   typeZset_content_dispositionr   r8   ZpopallCONTENT_LENGTHr   Zappend_payload)r#   Z
dispparamsr8   r-   partexcr   r   r$   _gen_form_data   sL   
zFormData._gen_form_datac                 C   s   | j r|  S |  S r&   )r   rP   rH   r'   r   r   r$   __call__   s   zFormData.__call__)r   TN)__name__
__module____qualname____doc__r   r   boolr   r3   r%   propertyr(   r9   r"   r   rD   rH   r   r   rP   r   rQ   r   r   r   r$   r      sH    

.%r   )r.   typingr   r   r   r   urllib.parser   Z	multidictr   r    r
   r   r   helpersr   r   __all__r   r   r   r   r$   <module>   s    