o
    9kbu                     @   sj  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re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   sp   |du r| S | du r|S t |trt | ts| S |t|}|t|  dd | D }|D ]}||= q0|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(   5usr/lib/python3.10/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   r3   c                 C   s@   |du s| dg kr| S | du s|  dg kr|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_hooks[   s
   r6   c                   @   sJ   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respr8   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r&|jdv r&|jdkr&|jdv r&dS |j|jk}|j|jk}t|jddf}|sI|j|v rI|j|v rIdS |pL|S )zFDecide whether Authorization header should be removed when redirectingThttp)P   Nhttps)i  NFN)r   hostnameschemeportr   r5   )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   sB   g }
|  |}t|jj}|r| }|
| |
dd |_z|j W n tt	t
fy9   |jjdd Y nw t|j| jkrMtd| j d|d|  |dret|j}d	t|j|g}t|}|jd
krw|rw|j|d}n|jr}|j}| }|jst|jt|}nt|}t||_| || |jtjtjfvrd}|D ]	}|j |d qd|_!|j}| dd t"|j#||j t$|j#| j% |&|j# | '||}| (|| |j)duod|v pd|v }|rt*| |}|r|V  n | j+|f|||||dd|	}t"| j%||j |  |}|V  |sdS dS )zBReceives a Response. Returns a generator of Responses or Requests.r   NF)Zdecode_contentz	Exceeded z redirects.)r4   z//: )fragment)Content-LengthzContent-TypeTransfer-EncodingZCookierJ   rK   )streamtimeoutverifycertproxiesallow_redirects),r?   r   urlrI   copyappendhistorycontentr   r   RuntimeErrorrawreadlenmax_redirectsr   close
startswithjoinr   rD   _replacegeturl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>   reqrL   rM   rN   rO   rP   yield_requestsZadapter_kwargsZhistrR   Zprevious_fragmentprepared_requestZparsed_rurlparsedZpurged_headersheaderr;   Z
rewindabler(   r(   r,   resolve_redirects   s   







z&SessionRedirectMixin.resolve_redirectsc                 C   sV   |j }|j}d|v r| |jj|r|d= | jrt|nd}|dur)|| dS 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;   rR   rF   request	trust_envr    Zprepare_auth)r=   rl   r4   r;   rR   Znew_authr(   r(   r,   rh     s   z!SessionRedirectMixin.rebuild_authc                 C   sx   |j }t|jj}t||| j}d|v r|d= z
t|| \}}W n ty.   d\}}Y nw |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
        zProxy-AuthorizationNN)	r;   r   rR   rD   r"   rq   r   KeyErrorr   )r=   rl   rP   r;   rD   Znew_proxiesusernamepasswordr(   r(   r,   rg   /  s   z$SessionRedirectMixin.rebuild_proxiesc                 C   sX   |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}||_ 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)methodrc   r   Z	see_otherfoundZmoved)r=   rl   r4   ry   r(   r(   r,   rb   L  s   
z#SessionRedirectMixin.rebuild_method)FNTNNF)	__name__
__module____qualname__r?   rF   ro   rh   rg   rb   r(   r(   r(   r,   r7   j   s    $
|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;   rf   authrP   hooksparamsrN   rO   adaptersrL   rq   r[   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   rP   r   r   r   rL   rN   rO   r   r[   rq   r   rf   r   r   mountr   r=   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'   )r\   )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r)|s)| js)t	|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   )
ry   rR   filesdatajsonr;   r   r   rf   r   )rf   r.   r	   Z	CookieJarr   r   r   r   rq   r    rR   r   Zpreparery   upperr   r   r   r3   r;   r   r   r6   r   )r=   rp   rf   Zmerged_cookiesr   pr(   r(   r,   prepare_request  s2   




zSession.prepare_requestNTc                 C   sx   t | ||||p
i ||pi |||d
}| |}|pi }| |j||||}|	|
d}|| | j|fi |}|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
        )
ry   rR   r;   r   r   r   r   r   rf   r   )rM   rQ   )r   r   r   merge_environment_settingsrR   r/   ri   )r=   ry   rR   r   r   r;   rf   r   r   rM   rQ   rP   r   rL   rN   rO   r   rj   ZprepsettingsZsend_kwargsr>   r(   r(   r,   rp     s.   =

zSession.requestc                 K       | 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
        rQ   Trw   
setdefaultrp   r=   rR   kwargsr(   r(   r,   r5   O     zSession.getc                 K   r   )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
        rQ   TZOPTIONSr   r   r(   r(   r,   optionsZ  r   zSession.optionsc                 K   r   )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
        rQ   Frv   r   r   r(   r(   r,   heade  r   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
        rx   )r   r   rp   )r=   rR   r   r   r   r(   r(   r,   postp  s   zSession.postc                 K      | 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=   rR   r   r   r(   r(   r,   put}     
zSession.putc                 K   r   )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  r   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,   delete  s   zSession.deletec                 K   s  | d| j | d| j | d| j d|vr#t|| j| j|d< t|tr,t	d|
dd}|d}|j}| j|jd}t }|j|fi |}t | }	t|	d	|_td
||fi |}|jrt|jD ]}
t| j|
j|
j qht| j||j |r| j||fi |}dd |D }ng }|r|d| |
 }||_|szt| j||fddi||_W n	 ty   Y nw |s|j |S )zISend a given PreparedRequest.

        :rtype: requests.Response
        rL   rN   rO   rP   z#You can only send PreparedRequests.rQ   T)rR   )secondsr4   c                 S   s   g | ]}|qS r(   r(   )r)   r>   r(   r(   r,   r-     s    z Session.send.<locals>.<listcomp>r   rk   )r   rL   rN   rO   r"   rP   rq   r.   r   
ValueErrorrd   r5   r   get_adapterrR   preferred_clockri   r   elapsedr   rU   r   rf   rp   rX   ro   insertnext_nextStopIterationrV   )r=   rp   r   rQ   rL   r   adapterstartrr   r>   genrU   r(   r(   r,   ri     sN   




zSession.sendc           
      C   s   | j r9|dur|dnd}t||d}| D ]
\}}	|||	 q|du s+|du r9tjdp8tjdp8|}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)rP   rL   rN   rO   )rq   r5   r   r0   r   osenvironr3   rP   rL   rN   rO   )
r=   rR   rP   rL   rN   rO   r   Zenv_proxiesr*   r+   r(   r(   r,   r     s    
z"Session.merge_environment_settingsc                 C   s<   | j  D ]\}}| | r|  S qtd|)z~
        Returns the appropriate connection adapter for the given URL.

        :rtype: requests.adapters.BaseAdapter
        z&No connection adapters were found for )r   r0   lowerr]   r   )r=   rR   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   valuesr\   )r=   r+   r(   r(   r,   r\     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(   )rZ   )r)   r*   r   r(   r,   r-   %  s     z!Session.mount.<locals>.<listcomp>N)r   rd   )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>+  s    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__)NNNNNNNTNNNNNNrr   r'   )r{   r|   r}   __doc__r   r   r   r   r   rp   r5   r   r   r   r   r   r   ri   r   r   r\   r   r   r   r(   r(   r(   r,   r~   c  sH    >/
[



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,   session3  s   r   )<r   r   systimecollectionsr   datetimer   Z_internal_utilsr   r   r   r   r   compatr   r	   r
   r   rf   r   r   r   r   
exceptionsr   r   r   r   r   r   r   Zmodelsr   r   r   r   Zstatus_codesr   Z
structuresr   utilsr   r   r   r   r    r!   r"   r#   r$   r%   platformperf_counterr   r3   r6   r7   r~   r   r(   r(   r(   r,   <module>   s:    0
 z   S