U
'�7`� � @ s~ d dl Z d dlmZmZmZmZ d dlmZ d dlm Z m
Z
ddl m Z m
Z
mZ ddlmZ ddlmZ d ZG d
d � d �ZdS ) � N)�Any�Iterable�List�Optional)� urlencode)� MultiDict�MultiDictProxy� )�hdrs� multipart�payload)�guess_filename)�Payload)�FormDatac @ s� e Zd ZdZdee eee dd�dd�Z e
ed�d d
��Z dddd �eeee ee ee dd �d
d�Z edd�dd�Z
ejd�dd�Zejd�dd�Zed�dd�ZdS )r z_Helper class for multipart/form-data and
application/x-www-form-urlencoded body generation.� TN)�fields� quote_fields�charset�returnc C sd t �d�| _g | _d| _d| _|| _|| _t|t �rBt
|� � �}nt|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/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/formdata.py�__init__ s
zFormData.__init__)r c C s | j S �N)r �r# r r r$ � is_multipart% s zFormData.is_multipart)� content_type�filename�content_transfer_encoding)�name�valuer) r* r+ r c C s t |tj�rd| _n$t |tttf�r8|d kr8|d kr8|}td|i�}|d k rbt |t�sbt d| ��|d kr�t |tj�r�t
||�}|d k r�||d< d| _i }|d k r�t |t�s�t d| ��||t j < d| _|d k r�t |t�s�t 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_TYPEZCONTENT_TRANSFER_ENCODINGr �append)r# r, r- r) r* r+ � type_options�headersr r r$ � add_field) sB
�
�
��
zFormData.add_field)r r c G s� t |�}|r�|�d�}t|tj�r:t|d�}| �||� qt|ttf�rX|� |�
� � qt|t t f�r�t |�dkr�|\}}| �||� qt
d�|���qd 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 Zto_addZrec�k�fpr r r$ r" W s
��zFormData.add_fieldsc C sn g }| j D ] \}}}|�|d |f� q
| jd k r8| jnd}|dkrJd}nd| }tjt|d|d��� |d�S )Nr, zutf-8z!application/x-www-form-urlencodedz-application/x-www-form-urlencoded; charset=%sT)Zdoseq�encoding)r) )r r6 r r � BytesPayloadr �encode)r# �datar7 �_r- r r) r r r$ �_gen_form_urlencodedo s �z FormData._gen_form_urlencodedc C s� | j rtd��| jD ]�\}}}z<tj|krFtj||tj || jd�}ntj||| jd�}W n: tk
r� } z t dt
|�||f �|�W 5 d}~X Y nX |r�|j d d| j i|�� |j
dk s�t�|j
�tjd� | j�|� qd| _ | jS )
zAEncode a list of fields using the multipart/form-data MIME formatz$Form data has been processed already)r) r8 rB )r8 rB z8Can not serialize value type: %r
headers: %r
value: %rNr r T)r )r � RuntimeErrorr r
r5 r Z get_payloadr � Exceptionr4 �typeZset_content_dispositionr r8 �AssertionErrorZpopallZCONTENT_LENGTHr Zappend_payload)r# Z
dispparamsr8 r- �part�excr r r$ �_gen_form_data� sJ
� �
��� ���zFormData._gen_form_datac C s | j r| �� S | �� S d S r& )r rN rG r' r r r$ �__call__� s zFormData.__call__)r TN)�__name__�
__module__� __qualname__�__doc__r r �boolr r3 r% �propertyr( r9 r" r rC rG r r rN r rO r r r r$ r s6 �� �� .%r )r. �typingr r r r Z urllib.parser Z multidictr r � r
r r Zhelpersr
r �__all__r r r r r$ �<module> s