
`[b                 @   s  d  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 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 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& d d l m' Z' m( Z( m) Z) m* Z* m+ Z+ d d l, m- Z- d d l m. Z. d Z/ e	 d d  Z0 e	 d d  Z1 Gd d   d e2  Z3 Gd d   d e3  Z4 d d   Z5 d S)z
requests.session
~~~~~~~~~~~~~~~~

This module provides a Session object to manage and persist settings across
requests (cookies, auth, proxies).

    N)Mapping)datetime   )_basic_auth_str)	cookielibOrderedDicturljoinurlparse)cookiejar_from_dictextract_cookies_to_jarRequestsCookieJarmerge_cookies)RequestPreparedRequestDEFAULT_REDIRECT_LIMIT)default_hooksdispatch_hook)to_key_val_listdefault_headersto_native_string)TooManyRedirectsInvalidSchemaChunkedEncodingErrorContentDecodingError)RecentlyUsedContainer)CaseInsensitiveDict)HTTPAdapter)requote_uriget_environ_proxiesget_netrc_authshould_bypass_proxiesget_auth_from_url)codes)REDIRECT_STATIi  c             C   s   | d k r |  S|  d k r  | St  | t  o; t  |  t  sB |  S| t |   } | j t |    d d   | j   D } x | D] } | | =q W| 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$   3/usr/lib/python3/dist-packages/requests/sessions.py
<listcomp>C   s   	 z!merge_setting.<locals>.<listcomp>)
isinstancer   r   updateitems)Zrequest_settingZsession_setting
dict_classZmerged_settingZ	none_keyskeyr$   r$   r(   merge_setting*   s    r/   c             C   sZ   | d k s! | j  d  g  k r% |  S|  d k sF |  j  d  g  k rJ | St |  | |  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)getr/   )Zrequest_hooksZsession_hooksr-   r$   r$   r(   merge_hooksJ   s
    !!r2   c               @   sO   e  Z d  Z d d   Z d d d d d d d  Z d d	   Z d
 d   Z d S)SessionRedirectMixinc             C   s   t  |  } t  |  } | j | j k r. d S| j d k rn | j d k rn | j d k rn | j d	 k rn d S| j | j k p | j | j k S)
zFDecide whether Authorization header should be removed when redirectingTZhttpP   NZhttps  F)r4   N)r5   N)r	   ZhostnameschemeZport)selfZold_urlnew_urlZ
old_parsedZ
new_parsedr$   r$   r(   should_strip_auth[   s    z&SessionRedirectMixin.should_strip_authFNTc             k   s%  d }	 g  }
 x| j  r | j   } |	 d k rR |
 j |  t |
  } | | _ y | j Wn. t t t f k
 r | j	 j
 d d  Yn X|	 |  j k r t d |  j   | j   | j d } | j } | j d  rt | j  } d | j | f } t |  } | j   } | j s=t | j t |   } n t |  } t |  | _ | j r| j | j k r| j |  j | j <| j t j k r| d k rd	 } | j t j k r| d k rd	 } | j t j k r| d
 k rd	 } | | _ | j t j  t j! f k r8d | j k r/| j d =d | _" | j } y | d =Wn t# k
 r`Yn Xt$ | j% | | j	  | j% j& |  j'  | j( | j%  |  j) | |  } |  j* | |  | } |  j+ | d | d | d | d | d | d d | } t$ |  j' | | j	  |	 d 7}	 | Vq Wd S)z6Receives a Response. Returns a generator of Responses.r   Zdecode_contentFzExceeded %s redirects.locationz//z%s:%sHEADGETPOSTzContent-LengthNZCookiestreamtimeoutverifycertproxiesallow_redirectsr   ),Zis_redirectcopyappendlisthistorycontentr   r   RuntimeErrorrawreadmax_redirectsr   closeheadersmethod
startswithr	   urlr6   ZgeturlZnetlocr   r   r   Zis_permanent_redirectredirect_cacheZstatus_coder"   Z	see_otherfoundZmovedZtemporary_redirectZpermanent_redirectZbodyKeyErrorr   Z_cookiesr+   cookiesZprepare_cookiesrebuild_proxiesrebuild_authsend)r7   respreqr>   r?   r@   rA   rB   Zadapter_kwargsiZhistprepared_requestZnew_histrQ   rO   Zparsed_rurlZparsedrN   r$   r$   r(   resolve_redirectsk   s~    	
			
			
z&SessionRedirectMixin.resolve_redirectsc             C   su   | j  } | j } d | k r= |  j | j j |  r= | d =|  j rR t |  n d } | d k	 rq | j |  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.
        ZAuthorizationN)rN   rQ   r9   request	trust_envr   Zprepare_auth)r7   r\   r0   rN   rQ   Znew_authr$   r$   r(   rW      s    		$z!SessionRedirectMixin.rebuild_authc             C   s   | j  } | j } t |  j } | d k	 r9 | j   n i  } |  j r t |  r t |  } | j |  } | r | j	 | | |  d | k r | d =y t
 | |  \ }	 }
 Wn t k
 r d \ }	 }
 Yn 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.
        NzProxy-Authorization)NN)rN   rQ   r	   r6   rD   r_   r    r   r1   
setdefaultr!   rT   r   )r7   r\   rB   rN   rQ   r6   Znew_proxiesZenviron_proxiesproxyZusernameZpasswordr$   r$   r(   rV      s$    		z$SessionRedirectMixin.rebuild_proxies)__name__
__module____qualname__r9   r]   rW   rV   r$   r$   r$   r(   r3   Z   s
   or3   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 d d d d d d d d d d d d d d d d  Z	 d d   Z
 d d   Z d d    Z d d d! d"  Z d d# d$  Z d d% d&  Z d' d(   Z d) d*   Z d+ d,   Z d- d.   Z d/ d0   Z d1 d2   Z d3 d4   Z d5 d6   Z d S)7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]>
    rN   rU   authrB   hooksparamsr@   rA   Zprefetchadaptersr>   r_   rL   c             C   s   t    |  _ d  |  _ i  |  _ t   |  _ i  |  _ d |  _ d |  _ d  |  _	 t
 |  _ d |  _ t i   |  _ t   |  _ |  j d t    |  j d t    t t  |  _ d  S)NFTzhttps://zhttp://)r   rN   rf   rB   r   rg   rh   r>   r@   rA   r   rL   r_   r
   rU   r   ri   mountr   r   REDIRECT_CACHE_SIZErR   )r7   r$   r$   r(   __init__/  s    								zSession.__init__c             C   s   |  S)Nr$   )r7   r$   r$   r(   	__enter__f  s    zSession.__enter__c             G   s   |  j    d  S)N)rM   )r7   argsr$   r$   r(   __exit__i  s    zSession.__exit__c             C   s$  | j  p i  } t | t j  s- t |  } t t t   |  j   |  } | j } |  j r} | r} |  j r} t	 | j
  } t   } | j d | j j   d | j
 d | j d | j d | j d t | j |  j d t d t | j |  j  d	 t | |  j  d
 | d t | j |  j   
| S)ab  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.
        rO   rQ   filesdatajsonrN   r-   rh   rf   rU   rg   )rU   r*   r   Z	CookieJarr
   r   r   rf   r_   r   rQ   r   ZpreparerO   upperrp   rq   rr   r/   rN   r   rh   r2   rg   )r7   r^   rU   Zmerged_cookiesrf   pr$   r$   r(   prepare_requestl  s*    								zSession.prepare_requestNTc             C   s   t  d | j   d | d | d | d | p- i  d | d | p? i  d | d	 | d
 |  
} |  j |  } | po i  } |  j | j | | | |  } d |	 d |
 i } | j |  |  j | |  } | S)aq  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) whether the SSL cert will be verified.
            A CA_BUNDLE path can also be provided. Defaults to ``True``.
        :param cert: (optional) if String, path to ssl client cert file (.pem).
            If Tuple, ('cert', 'key') pair.
        rO   rQ   rN   rp   rq   rr   rh   rf   rU   rg   r?   rC   )r   rs   ru   merge_environment_settingsrQ   r+   rX   )r7   rO   rQ   rh   rq   rN   rU   rp   rf   r?   rC   rB   rg   r>   r@   rA   rr   rZ   ZprepZsettingsZsend_kwargsrY   r$   r$   r(   r^     s(    2	zSession.requestc             K   s#   | j  d d  |  j d | |  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.
        rC   Tr<   )r`   r^   )r7   rQ   kwargsr$   r$   r(   r1     s    zSession.getc             K   s#   | j  d d  |  j d | |  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.
        rC   TZOPTIONS)r`   r^   )r7   rQ   rw   r$   r$   r(   options  s    zSession.optionsc             K   s#   | j  d d  |  j d | |  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.
        rC   Fr;   )r`   r^   )r7   rQ   rw   r$   r$   r(   head  s    zSession.headc             K   s   |  j  d | d | 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.
        r=   rq   rr   )r^   )r7   rQ   rq   rr   rw   r$   r$   r(   post  s    	zSession.postc             K   s   |  j  d | d | | S)a7  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.
        ZPUTrq   )r^   )r7   rQ   rq   rw   r$   r$   r(   put  s    zSession.putc             K   s   |  j  d | d | | S)a9  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.
        ZPATCHrq   )r^   )r7   rQ   rq   rw   r$   r$   r(   patch  s    zSession.patchc             K   s   |  j  d | |  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.
        ZDELETE)r^   )r7   rQ   rw   r$   r$   r(   delete!  s    zSession.deletec             K   s  | j  d |  j  | j  d |  j  | j  d |  j  | j  d |  j  t | t  sg t d   t   } xQ | j	 |  j
 k r | j | j	  |  j
 j | j	  } | | k r P| | _	 qs W| j d d  } | j d  } | j } |  j d | j	  } t j   }	 | j | |  }
 t j   |	 |
 _ t d	 | |
 |  }
 |
 j rx* |
 j D] } t |  j | j | j  q\Wt |  j | |
 j  |  j |
 | |  } | rd
 d   | D n g  } | r| j d |
  | j   }
 | |
 _ | s|
 j |
 S)zSend a given PreparedRequest.r>   r@   rA   rB   z#You can only send PreparedRequests.rC   TrQ   r0   c             S   s   g  |  ] } |  q Sr$   r$   )r%   rY   r$   r$   r(   r)   a  s   	 z Session.send.<locals>.<listcomp>r   )r`   r>   r@   rA   rB   r*   r   
ValueErrorsetrQ   rR   addr1   poprg   get_adapterr   ZutcnowrX   elapsedr   rG   r   rU   r^   rJ   r]   insertrH   )r7   r^   rw   Zchecked_urlsr8   rC   r>   rg   adapterstartrrY   genrG   r$   r$   r(   rX   *  sD    				zSession.sendc       	      C   s   |  j  r t |  p i  } x* | j   D] \ } } | j | |  q( W| d k s` | d k r t j j d  p t j j d  } t | |  j  } t | |  j	  } t | |  j
  } t | |  j  } d | d | d | d | i S)	z6Check the environment and merge it with some settings.TNZREQUESTS_CA_BUNDLEZCURL_CA_BUNDLEr@   rB   r>   rA   )r_   r   r,   r`   osenvironr1   r/   rB   r>   r@   rA   )	r7   rQ   rB   r>   r@   rA   Zenv_proxiesr&   r'   r$   r$   r(   rv   p  s    	z"Session.merge_environment_settingsc             C   sM   x6 |  j  j   D]% \ } } | j   j |  r | Sq Wt d |   d S)z=Returns the appropriate connection adapter for the given URL.z*No connection adapters were found for '%s'N)ri   r,   lowerrP   r   )r7   rQ   prefixr   r$   r$   r(   r     s    zSession.get_adapterc             C   s(   x! |  j  j   D] } | j   q Wd S)z+Closes all adapters and as such the sessionN)ri   valuesrM   )r7   r'   r$   r$   r(   rM     s    zSession.closec                sW   | |  j    <  f d d   |  j  D } x' | D] } |  j  j |  |  j  | <q0 Wd S)zkRegisters a connection adapter to a prefix.

        Adapters are sorted in descending order by key length.c                s.   g  |  ]$ } t  |  t     k  r |  q Sr$   )len)r%   r&   )r   r$   r(   r)     s   	 z!Session.mount.<locals>.<listcomp>N)ri   r   )r7   r   r   Zkeys_to_mover.   r$   )r   r(   rj     s    zSession.mountc                s9   t    f d d     j D  } t    j  | d <| S)Nc             3   s'   |  ] } | t    | d   f Vq d  S)N)getattr)r%   attr)r7   r$   r(   	<genexpr>  s    z'Session.__getstate__.<locals>.<genexpr>rR   )dict	__attrs__rR   )r7   stater$   )r7   r(   __getstate__  s    "zSession.__getstate__c             C   s|   | j  d i   } x* | j   D] \ } } t |  | |  q Wt t  |  _ x' | j   D] \ } } | |  j | <q[ Wd  S)NrR   )r   r,   setattrr   rk   rR   )r7   r   rR   r   valueZredirectZtor$   r$   r(   __setstate__  s    zSession.__setstate__)rb   rc   rd   __doc__r   rl   rm   ro   ru   r^   r1   rx   ry   rz   r{   r|   r}   rX   rv   r   rM   rj   r   r   r$   r$   r$   r(   re     sH   	7)B




	F
re   c               C   s   t    S)z2Returns a :class:`Session` for context-management.)re   r$   r$   r$   r(   session  s    r   )6r   r   collectionsr   r   rf   r   compatr   r   r   r	   rU   r
   r   r   r   Zmodelsr   r   r   rg   r   r   Zutilsr   r   r   
exceptionsr   r   r   r   Zpackages.urllib3._collectionsr   Z
structuresr   ri   r   r   r   r   r    r!   Zstatus_codesr"   r#   rk   r/   r2   objectr3   re   r   r$   r$   r$   r(   <module>
   s.   """(  