a
    }3e                     @   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ee eee ddddZ	e
edd	d
Zddddeeee ee ee ddddZeddddZejdddZejdddZed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	rBt
| }nt|t
tfsV|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   C/home/pi/bot/my_env/lib/python3.9/site-packages/aiohttp/formdata.py__init__   s    
zFormData.__init__)r   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+   r   c                C   s  t |tjrd| _n$t |tttfr8|d u r8|d u r8|}td|i}|d urbt |tsbt	d| |d u rt |tjrt
||}|d ur||d< d| _i }|d urt |tst	d| ||tj< d| _|d urt |tst	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_field)r   r   c                 G   s   t |}|r|d}t|tjr:t|d}| || qt|ttfrX|	|
  qt|t tfrt|dkr|\}}| || qtd|q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r8| jnd}|dkrJd}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 rFtj||tj || jd}ntj||| jd}W n< ty } z$t	dt
|||f |W Y d}~n
d}~0 0 |r|jd	d| ji| |jdusJ |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   sF    

 zFormData._gen_form_datac                 C   s   | j r|  S |  S d 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      s6      .%r   )r.   typingr   r   r   r   urllib.parser   Z	multidictr   r    r
   r   r   Zhelpersr   r   __all__r   r   r   r   r$   <module>   s   