���Z� � @ s{ d d l 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 Z Gd d � d � Z
d S)
� N)� urlencode)� MultiDict�MultiDictProxy� )�hdrs� multipart�payload)�guess_filename�FormDatac @ s� e Z d Z d Z f d d d d � Z e d d � � Z d d d d d
d d d � Z d
d � Z d d � Z d d � Z
d d � Z d S)r
z_Helper class for multipart/form-data and
application/x-www-form-urlencoded body generation.TNc C s� t j d � | _ g | _ d | _ | | _ | | _ t | t � rZ t | j
� � } n t | t t f � sx | f } | j | � d S)Nz form-dataF)
r ZMultipartWriter�_writer�_fields�
_is_multipart�
_quote_fields�_charset�
isinstance�dict�list�items�tuple�
add_fields)�self�fields� quote_fields�charset� r � /formdata.py�__init__ s zFormData.__init__c C s | j S)N)r
)r r r r � is_multipart s zFormData.is_multipart� content_type�filename�content_transfer_encodingc C s� t | t j � r d | _ n6 t | t t t f � rT | d k rT | d k rT | } t d | i � } | d k r� t | t � r� t d | � � | d k r� t | t j � r� t
| | � } | d k r� | | d <d | _ i } | d k r%t | t � st d | � � | | t j <d | _ | d k rft | t � sPt d | � � | | t j
<d | _ | j j | | | f � d S)NT�namez,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! �valuer r r � type_options�headersr r r � add_field" s8
zFormData.add_fieldc G s� t | � } x� | r� | j d � } t | t j � rX t | d � } | j | | � q t | t t f � r� | j | j
� � q t | t t f � r� t | � d k r� | \ } } | j | | � q t
d j | � � � q Wd 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 r. r r �extendr r �lenr( �format)r r Zto_addZrec�k�fpr r r r E s ' zFormData.add_fieldsc C s� g } x1 | j D]& \ } } } | j | d | f � q W| j d k rR | j n d } | d k rm d } n
d | } t j t | d d d | �j � d | �S) Nr! zutf-8z!application/x-www-form-urlencodedz-application/x-www-form-urlencoded; charset=%sZdoseqT�encodingr )r r* r r Z BytesPayloadr �encode)r �datar, �_r+ r r r r r �_gen_form_urlencoded\ s z FormData._gen_form_urlencodedc C s x| j D]� \ } } } y_ t j | k rY t j | d | t j d | d | j �} n t j | d | d | j �} WnG t k
r� } z' t d t | � | | f � | � WYd d } ~ Xn X| r� | j d d | j
| �| j j t j
d � | j j | � q
W| j S)zAEncode a list of fields using the multipart/form-data MIME formatr r- r7 z8Can not serialize value type: %r
headers: %r
value: %rNz form-datar )r r r) r Z get_payloadr � Exceptionr( �typeZset_content_dispositionr r- ZpopallZCONTENT_LENGTHr Zappend_payload)r Z
dispparamsr- r+ �part�excr r r �_gen_form_datan s$ /zFormData._gen_form_datac C s! | j r | j � S| j � Sd S)N)r
r@ r; )r r r r �__call__� s
zFormData.__call__) �__name__�
__module__� __qualname__�__doc__r �propertyr r. r r; r@ rA r r r r r
s
" )r
)r" Z urllib.parser Z multidictr r � r r r Zhelpersr �__all__r
r r r r �<module> s