3
[l                 @   s  d Z ddlZddlZddlZddlmZ ddlmZ ddlm	Z	m
Z
mZmZmZmZ ddlmZmZmZmZ ddlmZmZmZ dd	l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#m$Z$ ddl%m&Z& ddl'm(Z( ddlm)Z)m*Z*m+Z+m,Z,m-Z-m.Z. ddl/m0Z0 ddlm1Z1 ej2dkrJy
ej3Z4W n e5k
rF   ej6Z4Y nX nejZ4efddZ7efddZ8G dd de9Z:G dd de:Z;dd Z<dS )z
requests.session
~~~~~~~~~~~~~~~~

This module provides a Session object to manage and persist settings across
requests (cookies, auth, proxies).
    N)	timedelta   )_basic_auth_str)	cookielibis_py3OrderedDicturljoinurlparseMapping)cookiejar_from_dictextract_cookies_to_jarRequestsCookieJarmerge_cookies)RequestPreparedRequestDEFAULT_REDIRECT_LIMIT)default_hooksdispatch_hook)to_native_string)to_key_val_listdefault_headers)TooManyRedirectsInvalidSchemaChunkedEncodingErrorContentDecodingError)CaseInsensitiveDict)HTTPAdapter)requote_uriget_environ_proxiesget_netrc_authshould_bypass_proxiesget_auth_from_urlrewind_body)codes)REDIRECT_STATIwin32c             C   st   |dkr| S | dkr|S t |to*t | ts0| S |t|}|jt|  dd |j D }x|D ]
}||= qbW |S )zDetermines appropriate setting for a given request, taking into account
    the explicit setting on that request, and the setting in the session. If a
    setting is a dictionary, they will be merged together using `dict_class`
    Nc             S   s   g | ]\}}|d kr|qS )N ).0kvr&   r&   >/tmp/pip-install-65viz3iy/pip/pip/_vendor/requests/sessions.py
<listcomp>I   s    z!merge_setting.<locals>.<listcomp>)
isinstancer
   r   updateitems)request_settingsession_setting
dict_classmerged_setting	none_keyskeyr&   r&   r*   merge_setting1   s    



r5   c             C   s@   |dks|j dg kr| S | dks0| j dg kr4|S t| ||S )zProperly merges both requests and session hooks.

    This is necessary because when request_hooks == {'response': []}, the
    merge breaks Session hooks entirely.
    Nresponse)getr5   )request_hookssession_hooksr1   r&   r&   r*   merge_hooksP   s
    r:   c               @   s6   e Zd Zdd ZdddZdd	 Zd
d Zdd ZdS )SessionRedirectMixinc             C   s,   |j r(|jd }tr|jd}t|dS dS )z7Receives a Response. Returns a redirect URI or ``None``locationlatin1utf8N)is_redirectheadersr   encoder   )selfrespr<   r&   r&   r*   get_redirect_targeta   s    


z(SessionRedirectMixin.get_redirect_targetFNTc	             k   sd  g }
| j |}t|jj}xB|r^|j }|
j| |
dd |_y
|j W n( tt	t
fk
rv   |jjdd Y nX t|j| jkrtd| j |d|j  |jdrt|j}dt|j|f }t|}|jd	kr|r|j|d
}n|jr|j}|j }|jst|jt|}nt|}t||_| j|| |jtjtjfkrtd}x|D ]}|jj|d qVW d|_ |j}y
|d= W n t!k
r   Y nX t"|j#||j t$|j#| j% |j&|j# | j'||}| j(|| |j)dk	od|kpd|k}|r
t*| |}|r|V  q| j+|f|||||dd|	}t"| j%||j | j |}|V  qW dS )zBReceives a Response. Returns a generator of Responses or Requests.r   NF)decode_contentzExceeded %s redirects.)r6   z//z%s:%s )fragmentContent-LengthContent-TypeTransfer-EncodingCookie)streamtimeoutverifycertproxiesallow_redirects)rH   rI   rJ   ),rD   r	   urlrG   copyappendhistorycontentr   r   RuntimeErrorrawreadlenmax_redirectsr   close
startswithr   scheme_replacegeturlnetlocr   r   rebuild_methodstatus_coder#   temporary_redirectpermanent_redirectr@   popbodyKeyErrorr   _cookiesr   cookiesprepare_cookiesrebuild_proxiesrebuild_auth_body_positionr"   send)rB   rC   reqrL   rM   rN   rO   rP   yield_requestsadapter_kwargshistrR   Zprevious_fragmentprepared_requestparsed_rurlparsedpurged_headersheaderr@   
rewindabler&   r&   r*   resolve_redirectsv   s|    









z&SessionRedirectMixin.resolve_redirectsc             C   sb   |j }|j}d|kr:t|jj}t|}|j|jkr:|d= | jrHt|nd}|dk	r^|j| dS )zWhen being redirected we may want to strip authentication from the
        request to avoid leaking credentials. This method intelligently removes
        and reapplies authentication where possible to avoid credential loss.
        AuthorizationN)r@   rR   r	   requesthostname	trust_envr   prepare_auth)rB   rt   r6   r@   rR   original_parsedredirect_parsednew_authr&   r&   r*   rm      s    
z!SessionRedirectMixin.rebuild_authc             C   s   |dk	r|ni }|j }|j}t|j}|j }|jd}t||d}| jr~| r~t||d}	|	j||	jd}
|
r~|j	||
 d|kr|d= yt
|| \}}W n tk
r   d\}}Y nX |r|rt|||d< |S )a  This method re-evaluates the proxy configuration by considering the
        environment variables. If we are redirected to a URL covered by
        NO_PROXY, we strip the proxy configuration. Otherwise, we set missing
        proxy keys for this URL (in case they were stripped by a previous
        redirect).

        This method also replaces the Proxy-Authorization header where
        necessary.

        :rtype: dict
        Nno_proxy)r   allzProxy-Authorization)NN)r@   rR   r	   r^   rS   r7   r    r~   r   
setdefaultr!   rh   r   )rB   rt   rP   r@   rR   r^   new_proxiesr   bypass_proxyenviron_proxiesproxyusernamepasswordr&   r&   r*   rl      s*    

z$SessionRedirectMixin.rebuild_proxiesc             C   sX   |j }|jtjkr|dkrd}|jtjkr6|dkr6d}|jtjkrN|dkrNd}||_ dS )zWhen being redirected we may want to change the method of the request
        based on certain specs or browser behavior.
        HEADGETPOSTN)methodrc   r#   	see_otherfoundmoved)rB   rt   r6   r   r&   r&   r*   rb   (  s    z#SessionRedirectMixin.rebuild_method)FNTNNF)__name__
__module____qualname__rD   rz   rm   rl   rb   r&   r&   r&   r*   r;   _   s    
p)r;   c               @   s   e Zd ZdZdddddddd	d
ddddgZdd Zdd Zdd Zdd Zd7ddZ	dd Z
dd Zdd  Zd8d!d"Zd9d#d$Zd:d%d&Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 ZdS );Sessiona~  A Requests session.

    Provides cookie persistence, connection-pooling, and configuration.

    Basic Usage::

      >>> import requests
      >>> s = requests.Session()
      >>> s.get('http://httpbin.org/get')
      <Response [200]>

    Or as a context manager::

      >>> with requests.Session() as s:
      >>>     s.get('http://httpbin.org/get')
      <Response [200]>
    r@   rj   authrP   hooksparamsrN   rO   prefetchadaptersrL   r~   r[   c             C   sr   t  | _d | _i | _t | _i | _d| _d| _d | _	t
| _d| _ti | _t | _| jdt  | jdt  d S )NFTzhttps://zhttp://)r   r@   r   rP   r   r   r   rL   rN   rO   r   r[   r~   r   rj   r   r   mountr   )rB   r&   r&   r*   __init__X  s    
zSession.__init__c             C   s   | S )Nr&   )rB   r&   r&   r*   	__enter__  s    zSession.__enter__c             G   s   | j   d S )N)r\   )rB   argsr&   r&   r*   __exit__  s    zSession.__exit__c             C   s   |j pi }t|tjst|}ttt | j |}|j}| jrV| rV| j rVt	|j
}t }|j|jj |j
|j|j|jt|j| jtdt|j| jt|| j|t|j| jd
 |S )a  Constructs a :class:`PreparedRequest <PreparedRequest>` for
        transmission and returns it. The :class:`PreparedRequest` has settings
        merged from the :class:`Request <Request>` instance and those of the
        :class:`Session`.

        :param request: :class:`Request` instance to prepare with this
            session's settings.
        :rtype: requests.PreparedRequest
        )r1   )
r   rR   filesdatajsonr@   r   r   rj   r   )rj   r,   r   	CookieJarr   r   r   r   r~   r   rR   r   preparer   upperr   r   r   r5   r@   r   r   r:   r   )rB   r|   rj   merged_cookiesr   pr&   r&   r*   prepare_request  s*    



zSession.prepare_requestNTc             C   st   t |j ||||pi ||pi |||d
}| j|}|p8i }| j|j||||}|	|
d}|j| | j|f|}|S )a  Constructs a :class:`Request <Request>`, prepares it and sends it.
        Returns :class:`Response <Response>` object.

        :param method: method for the new :class:`Request` object.
        :param url: URL for the new :class:`Request` object.
        :param params: (optional) Dictionary or bytes to be sent in the query
            string for the :class:`Request`.
        :param data: (optional) Dictionary, bytes, or file-like object to send
            in the body of the :class:`Request`.
        :param json: (optional) json to send in the body of the
            :class:`Request`.
        :param headers: (optional) Dictionary of HTTP Headers to send with the
            :class:`Request`.
        :param cookies: (optional) Dict or CookieJar object to send with the
            :class:`Request`.
        :param files: (optional) Dictionary of ``'filename': file-like-objects``
            for multipart encoding upload.
        :param auth: (optional) Auth tuple or callable to enable
            Basic/Digest/Custom HTTP Auth.
        :param timeout: (optional) How long to wait for the server to send
            data before giving up, as a float, or a :ref:`(connect timeout,
            read timeout) <timeouts>` tuple.
        :type timeout: float or tuple
        :param allow_redirects: (optional) Set to True by default.
        :type allow_redirects: bool
        :param proxies: (optional) Dictionary mapping protocol or protocol and
            hostname to the URL of the proxy.
        :param stream: (optional) whether to immediately download the response
            content. Defaults to ``False``.
        :param verify: (optional) Either a boolean, in which case it controls whether we verify
            the server's TLS certificate, or a string, in which case it must be a path
            to a CA bundle to use. Defaults to ``True``.
        :param cert: (optional) if String, path to ssl client cert file (.pem).
            If Tuple, ('cert', 'key') pair.
        :rtype: requests.Response
        )
r   rR   r@   r   r   r   r   r   rj   r   )rM   rQ   )r   r   r   merge_environment_settingsrR   r-   ro   )rB   r   rR   r   r   r@   rj   r   r   rM   rQ   rP   r   rL   rN   rO   r   rp   prepsettingssend_kwargsrC   r&   r&   r*   r|     s(    )

zSession.requestc             K   s   |j dd | jd|f|S )zSends a GET request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        rQ   Tr   )r   r|   )rB   rR   kwargsr&   r&   r*   r7     s    zSession.getc             K   s   |j dd | jd|f|S )zSends a OPTIONS request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        rQ   TOPTIONS)r   r|   )rB   rR   r   r&   r&   r*   options  s    zSession.optionsc             K   s   |j dd | jd|f|S )zSends a HEAD request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        rQ   Fr   )r   r|   )rB   rR   r   r&   r&   r*   head  s    zSession.headc             K   s   | j d|f||d|S )a  Sends a POST request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param data: (optional) Dictionary, bytes, or file-like object to send in the body of the :class:`Request`.
        :param json: (optional) json to send in the body of the :class:`Request`.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        r   )r   r   )r|   )rB   rR   r   r   r   r&   r&   r*   post%  s    
zSession.postc             K   s   | j d|fd|i|S )aY  Sends a PUT request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param data: (optional) Dictionary, bytes, or file-like object to send in the body of the :class:`Request`.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        PUTr   )r|   )rB   rR   r   r   r&   r&   r*   put1  s    	zSession.putc             K   s   | j d|fd|i|S )a[  Sends a PATCH request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param data: (optional) Dictionary, bytes, or file-like object to send in the body of the :class:`Request`.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        PATCHr   )r|   )rB   rR   r   r   r&   r&   r*   patch<  s    	zSession.patchc             K   s   | j d|f|S )zSends a DELETE request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param \*\*kwargs: Optional arguments that ``request`` takes.
        :rtype: requests.Response
        DELETE)r|   )rB   rR   r   r&   r&   r*   deleteG  s    zSession.deletec             K   s~  |j d| j |j d| j |j d| j |j d| j t|trJtd|jdd}|j	d}|j
}| j|jd}t }|j|f|}t | }	t|	d	|_td
||f|}|jrx |jD ]}
t| j|
j|
j qW t| j||j | j||f|}|r
dd |D ng }|r.|jd| |j }||_|sny"t| j||fddi||_W n tk
rl   Y nX |sz|j |S )zISend a given PreparedRequest.

        :rtype: requests.Response
        rL   rN   rO   rP   z#You can only send PreparedRequests.rQ   T)rR   )secondsr6   c             S   s   g | ]}|qS r&   r&   )r'   rC   r&   r&   r*   r+     s    z Session.send.<locals>.<listcomp>r   rq   )r   rL   rN   rO   rP   r,   r   
ValueErrorrf   r7   r   get_adapterrR   preferred_clockro   r   elapsedr   rU   r   rj   r|   rX   rz   insertnext_nextStopIterationrV   )rB   r|   r   rQ   rL   r   adapterstartrr   rC   genrU   r&   r&   r*   ro   Q  sB    


"zSession.sendc       
      C   s   | j rr|dk	r|jdnd}t||d}x |j D ]\}}	|j||	 q2W |dksZ|dkrrtjjdpptjjd}t|| j}t|| j	}t|| j
}t|| j}||||dS )z^
        Check the environment and merge it with some settings.

        :rtype: dict
        Nr   )r   TREQUESTS_CA_BUNDLECURL_CA_BUNDLE)rN   rP   rL   rO   )r~   r7   r   r.   r   osenvironr5   rP   rL   rN   rO   )
rB   rR   rP   rL   rN   rO   r   env_proxiesr(   r)   r&   r&   r*   r     s    z"Session.merge_environment_settingsc             C   s>   x,| j j D ]\}}|j j|j r|S qW td| dS )z~
        Returns the appropriate connection adapter for the given URL.

        :rtype: requests.adapters.BaseAdapter
        z*No connection adapters were found for '%s'N)r   r.   lowerr]   r   )rB   rR   prefixr   r&   r&   r*   r     s    zSession.get_adapterc             C   s    x| j j D ]}|j  qW dS )z+Closes all adapters and as such the sessionN)r   valuesr\   )rB   r)   r&   r&   r*   r\     s    zSession.closec                sB   || j  <  fdd| j D }x|D ]}| j j|| j |< q$W dS )zwRegisters a connection adapter to a prefix.

        Adapters are sorted in descending order by prefix length.
        c                s    g | ]}t |t  k r|qS r&   )rZ   )r'   r(   )r   r&   r*   r+     s    z!Session.mount.<locals>.<listcomp>N)r   rf   )rB   r   r   keys_to_mover4   r&   )r   r*   r     s    

zSession.mountc                s   t  fdd jD }|S )Nc             3   s   | ]}|t  |d fV  qd S )N)getattr)r'   attr)rB   r&   r*   	<genexpr>  s    z'Session.__getstate__.<locals>.<genexpr>)dict	__attrs__)rB   stater&   )rB   r*   __getstate__  s    zSession.__getstate__c             C   s&   x |j  D ]\}}t| || q
W d S )N)r.   setattr)rB   r   r   valuer&   r&   r*   __setstate__  s    zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)r   r   r   __doc__r   r   r   r   r   r|   r7   r   r   r   r   r   r   ro   r   r   r\   r   r   r   r&   r&   r&   r*   r   ?  s2   
7)  
D



Ir   c               C   s   t  S )zQ
    Returns a :class:`Session` for context-management.

    :rtype: Session
    )r   r&   r&   r&   r*   session  s    r   )=r   r   systimedatetimer   r   r   compatr   r   r   r   r	   r
   rj   r   r   r   r   modelsr   r   r   r   r   r   _internal_utilsr   utilsr   r   
exceptionsr   r   r   r   
structuresr   r   r   r   r   r   r    r!   r"   status_codesr#   r$   platformperf_counterr   AttributeErrorclockr5   r:   objectr;   r   r   r&   r&   r&   r*   <module>	   s@     
 a   "