a
    Ba_u                     @   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/m0Z0 ddl1m2Z2 ddlm3Z3 ej4dkrTz
ej5Z6W n e7yP   ej8Z6Y n0 nejZ6efddZ9e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)codes)REDIRECT_STATIZwin32c                 C   sp   |du r| S | du r|S t |tr,t | ts0| S |t|}|t|  dd | D }|D ]
}||= q`|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 u r|qS N ).0kvr'   r'   3/usr/lib/python3/dist-packages/requests/sessions.py
<listcomp>J       z!merge_setting.<locals>.<listcomp>)
isinstancer
   r   updateitems)Zrequest_settingZsession_setting
dict_classZmerged_settingZ	none_keyskeyr'   r'   r+   merge_setting2   s    r3   c                 C   s@   |du s| dg kr| S | du 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)getr3   )Zrequest_hooksZsession_hooksr1   r'   r'   r+   merge_hooksQ   s
    r6   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)Zis_redirectheadersr   encoder   )selfrespr8   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v rL|jdkrL|jdv rLdS |j|jk}|j|jk}t|jddf}|s|j|v r|j|v rdS |p|S )zFDecide whether Authorization header should be removed when redirectingTZhttp)P   NZhttps)i  NFN)r	   ZhostnameschemeZportr   r5   )r=   Zold_urlZnew_urlZ
old_parsedZ
new_parsedZchanged_portZchanged_schemeZdefault_portr'   r'   r+   should_strip_authw   s$    z&SessionRedirectMixin.should_strip_authFNTc	              
   k   sF  g }
|  |}t|jj}|rB| }|
| |
dd |_z
|j W n& tt	t
fyp   |jjdd Y n0 t|j| jkrtd| j|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vrnd}|D ]}|j !|d qRd|_"|j }|!dd t#|j$||j t%|j$| j& |'|j$ | (||}| )|| |j*duod|v pd|v }|rt+| |}|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)Zdecode_contentzExceeded {} redirects.)r4   z//: )fragment)Content-LengthzContent-TypeTransfer-EncodingZCookierF   rG   )streamtimeoutverifycertproxiesallow_redirects)-r?   r	   urlrE   copyappendhistorycontentr   r   RuntimeErrorrawreadlenmax_redirectsr   formatclose
startswithjoinr   rA   _replaceZgeturlZnetlocr   r   rebuild_methodstatus_coder$   Ztemporary_redirectZpermanent_redirectr;   popbodyr   Z_cookiesr   cookiesZprepare_cookiesrebuild_proxiesrebuild_authZ_body_positionr#   send)r=   r>   reqrH   rI   rJ   rK   rL   yield_requestsZadapter_kwargsZhistrN   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|v r*| |jj|r*|d= | jr8t|nd}|du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;   rN   rB   request	trust_envr    Zprepare_auth)r=   rg   r4   r;   rN   Znew_authr'   r'   r+   rc      s    z!SessionRedirectMixin.rebuild_authc                 C   s   |dur|ni }|j }|j}t|j}| }|d}t||d}| jr||s|t||d}	|	||	d}
|
r||	||
 d|v r|d= zt
|| \}}W n ty   d\}}Y n0 |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rl   allzProxy-Authorization)NN)r;   rN   r	   rA   rO   r5   r!   rk   r   
setdefaultr"   KeyErrorr   )r=   rg   rL   r;   rN   rA   Znew_proxiesrl   Zbypass_proxyZenviron_proxiesproxyZusernameZpasswordr'   r'   r+   rb     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)methodr^   r$   Z	see_otherfoundZmoved)r=   rg   r4   ru   r'   r'   r+   r]   :  s    z#SessionRedirectMixin.rebuild_method)FNTNNF)	__name__
__module____qualname__r?   rB   ri   rc   rb   r]   r'   r'   r'   r+   r7   `   s     
n)r7   c                   @   s   e Zd ZdZg dZdd Zdd Zdd Zd	d
 Zd+ddZ	dd Z
dd Zdd Z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#d$ Zd%d& Zd'd( Zd)d* 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;   ra   authrL   hooksparamsrJ   rK   adaptersrH   rk   rW   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;   r{   rL   r   r|   r}   rH   rJ   rK   r   rW   rk   r   ra   r   r~   mountr   r=   r'   r'   r+   __init__j  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&   )rY   )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
        )r1   )
ru   rN   filesdatajsonr;   r}   r{   ra   r|   )ra   r.   r   Z	CookieJarr   r   r   r{   rk   r    rN   r   Zprepareru   upperr   r   r   r3   r;   r   r}   r6   r|   )r=   rj   ra   Zmerged_cookiesr{   pr'   r'   r+   prepare_request  s.    



zSession.prepare_requestNTc                 C   sx   t | ||||pi ||pi |||d
}| |}|p8i }| |j||||}|	|
d}|| | j|fi |}|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
        )
ru   rN   r;   r   r   r   r}   r{   ra   r|   )rI   rM   )r   r   r   merge_environment_settingsrN   r/   rd   )r=   ru   rN   r}   r   r;   ra   r   r{   rI   rM   rL   r|   rH   rJ   rK   r   re   ZprepZsettingsZsend_kwargsr>   r'   r'   r+   rj     s.    .

zSession.requestc                 K   s    | dd | jd|fi |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
        rM   Trs   ro   rj   r=   rN   kwargsr'   r'   r+   r5   "  s    zSession.getc                 K   s    | dd | jd|fi |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
        rM   TZOPTIONSr   r   r'   r'   r+   options-  s    zSession.optionsc                 K   s    | dd | jd|fi |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
        rM   Frr   r   r   r'   r'   r+   head8  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
        rt   )r   r   rj   )r=   rN   r   r   r   r'   r'   r+   postC  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   r   r=   rN   r   r   r'   r'   r+   putP  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   r   r   r'   r'   r+   patch\  s    
zSession.patchc                 K   s   | j d|fi |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DELETEr   r   r'   r'   r+   deleteh  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i |}t | }	t|	d	|_td
||fi |}|jr|jD ]}
t| j|
j|
j qt| j||j |r| j||fi |}dd |D }ng }|r8|d| | }||_|svz"t| j||fddi||_W n tyt   Y n0 |s|j |S )zISend a given PreparedRequest.

        :rtype: requests.Response
        rH   rJ   rK   rL   z#You can only send PreparedRequests.rM   T)rN   )Zsecondsr4   c                 S   s   g | ]}|qS r'   r'   )r(   r>   r'   r'   r+   r,     r-   z Session.send.<locals>.<listcomp>r   rf   )ro   rH   rJ   rK   rL   r.   r   
ValueErrorr_   r5   r|   get_adapterrN   preferred_clockrd   r   elapsedr   rQ   r   ra   rj   rT   ri   insertnextZ_nextStopIterationrR   )r=   rj   r   rM   rH   r|   adapterstartrr   r>   genrQ   r'   r'   r+   rd   r  sF    



"zSession.sendc           
      C   s   | j rn|dur|dnd}t||d}| D ]\}}	|||	 q0|du sV|du 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
        Nrl   rm   TZREQUESTS_CA_BUNDLEZCURL_CA_BUNDLE)rJ   rL   rH   rK   )rk   r5   r   r0   ro   osenvironr3   rL   rH   rJ   rK   )
r=   rN   rL   rH   rJ   rK   rl   Z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~   r0   lowerrZ   r   rX   )r=   rN   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~   valuesrY   )r=   r*   r'   r'   r+   rY     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'   )rV   )r(   r)   r   r'   r+   r,     r-   z!Session.mount.<locals>.<listcomp>N)r~   r_   )r=   r   r   Zkeys_to_mover2   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>  r-   z(Session.__getstate__.<locals>.<dictcomp>)	__attrs__)r=   stater'   r   r+   __getstate__  s    zSession.__getstate__c                 C   s"   |  D ]\}}t| || qd S r&   )r0   setattr)r=   r   r   valuer'   r'   r+   __setstate__  s    zSession.__setstate__)NNNNNNNTNNNNNN)NN)N)N)rw   rx   ry   __doc__r   r   r   r   r   rj   r5   r   r   r   r   r   r   rd   r   r   rY   r   r   r   r'   r'   r'   r+   rz   Q  s0   >)   
L



Krz   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
    )rz   r'   r'   r'   r+   session  s    r   )?r   r   systimeZdatetimer   collectionsr   r{   r   compatr   r   r   r	   r
   ra   r   r   r   r   Zmodelsr   r   r   r|   r   r   Z_internal_utilsr   Zutilsr   r   r   
exceptionsr   r   r   r   Z
structuresr   r~   r   r   r   r    r!   r"   r#   Zstatus_codesr$   r%   platformperf_counterr   AttributeErrorZclockr3   r6   objectr7   rz   r   r'   r'   r'   r+   <module>   sB    
 r   3