관리-도구
편집 파일: poolmanager.cpython-37.pyc
B ��Re�E � @ s( d dl mZ d dlZd dlZd dlZd dlZddlmZ ddlm Z m Z ddlm Z ddl m Z mZmZmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ d ddgZe� e �Z dZ dZ e�!de �Z"dd� Z#e�$e#e"�e�$e#e"�d�Z%e e d�Z&G dd � d e�Z'G dd� de'�Z(dd� Z)dS )� )�absolute_importN� )�RecentlyUsedContainer)�HTTPConnectionPool�HTTPSConnectionPool)�port_by_scheme)�LocationValueError� MaxRetryError�ProxySchemeUnknown� InvalidProxyConfigurationWarning)�six)�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_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_socket_options�key__socks_options�key_assert_hostname�key_assert_fingerprint�key_server_hostname�PoolKeyc C s� |� � }|d �� |d< |d �� |d< x4dD ],}||kr.|| dk r.t|| �� �||< q.W |�d�}|dk r|t|�|d< x&t|�� �D ]}|�|�|d| < q�W x | j D ]}||kr�d||< q�W | f |�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� rM ��/builddir/build/BUILDROOT/alt-python37-pip-20.2.4-6.el8.x86_64/opt/alt/python37/lib/python3.7/site-packages/pip/_vendor/urllib3/poolmanager.py�_default_key_normalizerJ s rO )�http�httpsc @ sx e Zd ZdZdZd dd�Zdd� Zdd � Zd d d �Zd d � Z d dd�Z dd� Z d dd�Z d dd�Z dd� Zd!dd�ZdS )"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|dd� d�| _t| _t�� | _d S )Nc S s | � � S )N)�close)�prM rM rN �<lambda>� � z&PoolManager.__init__.<locals>.<lambda>)� dispose_func)r �__init__�connection_pool_kwr �pools�pool_classes_by_scheme�key_fn_by_schemer= )�self� num_poolsr8 rY rM rM rN rX � s zPoolManager.__init__c C s | S )NrM )r] rM rM rN � __enter__� s zPoolManager.__enter__c C s | � � dS )NF)�clear)r] �exc_type�exc_val�exc_tbrM rM rN �__exit__� s zPoolManager.__exit__c C sf | j | }|dkr | j�� }xdD ]}|�|d� q"W |dkrXxtD ]}|�|d� qDW |||f|�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)r6 r7 �portrP )r[ rY r= rE � SSL_KEYWORDS)r] r6 r7 re rH �pool_clsrJ �kwrM rM rN � _new_pool� s zPoolManager._new_poolc C s | j �� dS )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)rZ r` )r] rM rM rN r` � s zPoolManager.clearrP 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. zNo host specified.rP r6 �P re r7 )r �_merge_pool_kwargsr rA r> �connection_from_context)r] r7 re r6 � pool_kwargsrH rM rM rN �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. r6 )rH )r> r\ �connection_from_pool_key)r] rH r6 �pool_key_constructor�pool_keyrM rM rN rl � s z#PoolManager.connection_from_contextc C s` | j j�N | j �|�}|r |S |d }|d }|d }| j||||d�}|| j |< W dQ R X |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. r6 r7 re )rH N)rZ �lockrA ri )r] rq rH �poolr6 r7 re rM rM rN ro � 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. )re r6 rm )r rn r7 re r6 )r] �urlrm �urM rM rN �connection_from_url s z PoolManager.connection_from_urlc C sZ | j �� }|rVxF|�� D ]:\}}|dkrJy ||= W qR tk rF Y qRX q|||< qW |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)rY r= r@ �KeyError)r] �override�base_pool_kwargsrJ �valuerM rM rN rk ' s z PoolManager._merge_pool_kwargsTc K s� t |�}| j|j|j|jd�}d|d< d|d< d|krD| j�� |d< | jdk rj|jdkrj|j||f|�}n|j||j f|�}|o�|� � }|s�|S t ||�}|j dkr�d }|� d �} t| t�s�tj| |d �} | j�r |�|��s tt�|d ��} x*| D ]"} | �� | jkr�|d �| d� q�W y| j||||d �} W n( tk �rZ | j�rV|�� � |S X | |d < ||d< t � d ||� |�� | j||f|�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. )re r6 F�assert_same_host�redirectr8 NrP i/ �GET�retries)r| )�response�_poolzRedirecting %s -> %s) r rn r7 re r6 r8 r= �proxy�urlopen� request_uri�get_redirect_locationr �statusrA � isinstancer �from_int�remove_headers_on_redirect� is_same_hostrC r �iterkeysr> rE � incrementr �raise_on_redirect� drain_conn�log�info) r] �methodrt r| rh ru �connr �redirect_locationr~ r8 �headerrM rM rN r� ; sH zPoolManager.urlopen)rR N)N)NrP N)N)N)T)�__name__� __module__� __qualname__�__doc__r� rX r_ rd ri r` rn rl ro rv rk r� rM rM rM rN r � s c sP e Zd ZdZd� fdd� Zd� fdd� Zdd d �Zd d � Zd� fdd� Z� Z S )r ax 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. 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 rR Nc s� t |t�r d|j|j|jf }t|�}|jsFt�|jd�}|j|d�}|jdkrZt |j��|| _ |pfi | _ | j |d< | j |d< t t | �j||f|� d S )Nz %s://%s:%irj )re )rP rQ �_proxyr9 )r� r r6 r7 re r r rA �_replacer r� � proxy_headers�superr rX )r] � proxy_urlr^ r8 r� rY r� re )� __class__rM rN rX � s zProxyManager.__init__rP c sD |dkr t t| �j||||d�S t t| �j| jj| jj| jj|d�S )NrQ )rm )r� r rn r� r7 re r6 )r] r7 re r6 rm )r� rM rN rn � s z!ProxyManager.connection_from_hostc C s0 ddi}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)r] rt r8 �headers_r� rM rM rN �_set_proxy_headers� s z ProxyManager._set_proxy_headersc C s( |dkr$| j jdkr$tjdtdd� d S )NrQ a Your proxy configuration specified an HTTPS scheme for the proxy. Are you sure you want to use HTTPS to contact the proxy? This most likely indicates an error in your configuration. Read this issue for more info: https://github.com/urllib3/urllib3/issues/1850� )� stacklevel)r� r6 �warnings�warnr )r] � url_schemerM rM rN �$_validate_proxy_scheme_url_selection� s z1ProxyManager._validate_proxy_scheme_url_selectionTc sZ t |�}| �|j� |jdkr<|�d| j�}| �||�|d< tt| �j||fd|i|��S )z@Same as HTTP(S)ConnectionPool.urlopen, ``url`` must be absolute.rP r8 r| ) r r� r6 rA r8 r� r� r r� )r] r� rt r| rh ru r8 )r� rM rN r� � s zProxyManager.urlopen)rR NN)NrP N)N)T) r� r� r� r� rX rn r� r� r� � __classcell__rM rM )r� rN r } s c K s t f d| i|��S )Nr� )r )rt rh rM rM rN r � s )*� __future__r � collections� functools�loggingr� � _collectionsr �connectionpoolr r r � exceptionsr r r r �packagesr Z packages.six.moves.urllib.parser �requestr Zutil.urlr Z util.retryr �__all__� getLoggerr� r� rf � _key_fields� namedtupler5 rO �partialr\ r[ r r r rM rM rM rN �<module> s4 6 wn