B
    +cu                 @   sl  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 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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/m0Z0m1Z1m2Z2m3Z3 ej4dkr$ej5Z6nejZ6efddZ7efddZ8G dd dZ9G dd de9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)OrderedDict)	timedelta   )to_native_string)HTTPAdapter)_basic_auth_str)Mapping	cookieliburljoinurlparse)RequestsCookieJarcookiejar_from_dictextract_cookies_to_jarmerge_cookies)ChunkedEncodingErrorContentDecodingErrorInvalidSchemaTooManyRedirects)default_hooksdispatch_hook)DEFAULT_REDIRECT_LIMITREDIRECT_STATIPreparedRequestRequest)codes)CaseInsensitiveDict)
DEFAULT_PORTSdefault_headersget_auth_from_urlget_environ_proxiesget_netrc_authrequote_uriresolve_proxiesrewind_bodyshould_bypass_proxiesto_key_val_list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 )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'   0lib/python3.7/site-packages/requests/sessions.py
<listcomp>T   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    

r2   c             C   s@   |dks| dg kr| S | dks0|  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)getr2   )Zrequest_hooksZsession_hooksr0   r'   r'   r+   merge_hooks[   s
    r5   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 }|d}t|dS dS )z7Receives a Response. Returns a redirect URI or ``None``locationlatin1utf8N)Zis_redirectheadersencoder   )selfrespr7   r'   r'   r+   get_redirect_targetk   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rdS |p|S )zFDecide whether Authorization header should be removed when redirectingTZhttp)P   NZhttps)i  NFN)r   ZhostnameschemeZportr   r4   )r<   Zold_urlZnew_urlZ
old_parsedZ
new_parsedZchanged_portZchanged_schemeZdefault_portr'   r'   r+   should_strip_auth   s"    





z&SessionRedirectMixin.should_strip_authFNTc	          
   k   sT  g }
|  |}t|jj}x2|rN| }|
| |
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|d|  |drt|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rzd}x|D ]}|j |d q\W d|_!|j}| dd t"|j#||j t$|j#| j% |&|j# | '||}| (|| |j)dk	od|kpd|k}|rt*| |}|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)Zdecode_contentz	Exceeded z redirects.)r3   z//: )fragment)zContent-LengthzContent-TypezTransfer-EncodingZCookiezContent-LengthzTransfer-Encoding)streamtimeoutverifycertproxiesallow_redirects),r>   r   urlrD   copyappendhistorycontentr   r   RuntimeErrorrawreadlenmax_redirectsr   close
startswithjoinr   r@   _replaceZgeturlZnetlocr
   r!   rebuild_methodstatus_coder   Ztemporary_redirectZpermanent_redirectr:   popZbodyr   Z_cookiesr   cookiesZprepare_cookiesrebuild_proxiesrebuild_authZ_body_positionr#   send)r<   r=   reqrE   rF   rG   rH   rI   yield_requestsZadapter_kwargsZhistrK   Zprevious_fragmentprepared_requestZparsed_rurlZparsedZpurged_headersheaderr:   Z
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 )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.
        ZAuthorizationN)r:   rK   rA   request	trust_envr    Zprepare_auth)r<   rb   r3   r:   rK   Znew_authr'   r'   r+   r^     s    z!SessionRedirectMixin.rebuild_authc             C   sz   |j }t|jj}t||| j}d|kr.|d= y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)	r:   r   rK   r@   r"   rf   r   KeyErrorr   )r<   rb   rI   r:   r@   Znew_proxiesZusernameZ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 )zWhen being redirected we may want to change the method of the request
        based on certain specs or browser behavior.
        HEADGETPOSTN)methodrZ   r   Z	see_otherfoundZmoved)r<   rb   r3   rk   r'   r'   r+   rY   L  s    z#SessionRedirectMixin.rebuild_method)FNTNNF)	__name__
__module____qualname__r>   rA   rd   r^   r]   rY   r'   r'   r'   r+   r6   j   s   $     
sr6   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]>
    r:   r\   authrI   hooksparamsrG   rH   adaptersrE   rf   rT   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   r:   rq   rI   r   rr   rs   rE   rG   rH   r   rT   rf   r   r\   r   rt   mountr   )r<   r'   r'   r+   __init__  s    

zSession.__init__c             C   s   | S )Nr'   )r<   r'   r'   r+   	__enter__  s    zSession.__enter__c             G   s   |    d S )N)rU   )r<   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
        )r0   )
rk   rK   filesdatajsonr:   rs   rq   r\   rr   )r\   r-   r	   Z	CookieJarr   r   r   rq   rf   r    rK   r   Zpreparerk   upperrz   r{   r|   r2   r:   r   rs   r5   rr   )r<   re   r\   Zmerged_cookiesrq   pr'   r'   r+   prepare_request  s,    



zSession.prepare_requestNTc             C   st   t | ||||pi ||pi |||d
}| |}|p8i }| |j||||}|	|
d}|| | j|f|}|S )aw	  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
        )
rk   rK   r:   rz   r{   r|   rs   rq   r\   rr   )rF   rJ   )r   r}   r   merge_environment_settingsrK   r.   r_   )r<   rk   rK   rs   r{   r:   r\   rz   rq   rF   rJ   rI   rr   rE   rG   rH   r|   r`   ZprepZsettingsZsend_kwargsr=   r'   r'   r+   re     s(    =

zSession.requestc             K   s   | 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
        rJ   Tri   )
setdefaultre   )r<   rK   kwargsr'   r'   r+   r4   O  s    zSession.getc             K   s   | 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
        rJ   TZOPTIONS)r   re   )r<   rK   r   r'   r'   r+   optionsZ  s    zSession.optionsc             K   s   | 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
        rJ   Frh   )r   re   )r<   rK   r   r'   r'   r+   heade  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
        rj   )r{   r|   )re   )r<   rK   r{   r|   r   r'   r'   r+   postp  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
        ZPUTr{   )re   )r<   rK   r{   r   r'   r'   r+   put}  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
        ZPATCHr{   )re   )r<   rK   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
        ZDELETE)re   )r<   rK   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rx |jD ]}
t| j|
j|
j qW t| j||j |r| j||f|}dd |D }ng }|r>|d| |
 }||_|s~y"t| j||fddi||_W n tk
r|   Y nX |s|j |S )zISend a given PreparedRequest.

        :rtype: requests.Response
        rE   rG   rH   rI   z#You can only send PreparedRequests.rJ   T)rK   )Zsecondsr3   c             S   s   g | ]}|qS r'   r'   )r(   r=   r'   r'   r+   r,     s    z Session.send.<locals>.<listcomp>r   ra   )r   rE   rG   rH   r"   rI   rf   r-   r   
ValueErrorr[   r4   rr   get_adapterrK   preferred_clockr_   r   elapsedr   rN   r   r\   re   rQ   rd   insertnextZ_nextStopIterationrO   )r<   re   r   rJ   rE   rr   adapterstartrr   r=   genrN   r'   r'   r+   r_     sJ    


 zSession.sendc       
      C   s   | j rv|dk	r|dnd}t||d}x | D ]\}}	|||	 q2W |dksZ|dkrvtjdpttjdpt|}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   TZREQUESTS_CA_BUNDLEZCURL_CA_BUNDLE)rI   rE   rG   rH   )rf   r4   r   r/   r   osenvironr2   rI   rE   rG   rH   )
r<   rK   rI   rE   rG   rH   r   Z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 N)rt   r/   lowerrV   r   )r<   rK   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)rt   valuesrU   )r<   r*   r'   r'   r+   rU     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'   )rS   )r(   r)   )r   r'   r+   r,   %  s    z!Session.mount.<locals>.<listcomp>N)rt   r[   )r<   r   r   Zkeys_to_mover1   r'   )r   r+   ru     s    

zSession.mountc                s    fdd j D }|S )Nc                s   i | ]}t  |d |qS )N)getattr)r(   attr)r<   r'   r+   
<dictcomp>+  s    z(Session.__getstate__.<locals>.<dictcomp>)	__attrs__)r<   stater'   )r<   r+   __getstate__*  s    zSession.__getstate__c             C   s&   x |  D ]\}}t| || q
W d S )N)r/   setattr)r<   r   r   valuer'   r'   r+   __setstate__.  s    zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rm   rn   ro   __doc__r   rv   rw   ry   r   re   r4   r   r   r   r   r   r   r_   r   r   rU   ru   r   r   r'   r'   r'   r+   rp   c  sZ   >/             
J



Nrp   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
    )rp   r'   r'   r'   r+   session3  s    r   )<r   r   sysZtimecollectionsr   Zdatetimer   Z_internal_utilsr   rt   r   rq   r   compatr   r	   r
   r   r\   r   r   r   r   
exceptionsr   r   r   r   rr   r   r   Zmodelsr   r   r   r   Zstatus_codesr   Z
structuresr   Zutilsr   r   r   r   r    r!   r"   r#   r$   r%   platformZperf_counterr   r2   r5   r6   rp   r   r'   r'   r'   r+   <module>   s8   0 z   S