U
    Â½„b‹t  ã                   @   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	 ddl
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/m0Z0m1Z1 ddl2m3Z3 ddlm4Z4 ej5dk�rZz
ej6Z7W n e8k
�rV   ej9Z7Y nX nejZ7efdd„Z:efdd„Z;G dd„ de<ƒZ=G dd„ de=ƒZ>dd„ Z?dS )z•
requests.sessions
~~~~~~~~~~~~~~~~~

This module provides a Session object to manage and persist settings across
requests (cookies, auth, proxies).
é    N)Ú	timedelta)ÚOrderedDicté   )Ú_basic_auth_str)Ú	cookielibÚis_py3Ú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Úresolve_proxies)Úcodes)ÚREDIRECT_STATIÚwin32c                 C   sp   |dkr| S | dkr|S t |tƒr,t | tƒs0| S |t|ƒƒ}| t| ƒ¡ dd„ | ¡ D ƒ}|D ]
}||= q`|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)   ú5/tmp/pip-unpacked-wheel-pr7jkl2n/requests/sessions.pyÚ
<listcomp>J   s      z!merge_setting.<locals>.<listcomp>)Ú
isinstancer
   r   ÚupdateÚitems)Úrequest_settingÚsession_settingÚ
dict_classÚmerged_settingÚ	none_keysÚkeyr)   r)   r-   Úmerge_setting2   s    ÿþr8   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)Úgetr8   )Úrequest_hooksÚsession_hooksr4   r)   r)   r-   Úmerge_hooksQ   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_targetb   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   r:   )rE   Úold_urlÚnew_urlÚ
old_parsedÚ
new_parsedÚchanged_portÚchanged_schemeÚdefault_portr)   r)   r-   Úshould_strip_authw   s$    ÿÿÿz&SessionRedirectMixin.should_strip_authFNTc	              
   k   sH  g }
|   |¡}t|jƒj}|�rD| ¡ }|
 |¡ |
dd… |_z
|j W n( tt	t
fk
rr   |jjdd� Y nX t|jƒ| jkr˜td | j¡|d�‚| ¡  | d¡rÈt|jƒ}d t|jƒ|g¡}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�rpd}|D ]}|j  !|d¡ �qTd|_"|j }| !dd¡ 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dS )zBReceives a Response. Returns a generator of Responses or Requests.r   NF)Údecode_contentzExceeded {} redirects.)r9   z//ú:Ú )Úfragment)úContent-LengthzContent-TypeúTransfer-EncodingÚCookierZ   r[   )ÚstreamÚtimeoutÚverifyÚcertÚproxiesÚallow_redirects)-rG   r	   ÚurlrY   ÚcopyÚappendÚhistoryÚcontentr   r   ÚRuntimeErrorÚrawÚreadÚlenÚmax_redirectsr   ÚformatÚcloseÚ
startswithÚjoinr   rL   Ú_replaceÚgeturlÚnetlocr   r   Úrebuild_methodÚstatus_coder%   Útemporary_redirectÚpermanent_redirectrC   ÚpopÚbodyr   Ú_cookiesr   ÚcookiesÚprepare_cookiesÚrebuild_proxiesÚrebuild_authÚ_body_positionr#   Úsend)rE   rF   Úreqr]   r^   r_   r`   ra   Úyield_requestsÚadapter_kwargsÚhistrc   Úprevious_fragmentÚprepared_requestÚparsed_rurlÚparsedÚpurged_headersÚheaderrC   Ú
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)rC   rc   rU   ÚrequestÚ	trust_envr    Úprepare_auth)rE   r†   r9   rC   rc   Únew_authr)   r)   r-   r~   þ   s    z!SessionRedirectMixin.rebuild_authc                 C   sz   |j }t|jƒj}t||| jƒ}d|kr.|d= zt|| ƒ\}}W n tk
r^   d\}}Y nX |rv|rvt||ƒ|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
        zProxy-Authorization)NN)	rC   r	   rc   rL   r$   r�   r"   ÚKeyErrorr   )rE   r†   ra   rC   rL   Únew_proxiesÚusernameÚpasswordr)   r)   r-   r}     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)Úmethodru   r%   Ú	see_otherÚfoundÚmoved)rE   r†   r9   r™   r)   r)   r-   rt   -  s    z#SessionRedirectMixin.rebuild_method)FNTNNF)	Ú__name__Ú
__module__Ú__qualname__rG   rU   rŒ   r~   r}   rt   r)   r)   r)   r-   r>   `   s           ÿ
nr>   c                   @   sÌ   e Zd ZdZ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d6dd„Z	dd„ Z
dd„ Zdd„ Zd7d d!„Zd8d"d#„Zd9d$d%„Zd&d'„ Zd(d)„ Zd*d+„ Zd,d-„ Zd.d/„ Zd0d1„ Zd2d3„ Zd4d5„ 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]>
    rC   r{   Úauthra   ÚhooksÚparamsr_   r`   Úadaptersr]   r�   rl   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   rC   r¡   ra   r   r¢   r£   r]   r_   r`   r   rl   r�   r   r{   r   r¤   Úmountr   ©rE   r)   r)   r-   Ú__init__]  s    

zSession.__init__c                 C   s   | S r(   r)   r¦   r)   r)   r-   Ú	__enter__›  s    zSession.__enter__c                 G   s   |   ¡  d S r(   )rn   )rE   Ú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
        )r4   )
r™   rc   ÚfilesÚdataÚjsonrC   r£   r¡   r{   r¢   )r{   r/   r   Ú	CookieJarr   r   r   r¡   r�   r    rc   r   Úpreparer™   Úupperr«   r¬   r­   r8   rC   r   r£   r=   r¢   )rE   rŽ   r{   Úmerged_cookiesr¡   Úpr)   r)   r-   Úprepare_request¡  s0    

 ÿ

özSession.prepare_requestNTc                 C   st   t | ¡ ||||pi ||pi |||d�
}|  |¡}|p8i }|  |j||||¡}|	|
dœ}| |¡ | j|f|Ž}|S )ax	  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``. When set to
            ``False``, requests will accept any TLS certificate presented by
            the server, and will ignore hostname mismatches and/or expired
            certificates, which will make your application vulnerable to
            man-in-the-middle (MitM) attacks. Setting verify to ``False`` 
            may be useful during local development or testing.
        :param cert: (optional) if String, path to ssl client cert file (.pem).
            If Tuple, ('cert', 'key') pair.
        :rtype: requests.Response
        )
r™   rc   rC   r«   r¬   r­   r£   r¡   r{   r¢   )r^   rb   )r   r°   r³   Úmerge_environment_settingsrc   r0   r€   )rE   r™   rc   r£   r¬   rC   r{   r«   r¡   r^   rb   ra   r¢   r]   r_   r`   r­   r�   ÚprepÚsettingsÚsend_kwargsrF   r)   r)   r-   rŽ   É  s6    .ö
    ÿþ
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
        rb   Tr—   ©Ú
setdefaultrŽ   ©rE   rc   Úkwargsr)   r)   r-   r:     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
        rb   TÚOPTIONSr¸   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
        rb   Fr–   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Ž   )rE   rc   r¬   r­   r»   r)   r)   r-   Úpost6  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¿   ©rE   rc   r¬   r»   r)   r)   r-   ÚputC  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¿   rÂ   r)   r)   r-   ÚpatchO  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
        ÚDELETEr¿   rº   r)   r)   r-   Údelete[  s    zSession.deletec                 K   sŠ  |  d| j¡ |  d| j¡ |  d| j¡ d|krFt|| j| jƒ|d< t|tƒrXt	dƒ‚| 
dd¡}| d¡}|j}| j|jd�}tƒ }|j|f|Ž}tƒ | }	t|	d	�|_td
||f|Ž}|jrà|jD ]}
t| j|
j|
jƒ qÈt| j||jƒ |�r| j||f|Ž}dd„ |D ƒ}ng }|�r:| d|¡ | 
¡ }||_|�szz"t| j||fddi|—Žƒ|_W n tk
�rx   Y nX |�s†|j |S )zISend a given PreparedRequest.

        :rtype: requests.Response
        r]   r_   r`   ra   z#You can only send PreparedRequests.rb   T)rc   )Úsecondsr9   c                 S   s   g | ]}|‘qS r)   r)   )r*   rF   r)   r)   r-   r.   ›  s     z Session.send.<locals>.<listcomp>r   r‚   )r¹   r]   r_   r`   r$   ra   r�   r/   r   Ú
ValueErrorrx   r:   r¢   Úget_adapterrc   Úpreferred_clockr€   r   Úelapsedr   rf   r   r{   rŽ   ri   rŒ   ÚinsertÚnextÚ_nextÚStopIterationrg   )rE   rŽ   r»   rb   r]   r¢   ÚadapterÚstartÚrrÌ   rF   Úgenrf   r)   r)   r-   r€   e  sP      ÿ



"zSession.sendc           
      C   s¬   | j rn|dk	r| d¡nd}t||d�}| ¡ D ]\}}	| ||	¡ q0|dksV|dkrntj d¡pltj d¡}t|| jƒ}t|| j	ƒ}t|| j
ƒ}t|| jƒ}||||dœS )z^
        Check the environment and merge it with some settings.

        :rtype: dict
        NÚno_proxy)rÕ   TÚREQUESTS_CA_BUNDLEÚCURL_CA_BUNDLE)r_   ra   r]   r`   )r�   r:   r   r1   r¹   ÚosÚenvironr8   ra   r]   r_   r`   )
rE   rc   ra   r]   r_   r`   rÕ   Úenv_proxiesr+   r,   r)   r)   r-   r´   ³  s     
ÿÿz"Session.merge_environment_settingsc                 C   s@   | j  ¡ D ]"\}}| ¡  | ¡ ¡r
|  S q
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 {!r}N)r¤   r1   Úlowerro   r   rm   )rE   rc   ÚprefixrÑ   r)   r)   r-   rÊ   Ð  s    
zSession.get_adapterc                 C   s   | j  ¡ D ]}| ¡  q
dS )z+Closes all adapters and as such the sessionN)r¤   Úvaluesrn   )rE   r,   r)   r)   r-   rn   Þ  s    zSession.closec                    s>   || j ˆ < ‡ fdd„| j D ƒ}|D ]}| j  |¡| j |< q"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)   )rk   )r*   r+   ©rÜ   r)   r-   r.   é  s      z!Session.mount.<locals>.<listcomp>N)r¤   rx   )rE   rÜ   rÑ   Úkeys_to_mover7   r)   rÞ   r-   r¥   ã  s    
zSession.mountc                    s   ‡ fdd„ˆ j D ƒ}|S )Nc                    s   i | ]}|t ˆ |d ƒ“qS r(   )Úgetattr)r*   Úattrr¦   r)   r-   Ú
<dictcomp>ï  s      z(Session.__getstate__.<locals>.<dictcomp>)Ú	__attrs__)rE   Ústater)   r¦   r-   Ú__getstate__î  s    zSession.__getstate__c                 C   s"   |  ¡ D ]\}}t| ||ƒ qd S r(   )r1   Úsetattr)rE   rä   rá   Úvaluer)   r)   r-   Ú__setstate__ò  s    zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)r�   rž   rŸ   Ú__doc__rã   r§   r¨   rª   r³   rŽ   r:   r½   r¾   rÀ   rÃ   rÅ   rÇ   r€   r´   rÊ   rn   r¥   rå   rè   r)   r)   r)   r-   r    D  s^            ý>)                         ý
L



N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   Úcollectionsr   r¡   r   Úcompatr   r   r   r	   r
   r{   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#   r$   Ústatus_codesr%   r&   ÚplatformÚperf_counterrË   ÚAttributeErrorZclockr8   r=   Úobjectr>   r    rê   r)   r)   r)   r-   Ú<module>   sB   $
 e   6