o

��Yh�Q�@s6
dd
lm
Z

ddlZddlZddlZddlmZmZ
ddlm	Z	m
Z
mZ
ddlm
Z
mZmZmZmZ
ddlZddlmZ
ddlmZ
dd	lmZ
dd
lmZ
ddlmZ
gd�
Ze�e�
Z d
Z!dZ"e�#de"�Z$dZ%e�#de%�Z&dd�Z'e�(e'e$�e�(e'e$�d�Z)e	e
d�Z*Gdd�de�Z+Gdd�de+�Z,dd�Z-dS)�)
�absolute_importN�
)�HTTPHeaderDict�RecentlyUsedContainer)�HTTPConnectionPool�HTTPSConnectionPool�port_by_scheme)�LocationValueError�
MaxRetryError�ProxySchemeUnknown�ProxySchemeUnsupported�URLSchemeUnknown)
�urljoin)
�RequestMethods)
�connection_requires_http_tunnel)
�Retry)
�	parse_url)�PoolManager�ProxyManager�proxy_from_url)�key_file�	cert_file�	cert_reqs�ca_certs�ssl_version�ca_cert_dir�ssl_context�key_password)�
key_scheme�key_host�key_port�key_timeout�key_retries�
key_strict�	key_block�key_source_address�key_key_file�key_key_password�
key_cert_file�
key_cert_reqs�key_ca_certs�key_ssl_version�key_ca_cert_dir�key_ssl_context�key_maxsize�key_headers�
key__proxy�key__proxy_headers�key__proxy_config�key_socket_options�key__socks_options�key_assert_hostname�key_assert_fingerprint�key_server_hostname�PoolKey)r�use_forwarding_for_https�ProxyConfigcCs�|
��}|d
�
�|d
<|d�
�|d<dD]}||vr,||du
r,t||���
||<q|�d�
}|du
r<t|�
|d<t|���
D]}|�|�
|d|<qB|j	D]
}||v
r[d||<qQ|di|�
�
S)a�
    Create a pool key out of a request context dictionary.

    According to RFC 3986, both the scheme and host are case-insensitive.
    Therefore, this function normalizes both before constructing the pool
    key for an HTTPS request. If you wish to change this behaviour, provide
    alternate callables to ``key_fn_by_scheme``.

    :param key_class:
        The class to use when constructing the key. This should be a namedtuple
        with the ``scheme`` and ``host`` keys at a minimum.
    :type  key_class: namedtuple
    :param request_context:
        A dictionary-like object that contain the context for a request.
    :type  request_context: dict

    :return: A namedtuple that can be used as a connection pool key.
    :rtype:  PoolKey
    �scheme�host)�headers�_proxy_headers�_socks_optionsN�socket_options�key_�)
�copy�lower�	frozenset�items�get�tuple�list�keys�pop�_fields)�	key_class�request_context�context�key�socket_opts�fieldrBrB�5/usr/lib/python3/dist-packages/urllib3/poolmanager.py�_default_key_normalizerNs"



�






�rT��http�httpsc@s�eZ
dZd
ZdZdZd dd�
Zdd�Zdd	�Zd!d
d�
Z	dd
�Z
d"dd�
Zdd�Zd!dd�
Z
d!dd�
Zdd�Zdd�Zdd�Zd#dd�
ZdS)$ra$
    Allows for arbitrary requests while transparently keeping track of
    necessary connection pools for you.

    :param num_pools:
        Number of connection pools to cache before discarding the least
        recently used pool.

    :param headers:
        Headers to include with all requests, unless other headers are given
        explicitly.

    :param \**connection_pool_kw:
        Additional parameters are used to create fresh
        :class:`urllib3.connectionpool.ConnectionPool` instances.

    Example::

        >>> manager = PoolManager(num_pools=2)
        >>> r = manager.request('GET', 'http://google.com/')
        >>> r = manager.request('GET', 'http://google.com/mail')
        >>> r = manager.request('GET', 'http://yahoo.com/')
        >>> len(manager.pools)
        2

    N�
cKs~t�
||�
d
|vr)|d
}t|t�s)|du
}tj|dd�}||_|��}||d
<||_t|
dd�d�|_	t
|_
t��|_dS)N�retriesF�
�redirectc

Ss|��S�
N)
�close)
�
prBrBrS�<lambda>�sz&PoolManager.__init__.<locals>.<lambda>)
�dispose_func)r�__init__�
isinstancer�from_int�raise_on_redirectrC�connection_pool_kwr�pools�pool_classes_by_scheme�key_fn_by_scheme)�self�	num_poolsr=rerYrdrBrBrSra�s




	






zPoolManager.__init__c


Cs|Sr\rB�
rirBrBrS�	__enter__�s
zPoolManager.__enter__cCs|��
d
S)NF)
�clear)ri�exc_type�exc_val�exc_tbrBrBrS�__exit__�s
zPoolManager.__exit__cCsb|j|
}|d
ur|j
��}dD]}|�|d
�
q|
dkr(tD]}|�|d
�
q|||fi|�
�
S)a�

        Create a new :class:`urllib3.connectionpool.ConnectionPool` based on host, port, scheme, and
        any additional pool keyword arguments.

        If ``request_context`` is provided, it is provided as keyword arguments
        to the pool class used. This method is used to actually create the
        connection pools handed out by :meth:`connection_from_url` and
        companion methods. It is intended to be overridden for customization.
        N)r;r<�portrV)rgrerCrK�SSL_KEYWORDS)rir;r<rrrN�pool_clsrP�kwrBrBrS�	_new_pool�s







zPoolManager._new_poolc

Cs|j�
�
d
S)z�
        Empty our store of pools and direct them all to close.

        This will not affect in-flight connections, but they will not be
        re-used after completion.
        N)rfrmrkrBrBrSrm�szPoolManager.clearrVcCsT|
std
�
�
|�
|�
}|pd|d<|st�|d��d�}||d<|
|d<|�|�
S)a�

        Get a :class:`urllib3.connectionpool.ConnectionPool` based on the host, port, and scheme.

        If ``port`` isn't given, it will be derived from the ``scheme`` using
        ``urllib3.connectionpool.port_by_scheme``. If ``pool_kwargs`` is
        provided, it is merged with the instance's ``connection_pool_kw``
        variable and used to create the new connection pool, if one is
        needed.
        zNo host specified.rVr;�Prrr<)r	�_merge_pool_kwargsrrGrD�connection_from_context)rir<rrr;�pool_kwargsrNrBrBrS�connection_from_host�s







z PoolManager.connection_from_hostcCs:|
d
��}|j
�|�
}|st|�
�
||
�
}|j||
d�S)z�
        Get a :class:`urllib3.connectionpool.ConnectionPool` based on the request context.

        ``request_context`` must at least contain the ``scheme`` key and its
        value must be a key in ``key_fn_by_scheme`` instance variable.
        r;�
rN)rDrhrGr
�connection_from_pool_key)rirNr;�pool_key_constructor�pool_keyrBrBrSry
s



z#PoolManager.connection_from_contextcCs�|jj
�4
|j�|
�
}|r|Wd
�
S|d}|d}|d}|j||||d�}||j|
<Wd
�
|S1s;w



Y
|S)a

        Get a :class:`urllib3.connectionpool.ConnectionPool` based on the provided pool key.

        ``pool_key`` should be a namedtuple that only contains immutable
        objects. At a minimum it must have the ``scheme``, ``host``, and
        ``port`` fields.
        Nr;r<rrr|)rf�lockrGrv)rirrN�poolr;r<rrrBrBrSr}
s


�




��z$PoolManager.connection_from_pool_keycCs t|
�
}|j
|j|j|j|d
�S)a�

        Similar to :func:`urllib3.connectionpool.connection_from_url`.

        If ``pool_kwargs`` is not provided and a new pool needs to be
        constructed, ``self.connection_pool_kw`` is used to initialize
        the :class:`urllib3.connectionpool.ConnectionPool`. If ``pool_kwargs``
        is provided, it is used instead. Note that if a new pool does not
        need to be created for the request, the provided ``pool_kwargs`` are
        not used.
        )rrr;rz)rr{r<rrr;)ri�urlrz�
urBrBrS�connection_from_url.
s

�zPoolManager.connection_from_urlc	CsR|j�
�}|
r'|
��D]\}}|d
ur"z||=Wqty!


Yqw|||<q|S)a

        Merge a dictionary of override values for self.connection_pool_kw.

        This does not modify self.connection_pool_kw and returns a new dict.
        Any keys in the override dictionary with a value of ``None`` are
        removed from the merged dictionary.
        N)rerCrF�KeyError)ri�override�base_pool_kwargsrP�valuerBrBrSrx>
s








�

zPoolManager._merge_pool_kwargscCs"|jd
urdSt
|j|j|
j�S)z�
        Indicates if the proxy requires the complete destination URL in the
        request.  Normally this is only needed when not using an HTTP CONNECT
        tunnel.
        NF)�proxyr�proxy_configr;)ri�
parsed_urlrBrBrS�!_proxy_requires_url_absolute_formR
s



�z-PoolManager._proxy_requires_url_absolute_formcCsD|jd
us	|
dkrd
S|jj
dkrd
Stjr|jjs td�
�
d
Sd
S)z�
        Validates that were not attempting to do TLS in TLS connections on
        Python2 or with unsupported SSL implementations.
        NrWzfContacting HTTPS destinations through HTTPS proxies 'via CONNECT tunnels' is not supported in Python 2)r�r;�six�PY2r�r9r)ri�
url_schemerBrBrS�$_validate_proxy_scheme_url_selection_
s



��z0PoolManager._validate_proxy_scheme_url_selectionTcKs�
t|�
}|�
|j�

|j|j|j|jd
�}d|d<d|d<d|v
r(|j��|d<|�|�
r8|j	|
|fi|�
�
}n|j	|
|j
fi|�
�
}|oH|��}|sM|St||�}|j
dkrgd}
d|d	<t|d�
��|d<|�d
|j�}	t|	t�sztj|	|d�}	|	jr�|�|�
s�tt�|d�
�
}
|
D]}|��|	jvr�|d�|d�
q�z|	j|
|||d�}	Wnty�


|	jr�|��
�|YSw|	|d
<||d<t �!d
||�
|��
|j	|
|fi|�
�
S)aN

        Same as :meth:`urllib3.HTTPConnectionPool.urlopen`
        with custom cross-host redirect logic and only sends the request-uri
        portion of the ``url``.

        The given ``url`` parameter must be absolute, such that an appropriate
        :class:`urllib3.connectionpool.ConnectionPool` can be chosen for it.
        )rrr;F�assert_same_hostr[r=i/
�GETN�bodyrYrZ)�response�_poolzRedirecting %s -> %s)"rr�r;r{r<rrr=rCr��urlopen�request_uri�get_redirect_locationr�statusr�_prepare_for_method_changerGrYrbrrc�remove_headers_on_redirect�is_same_hostrIr��iterkeysrDrK�	incrementr
rd�
drain_conn�log�info)ri�methodr�r[rur��connr��redirect_locationrYr=�headerrBrBrSr�p
sT	














�


�





�

zPoolManager.urlopen)rXNr\�NrVN�
T)�__name__�
__module__�__qualname__�__doc__r�r�rarlrqrvrmr{ryr}r�rxr�r�r�rBrBrBrSr�s"




	


rcsReZ
dZd
Z					d�f
dd�	Zd�f
dd	�	Zdd
d�
Zd�f
d
d�	Z�ZS)ra>
    Behaves just like :class:`PoolManager`, but sends all requests through
    the defined proxy, using the CONNECT method for HTTPS URLs.

    :param proxy_url:
        The URL of the proxy to be used.

    :param proxy_headers:
        A dictionary containing headers that will be sent to the proxy. In case
        of HTTP they are being sent with each request, while in the
        HTTPS/CONNECT case they are sent only once. Could be used for proxy
        authentication.

    :param proxy_ssl_context:
        The proxy SSL context is used to establish the TLS connection to the
        proxy when using HTTPS proxies.

    :param use_forwarding_for_https:
        (Defaults to False) If set to True will forward requests to the HTTPS
        proxy to be made on behalf of the client instead of creating a TLS
        tunnel via the CONNECT method. **Enabling this flag means that request
        and response headers and content will be visible from the HTTPS proxy**
        whereas tunneling keeps request and response headers and content
        private.  IP address, target hostname, SNI, and port are always visible
        to an HTTPS proxy even when this flag is disabled.

    Example:
        >>> proxy = urllib3.ProxyManager('http://localhost:3128/')
        >>> r1 = proxy.request('GET', 'http://google.com/')
        >>> r2 = proxy.request('GET', 'http://httpbin.org/')
        >>> len(proxy.pools)
        1
        >>> r3 = proxy.request('GET', 'https://httpbin.org/')
        >>> r4 = proxy.request('GET', 'https://twitter.com/')
        >>> len(proxy.pools)
        3

    rXNFc
s�t|
t
�rd
|
j|
j|
jf}
t|
�
}|jdv
rt|j�
�
|js-t�|jd�}	|j	|	d�
}||_
|p3i|_||_t
||�|_|j
|d<|j|d<|j|d<tt|�j||fi|�
�

dS)Nz
%s://%s:%irUrw)
rr�_proxyr>�
_proxy_config)rbrr;r<rrrrrrG�_replacer��
proxy_headers�proxy_ssl_contextr:r��superrra)
ri�	proxy_urlrjr=r�r�r9rer�rr�
�	__class__rBrSra�
s(




�













zProxyManager.__init__rVcsD|d
krtt
|�j|
|||d�Stt
|�j|jj|jj|jj|d�S)NrW)
rz)r�rr{r�r<rrr;)rir<rrr;rzr�rBrSr{s



�

�z!ProxyManager.connection_from_hostcCs0d
di
}t|
�
j
}|r||d<|r|�|�

|S)z�
        Sets headers needed by proxies: specifically, the Accept and Host
        headers. Only sets headers not provided by the user.
        �Acceptz*/*�Host)r�netloc�update)rir�r=�headers_r�rBrBrS�_set_proxy_headerss





zProxyManager._set_proxy_headersTcsVt|�
}t
|j|j|j�s|�d
|j�}|�||�|d
<tt	|�j
|
|fd|i
|�
�
S)z@Same as HTTP(S)ConnectionPool.urlopen, ``url`` must be absolute.r=r[)rrr�r�r;rGr=r�r�rr�)rir�r�r[rur�r=r�rBrSr�s


zProxyManager.urlopen)rXNNNFr�r\r�)	r�r�r�r�rar{r�r��
__classcell__rBrBr�rSr�
s
*



�%

rc
Kstdd
|i
|
�
�
S)Nr�rB)
r)r�rurBrBrSr*s
r).�
__future__r�collections�	functools�logging�_collectionsrr�connectionpoolrrr�
exceptionsr	r
rrr
r��six.moves.urllib.parser�requestr�
util.proxyr�
util.retryr�util.urlr�__all__�	getLoggerr�r�rs�_key_fields�
namedtupler8�_proxy_config_fieldsr:rT�partialrhrgrrrrBrBrBrS�<module>
s<












6

�
.s