a
    db?                     @   s  d dl mZ d dlZd dlZd dlZddl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 dd	lmZ dd
lmZ ddlmZ g dZeeZdZdZedeZdd Ze eee eedZ!ee	dZ"G dd deZ#G dd de#Z$dd Z%dS )    )absolute_importN   )RecentlyUsedContainer)HTTPConnectionPoolHTTPSConnectionPool)port_by_scheme)LocationValueErrorMaxRetryErrorProxySchemeUnknown)urljoin)RequestMethods)	parse_url)Retry)PoolManagerProxyManagerproxy_from_url)key_file	cert_file	cert_reqsca_certsssl_versionca_cert_dirssl_context)
key_schemekey_hostkey_portkey_timeoutkey_retries
key_strict	key_blockkey_source_addresskey_key_file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_socket_optionskey__socks_optionskey_assert_hostnamekey_assert_fingerprintPoolKeyc                 C   s   |  }|d  |d< |d  |d< dD ],}||v r,|| dur,t||  ||< q,|d}|durxt||d< t| D ]}|||d| < q| j	D ]}||vrd||< q| f i |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field rH   d/home/tom/ab/renpy-build/tmp/install.linux-x86_64/lib/python3.9/site-packages/urllib3/poolmanager.py_default_key_normalizer9   s    


rJ   httphttpsc                   @   sx   e Zd ZdZdZdddZdd Zdd	 Zdd
dZdd Z	dddZ
dd ZdddZd ddZdd Zd!ddZdS )"r   a$  
    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
   c                 K   s8   t | | || _t|dd d| _t| _t | _d S )Nc                 S   s   |   S N)close)prH   rH   rI   <lambda>       z&PoolManager.__init__.<locals>.<lambda>)dispose_func)r   __init__connection_pool_kwr   poolspool_classes_by_schemekey_fn_by_schemer8   )self	num_poolsr3   rV   rH   rH   rI   rU      s    zPoolManager.__init__c                 C   s   | S rO   rH   rZ   rH   rH   rI   	__enter__   s    zPoolManager.__enter__c                 C   s   |    dS )NF)clear)rZ   exc_typeexc_valexc_tbrH   rH   rI   __exit__   s    zPoolManager.__exit__c                 C   sb   | j | }|du r| j }dD ]}||d q |dkrPtD ]}||d q>|||fi |S )a  
        Create a new :class:`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)r1   r2   portrL   )rX   rV   r8   r@   SSL_KEYWORDS)rZ   r1   r2   rc   rC   pool_clsrE   kwrH   rH   rI   	_new_pool   s    


zPoolManager._new_poolc                 C   s   | 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)rW   r^   r\   rH   rH   rI   r^      s    zPoolManager.clearrL   c                 C   sT   |st d| |}|pd|d< |s:t|d  d}||d< ||d< | |S )a  
        Get a :class:`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.
        zNo host specified.rL   r1   P   rc   r2   )r   _merge_pool_kwargsr   r<   r9   connection_from_context)rZ   r2   rc   r1   pool_kwargsrC   rH   rH   rI   connection_from_host   s    
z PoolManager.connection_from_hostc                 C   s,   |d   }| j| }||}| j||dS )z
        Get a :class:`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.
        r1   rC   )r9   rY   connection_from_pool_key)rZ   rC   r1   pool_key_constructorpool_keyrH   rH   rI   rj      s    
z#PoolManager.connection_from_contextc                 C   s   | j jf | j |}|r,|W  d   S |d }|d }|d }| j||||d}|| j |< W d   n1 st0    Y  |S )z
        Get a :class:`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.
        Nr1   r2   rc   rm   )rW   lockr<   rg   )rZ   rp   rC   poolr1   r2   rc   rH   rH   rI   rn      s    
(z$PoolManager.connection_from_pool_keyc                 C   s    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.
        )rc   r1   rk   )r   rl   r2   rc   r1   )rZ   urlrk   urH   rH   rI   connection_from_url
  s    zPoolManager.connection_from_urlc              	   C   sT   | j  }|rP| D ]8\}}|du rFz
||= W qN tyB   Y qN0 q|||< 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)rV   r8   r;   KeyError)rZ   overridebase_pool_kwargsrE   valuerH   rH   rI   ri     s    


zPoolManager._merge_pool_kwargsTc           
      K   s8  t |}| j|j|j|jd}d|d< d|d< d|vr@| j|d< | jdurj|jdkrj|j||fi |}n|j||jfi |}|o|	 }|s|S t
||}|jdkrd	}|d
}	t|	tstj|	|d}	z|	j||||d}	W n  ty   |	jr | Y S 0 |	|d
< ||d< td|| | j||fi |S )a]  
        Same as :meth:`urllib3.connectionpool.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.
        )rc   r1   Fassert_same_hostredirectr3   NrL   i/  GETretries)r{   )response_poolzRedirecting %s -> %s)r   rl   r2   rc   r1   r3   proxyurlopenrequest_uriget_redirect_locationr   statusr<   
isinstancer   from_int	incrementr	   raise_on_redirectloginfo)
rZ   methodrs   r{   rf   rt   connr~   redirect_locationr}   rH   rH   rI   r   -  s8    	





zPoolManager.urlopen)rN   N)N)NrL   N)N)N)T)__name__
__module____qualname____doc__r   rU   r]   rb   rg   r^   rl   rj   rn   ru   ri   r   rH   rH   rH   rI   r   y   s   

	


r   c                       sH   e Zd ZdZd fdd	Zd fdd	Zdd	d
Zd fdd	Z  ZS )r   aw  
    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 contaning 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.

    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

    rN   Nc                    s   t |trd|j|j|jf }t|}|jsFt|jd}|j|d}|jdvrZt	|j|| _
|pfi | _| j
|d< | j|d< tt| j||fi | d S )Nz
%s://%s:%irh   )rc   rK   _proxyr4   )r   r   r1   r2   rc   r   r   r<   _replacer
   r   proxy_headerssuperr   rU   )rZ   	proxy_urlr[   r3   r   rV   r   rc   	__class__rH   rI   rU   {  s&    







zProxyManager.__init__rL   c                    sD   |dkr t t| j||||dS t t| j| jj| jj| jj|dS )NrM   )rk   )r   r   rl   r   r2   rc   r1   )rZ   r2   rc   r1   rk   r   rH   rI   rl     s    

z!ProxyManager.connection_from_hostc                 C   s0   ddi}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)rZ   rs   r3   headers_r   rH   rH   rI   _set_proxy_headers  s    

zProxyManager._set_proxy_headersTc                    sN   t |}|jdkr0|d| j}| |||d< tt| j||fd|i|S )z@Same as HTTP(S)ConnectionPool.urlopen, ``url`` must be absolute.rL   r3   r{   )r   r1   r<   r3   r   r   r   r   )rZ   r   rs   r{   rf   rt   r3   r   rH   rI   r     s
    
zProxyManager.urlopen)rN   NN)NrL   N)N)T)	r   r   r   r   rU   rl   r   r   __classcell__rH   rH   r   rI   r   `  s     
r   c                 K   s   t f d| i|S )Nr   )r   )rs   rf   rH   rH   rI   r     s    r   )&
__future__r   collections	functoolslogging_collectionsr   connectionpoolr   r   r   
exceptionsr   r	   r
   Zpackages.six.moves.urllib.parser   requestr   Zutil.urlr   Z
util.retryr   __all__	getLoggerr   r   rd   _key_fields
namedtupler0   rJ   partialrY   rX   r   r   r   rH   rH   rH   rI   <module>   s6   
6

 hW