B
    Kkd”r  ã               @   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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l0m1Z1 ddlm2Z2 ej3dkrNy
ej4Z5W n e6k
rJ   ej7Z5Y nX nejZ5efdd„Z8efdd„Z9G dd„ de:ƒ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ÚDEFAULT_PORTS)Ú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ƒr,t | tƒs0| S |t|ƒƒ}| t| ƒ¡ dd„ | ¡ D ƒ}x|D ]
}||= qbW |S )zèDetermines 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-gxxfd9b7/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    



r6   c             C   s@   |dks|  d¡g kr| S | dks0|   d¡g kr4|S t| ||ƒS )z©Properly merges both requests and session hooks.

    This is necessary because when request_hooks == {'response': []}, the
    merge breaks Session hooks entirely.
    NÚresponse)Úgetr6   )Úrequest_hooksÚsession_hooksr2   r'   r'   r+   Úmerge_hooksP   s
    r;   c               @   s>   e Zd Zd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| 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_targetc             C   sš   t |ƒ}t |ƒ}|j|jkr dS |jdkrL|jdkrL|jdkrL|jdkrLdS |j|jk}|j|jk}t |jd¡df}|s’|j|kr’|j|kr’dS |p˜|S )zFDecide whether Authorization header should be removed when redirectingTÚhttp)éP   NÚhttps)i»  NFN)r	   ÚhostnameÚschemeÚportr   r8   )rC   Úold_urlÚnew_urlÚ
old_parsedÚ
new_parsedÚchanged_portÚchanged_schemeÚdefault_portr'   r'   r+   Úshould_strip_authv   s    
z&SessionRedirectMixin.should_strip_authFNTc	          
   k   sd  g }
|   |¡}t|jƒj}xB|r^| ¡ }|
 |¡ |
dd… |_y
|j W n( tt	t
fk
rv   |jjdd Y nX t|jƒ| jkrštd| j |d‚| ¡  | d¡rÈt|jƒ}dt|jƒ|f }t|ƒ}|jd	krì|rì|j|d
}n|jrø|j}| ¡ }|jst|jt|ƒƒ}nt|ƒ}t|ƒ|_|  ||¡ |jtjtjfkrtd}x|D ]}|j |d¡ qVW d|_ |j}y
|d= W n t!k
rš   Y nX t"|j#||jƒ t$|j#| j%ƒ | &|j#¡ |  '||¡}|  (||¡ |j)dk	oúd|kpúd|k}|r
t*|ƒ |}|r|V  q| j+|f|||||ddœ|	—Ž}t"| j%||jƒ |   |¡}|V  qW dS )zBReceives a Response. Returns a generator of Responses or Requests.r   NF)Údecode_contentzExceeded %s redirects.)r7   z//z%s:%sÚ )Úfragment)zContent-LengthzContent-TypezTransfer-EncodingÚCookiezContent-LengthzTransfer-Encoding)ÚstreamÚtimeoutÚverifyÚcertÚproxiesÚallow_redirects),rE   r	   ÚurlrV   ÚcopyÚappendÚhistoryÚcontentr   r   ÚRuntimeErrorÚrawÚreadÚlenÚmax_redirectsr   ÚcloseÚ
startswithr   rJ   Ú_replaceÚgeturlÚnetlocr   r   Úrebuild_methodÚstatus_coder$   Útemporary_redirectÚpermanent_redirectrA   ÚpopÚbodyÚKeyErrorr   Ú_cookiesr   ÚcookiesÚprepare_cookiesÚrebuild_proxiesÚrebuild_authÚ_body_positionr#   Úsend)rC   rD   ÚreqrX   rY   rZ   r[   r\   Úyield_requestsÚadapter_kwargsÚhistr^   Úprevious_fragmentÚprepared_requestÚparsed_rurlÚparsedÚpurged_headersÚheaderrA   Ú
rewindabler'   r'   r+   Úresolve_redirects   s|    









z&SessionRedirectMixin.resolve_redirectsc             C   sR   |j }|j}d|kr*|  |jj|¡r*|d= | jr8t|ƒnd}|dk	rN| |¡ dS )zéWhen 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)rA   r^   rS   ÚrequestÚ	trust_envr    Úprepare_auth)rC   r€   r7   rA   r^   Únew_authr'   r'   r+   rx      s    
z!SessionRedirectMixin.rebuild_authc             C   sÖ   |dk	r|ni }|j }|j}t|ƒj}| ¡ }| d¡}t||d}| jr||s|t||d}	|	 ||	 d¡¡}
|
r|| 	||
¡ d|krŠ|d= yt
|| ƒ\}}W n tk
rº   d\}}Y nX |rÒ|rÒt||ƒ|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)rA   r^   r	   rJ   r_   r8   r!   r‰   r   Ú
setdefaultr"   rs   r   )rC   r€   r\   rA   r^   rJ   Únew_proxiesrŒ   Úbypass_proxyÚenviron_proxiesÚproxyÚusernameÚpasswordr'   r'   r+   rw     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 )z‚When being redirected we may want to change the method of the request
        based on certain specs or browser behavior.
        ÚHEADÚGETÚPOSTN)Úmethodrn   r$   Ú	see_otherÚfoundÚmoved)rC   r€   r7   r˜   r'   r'   r+   rm   =  s    z#SessionRedirectMixin.rebuild_method)FNTNNF)	Ú__name__Ú
__module__Ú__qualname__rE   rS   r†   rx   rw   rm   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('https://httpbin.org/get')
      <Response [200]>

    Or as a context manager::

      >>> with requests.Session() as s:
      >>>     s.get('https://httpbin.org/get')
      <Response [200]>
    rA   ru   Úauthr\   ÚhooksÚparamsrZ   r[   ÚprefetchÚadaptersrX   r‰   rg   c             C   sr   t ƒ | _d | _i | _tƒ | _i | _d| _d| _d | _	t
| _d| _ti ƒ| _tƒ | _|  dtƒ ¡ |  dtƒ ¡ d S )NFTzhttps://zhttp://)r   rA   r    r\   r   r¡   r¢   rX   rZ   r[   r   rg   r‰   r   ru   r   r¤   Úmountr   )rC   r'   r'   r+   Ú__init__m  s    
zSession.__init__c             C   s   | S )Nr'   )rC   r'   r'   r+   Ú	__enter__¤  s    zSession.__enter__c             G   s   |   ¡  d S )N)rh   )rC   Ú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rR|sR| jsRt	|j
ƒ}tƒ }|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
        )r2   )
r˜   r^   ÚfilesÚdataÚjsonrA   r¢   r    ru   r¡   )ru   r-   r   Ú	CookieJarr   r   r   r    r‰   r    r^   r   Úpreparer˜   Úupperrª   r«   r¬   r6   rA   r   r¢   r;   r¡   )rC   rˆ   ru   Úmerged_cookiesr    Úpr'   r'   r+   Úprepare_requestª  s*    



zSession.prepare_requestNTc             C   st   t | ¡ ||||pi ||pi |||d
}|  |¡}|p8i }|  |j||||¡}|	|
dœ}| |¡ | 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, list of tuples, 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˜   r^   rA   rª   r«   r¬   r¢   r    ru   r¡   )rY   r]   )r   r¯   r²   Úmerge_environment_settingsr^   r.   rz   )rC   r˜   r^   r¢   r«   rA   ru   rª   r    rY   r]   r\   r¡   rX   rZ   r[   r¬   r{   ÚprepÚsettingsÚsend_kwargsrD   r'   r'   r+   rˆ   Ò  s(    )

zSession.requestc             K   s   |  dd¡ | jd|f|ŽS )zåSends 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
        r]   Tr–   )rŽ   rˆ   )rC   r^   Úkwargsr'   r'   r+   r8     s    zSession.getc             K   s   |  dd¡ | jd|f|ŽS )zéSends 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
        r]   TÚOPTIONS)rŽ   rˆ   )rC   r^   r·   r'   r'   r+   Úoptions$  s    zSession.optionsc             K   s   |  dd¡ | jd|f|ŽS )zæSends 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
        r]   Fr•   )rŽ   rˆ   )rC   r^   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, list of tuples, 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ˆ   )rC   r^   r«   r¬   r·   r'   r'   r+   Úpost:  s    zSession.postc             K   s   | j d|fd|i|—ŽS )au  Sends a PUT request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param data: (optional) Dictionary, list of tuples, 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ˆ   )rC   r^   r«   r·   r'   r'   r+   ÚputG  s    
zSession.putc             K   s   | j d|fd|i|—ŽS )aw  Sends a PATCH request. Returns :class:`Response` object.

        :param url: URL for the new :class:`Request` object.
        :param data: (optional) Dictionary, list of tuples, 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ˆ   )rC   r^   r«   r·   r'   r'   r+   ÚpatchS  s    
zSession.patchc             K   s   | j d|f|ŽS )zèSends 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ˆ   )rC   r^   r·   r'   r'   r+   Údelete_  s    zSession.deletec             K   s~  |  d| j¡ |  d| j¡ |  d| j¡ |  d| j¡ t|tƒrJtdƒ‚| dd¡}| 	d¡}|j
}| j|jd}tƒ }|j|f|Ž}tƒ | }	t|	d	|_td
||f|Ž}|jrÖx |jD ]}
t| j|
j|
jƒ q¼W t| j||jƒ | j||f|Ž}|r
dd„ |D ƒng }|r.| d|¡ | ¡ }||_|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
        rX   rZ   r[   r\   z#You can only send PreparedRequests.r]   T)r^   )Úsecondsr7   c             S   s   g | ]}|‘qS r'   r'   )r(   rD   r'   r'   r+   r,   œ  s    z Session.send.<locals>.<listcomp>r   r|   )rŽ   rX   rZ   r[   r\   r-   r   Ú
ValueErrorrq   r8   r¡   Úget_adapterr^   Úpreferred_clockrz   r   Úelapsedr   ra   r   ru   rˆ   rd   r†   ÚinsertÚnextÚ_nextÚStopIterationrb   )rC   rˆ   r·   r]   rX   r¡   ÚadapterÚstartÚrrÆ   rD   Úgenra   r'   r'   r+   rz   i  sB    


"zSession.sendc       
      C   s°   | j rr|dk	r| d¡nd}t||d}x | ¡ D ]\}}	| ||	¡ q2W |dksZ|dkrrtj d¡ppt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)rZ   r\   rX   r[   )r‰   r8   r   r/   rŽ   ÚosÚenvironr6   r\   rX   rZ   r[   )
rC   r^   r\   rX   rZ   r[   rŒ   Úenv_proxiesr)   r*   r'   r'   r+   r³   ²  s    z"Session.merge_environment_settingsc             C   s>   x,| j  ¡ D ]\}}| ¡  | ¡ ¡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/   Úlowerri   r   )rC   r^   ÚprefixrË   r'   r'   r+   rÄ   Ï  s    zSession.get_adapterc             C   s    x| j  ¡ D ]}| ¡  qW dS )z+Closes all adapters and as such the sessionN)r¤   Úvaluesrh   )rC   r*   r'   r'   r+   rh   Ý  s    zSession.closec                sB   || j ˆ < ‡ fdd„| j D ƒ}x|D ]}| 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'   )rf   )r(   r)   )rÕ   r'   r+   r,   è  s    z!Session.mount.<locals>.<listcomp>N)r¤   rq   )rC   rÕ   rË   Úkeys_to_mover5   r'   )rÕ   r+   r¥   â  s    

zSession.mountc                s   ‡ fdd„ˆ j D ƒ}|S )Nc                s   i | ]}t ˆ |d ƒ|“qS )N)Úgetattr)r(   Úattr)rC   r'   r+   ú
<dictcomp>î  s    z(Session.__getstate__.<locals>.<dictcomp>)Ú	__attrs__)rC   Ústater'   )rC   r+   Ú__getstate__í  s    zSession.__getstate__c             C   s&   x |  ¡ D ]\}}t| ||ƒ q
W d S )N)r/   Úsetattr)rC   rÜ   rÙ   Úvaluer'   r'   r+   Ú__setstate__ñ  s    zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rœ   r   rž   Ú__doc__rÛ   r¦   r§   r©   r²   rˆ   r8   r¹   rº   r»   r½   r¿   rÁ   rz   r³   rÄ   rh   r¥   rÝ   rà   r'   r'   r'   r+   rŸ   T  s2   
7)  
D



IrŸ   c               C   s   t ƒ S )aZ  
    Returns a :class:`Session` for context-management.

    .. deprecated:: 1.0.0

        This method has been deprecated since version 1.0.0 and is only kept for
        backwards compatibility. New code should use :class:`~requests.sessions.Session`
        to create a session. This may be removed at a future date.

    :rtype: Session
    )rŸ   r'   r'   r'   r+   Úsessionö  s    râ   )>rá   rÑ   ÚsysÚtimeÚdatetimer   r    r   Úcompatr   r   r   r   r	   r
   ru   r   r   r   r   Úmodelsr   r   r   r¡   r   r   Ú_internal_utilsr   Úutilsr   r   r   Ú
exceptionsr   r   r   r   Ú
structuresr   r¤   r   r   r   r    r!   r"   r#   Ústatus_codesr$   r%   ÚplatformÚperf_counterrÅ   ÚAttributeErrorÚclockr6   r;   Úobjectr<   rŸ   râ   r'   r'   r'   r+   Ú<module>	   s@     
 v   %