U
    
"gZ!                     @   s   d dl mZ d dlZd dlZd dlZd dlZdddZdd Zdd	d	d
Z	e	
dd ed dD  dd Zdd ZeZG dd deZdS )    )absolute_importNapplication/octet-streamc                 C   s   | rt | d p|S |S )z
    Guess the "Content-Type" of a file.

    :param filename:
        The filename to guess the "Content-Type" of using :mod:`mimetypes`.
    :param default:
        If no "Content-Type" can be guessed, default to `default`.
    r   )	mimetypesZ
guess_type)filenamedefault r   0/usr/lib/python3/dist-packages/urllib3/fields.pyguess_content_type	   s    	r	   c              	      s   t  tjr d t fdddD sdd|  f }z|d W n ttfk
r^   Y nX |S tjrt d t	j
 d d|  f  tjr d  S )a  
    Helper function to format and quote a single header parameter using the
    strategy defined in RFC 2231.

    Particularly useful for header parameters which might contain
    non-ASCII values, like file names. This follows RFC 2388 Section 4.4.

    :param name:
        The name of the parameter, a string expected to be ASCII only.
    :param value:
        The value of the parameter, provided as ``bytes`` or `str``.
    :ret:
        An RFC-2231-formatted unicode string.
    utf-8c                 3   s   | ]}| kV  qd S Nr   ).0Zchvaluer   r   	<genexpr>)   s     z.format_header_param_rfc2231.<locals>.<genexpr>z"\
%s="%s"asciiz%s*=%s)
isinstancesixbinary_typedecodeanyencodeUnicodeEncodeErrorUnicodeDecodeErrorZPY2emailZutilsZencode_rfc2231)namer   resultr   r   r   format_header_param_rfc2231   s     


r   z%22z\\)"\r   c                 C   s&   i | ]}|d krt |d|qS ))   z%{:02X})r   Zunichrformat)r   Zccr   r   r   
<dictcomp>I   s    r"       c                    s:    fdd}t ddd   D }||| }|S )Nc                    s    |  d S )Nr   )group)matchneedles_and_replacementsr   r   replacerR   s    z#_replace_multiple.<locals>.replacer|c                 S   s   g | ]}t |qS r   )reescape)r   Zneedler   r   r   
<listcomp>V   s     z%_replace_multiple.<locals>.<listcomp>)r*   compilejoinkeyssub)r   r'   r(   patternr   r   r&   r   _replace_multipleQ   s    r2   c                 C   s,   t |tjr|d}t|t}d| |f S )a  
    Helper function to format and quote a single header parameter using the
    HTML5 strategy.

    Particularly useful for header parameters which might contain
    non-ASCII values, like file names. This follows the `HTML5 Working Draft
    Section 4.10.22.7`_ and matches the behavior of curl and modern browsers.

    .. _HTML5 Working Draft Section 4.10.22.7:
        https://w3c.github.io/html/sec-forms.html#multipart-form-data

    :param name:
        The name of the parameter, a string expected to be ASCII only.
    :param value:
        The value of the parameter, provided as ``bytes`` or `str``.
    :ret:
        A unicode string, stripped of troublesome characters.
    r
   r   )r   r   r   r   r2   _HTML5_REPLACEMENTS)r   r   r   r   r   format_header_param_html5^   s    

r4   c                   @   sR   e Zd ZdZddefddZeefddZdd Zd	d
 Z	dd Z
dddZdS )RequestFielda  
    A data container for request body parameters.

    :param name:
        The name of this request field. Must be unicode.
    :param data:
        The data/value body.
    :param filename:
        An optional filename of the request field. Must be unicode.
    :param headers:
        An optional dict-like object of headers to initially use for the field.
    :param header_formatter:
        An optional callable that is used to encode and format the headers. By
        default, this is :func:`format_header_param_html5`.
    Nc                 C   s0   || _ || _|| _i | _|r&t|| _|| _d S r   )_name	_filenamedataheadersdictheader_formatter)selfr   r8   r   r9   r;   r   r   r   __init__   s    
zRequestField.__init__c                 C   s`   t |tr4t|dkr"|\}}}q@|\}}t|}nd}d}|}| ||||d}|j|d |S )a  
        A :class:`~urllib3.fields.RequestField` factory from old-style tuple parameters.

        Supports constructing :class:`~urllib3.fields.RequestField` from
        parameter of key/value strings AND key/filetuple. A filetuple is a
        (filename, data, MIME type) tuple where the MIME type is optional.
        For example::

            'foo': 'bar',
            'fakefile': ('foofile.txt', 'contents of foofile'),
            'realfile': ('barfile.txt', open('realfile').read()),
            'typedfile': ('bazfile.bin', open('bazfile').read(), 'image/jpeg'),
            'nonamefile': 'contents of nonamefile field',

        Field names and filenames must be unicode.
           N)r   r;   )content_type)r   tuplelenr	   make_multipart)clsZ	fieldnamer   r;   r   r8   r?   Zrequest_paramr   r   r   from_tuples   s     

   zRequestField.from_tuplesc                 C   s   |  ||S )aI  
        Overridable helper function to format a single header parameter. By
        default, this calls ``self.header_formatter``.

        :param name:
            The name of the parameter, a string expected to be ASCII only.
        :param value:
            The value of the parameter, provided as a unicode string.
        )r;   )r<   r   r   r   r   r   _render_part   s    zRequestField._render_partc                 C   sL   g }|}t |tr| }|D ]"\}}|dk	r|| || qd|S )aO  
        Helper function to format and quote a single header.

        Useful for single headers that are composed of multiple items. E.g.,
        'Content-Disposition' fields.

        :param header_parts:
            A sequence of (k, v) tuples or a :class:`dict` of (k, v) to format
            as `k1="v1"; k2="v2"; ...`.
        N; )r   r:   itemsappendrE   r.   )r<   Zheader_partspartsiterabler   r   r   r   r   _render_parts   s    
zRequestField._render_partsc                 C   s   g }dddg}|D ]*}| j |dr|d|| j | f  q| j  D ]&\}}||krH|rH|d||f  qH|d d|S )z=
        Renders the headers for this request field.
        Content-DispositionContent-TypeContent-LocationFz%s: %sz
)r9   getrH   rG   r.   )r<   linesZ	sort_keysZsort_keyZheader_nameZheader_valuer   r   r   render_headers   s    

zRequestField.render_headersc                 C   sX   |pd| j d< | j d  dd| d| jfd| jffg7  < || j d< || j d< d	S )
a|  
        Makes this request field into a multipart request field.

        This method overrides "Content-Disposition", "Content-Type" and
        "Content-Location" headers to the request parameter.

        :param content_type:
            The 'Content-Type' of the request body.
        :param content_location:
            The 'Content-Location' of the request body.

        z	form-datarL   rF    r   r   rM   rN   N)r9   r.   rK   r6   r7   )r<   Zcontent_dispositionr?   Zcontent_locationr   r   r   rB      s    
zRequestField.make_multipart)NNN)__name__
__module____qualname____doc__r4   r=   classmethodrD   rE   rK   rQ   rB   r   r   r   r   r5   }   s   
#     r5   )r   )Z
__future__r   Zemail.utilsr   r   r*   r   r	   r   r3   updateranger2   r4   Zformat_header_paramobjectr5   r   r   r   r   <module>   s&   
*	