ReE @ s d d l m Z d d l 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 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 d
d d g Z e j e Z d9 Z d: Z e j! d1 e Z" d2 d3 Z# d4 e j$ e# e" d5 e j$ e# e" i Z% d4 e d5 e
i Z& Gd6 d
d
e Z' Gd7 d d e' Z( d8 d Z) d S); )absolute_importN )RecentlyUsedContainer)HTTPConnectionPoolHTTPSConnectionPool)port_by_scheme)LocationValueError
MaxRetryErrorProxySchemeUnknown InvalidProxyConfigurationWarning)six)urljoin)RequestMethods) parse_url)Retry PoolManager ProxyManagerproxy_from_urlkey_file cert_file cert_reqsca_certs ssl_version ca_cert_dir ssl_context key_password
key_schemekey_hostkey_port key_timeout key_retries
key_strict key_blockkey_source_address key_key_filekey_key_password
key_cert_file
key_cert_reqs key_ca_certskey_ssl_versionkey_ca_cert_dirkey_ssl_context key_maxsize key_headers
key__proxykey__proxy_headerskey_socket_optionskey__socks_optionskey_assert_hostnamekey_assert_fingerprintkey_server_hostnamePoolKeyc C s | j } | d j | d <| d j | d >> 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 sP t j | | | | _ t | d d d | _ t | _ t j | _ d S)N dispose_funcc S s
| j S)N)close)prM rM rN s z&PoolManager.__init__..)r __init__connection_pool_kwr poolspool_classes_by_schemekey_fn_by_schemer= )self num_poolsr8 rX rM rM rN rW s
zPoolManager.__init__c C s | S)NrM )r\ rM rM rN __enter__ s zPoolManager.__enter__c C s | j d S)NF)clear)r\ exc_typeexc_valexc_tbrM rM rN __exit__ s
zPoolManager.__exit__c C s | j | } | d k r( | j j } x d D] } | j | d q/ W| d k rv x t D] } | j | d q\ W| | | | 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.
Nr6 r7 portrP )r6 r7 rd )rZ rX r= rE SSL_KEYWORDS)r\ r6 r7 rd rH pool_clsrJ kwrM rM rN _new_pool s
zPoolManager._new_poolc C s | j 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)rY r_ )r\ rM rM rN r_ s zPoolManager.clearrP c C st | s t d | j | } | p* d | d <| sS t j | d j d } | | d <| | d <| j | 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 rd r7 )r _merge_pool_kwargsr rA r> connection_from_context)r\ r7 rd r6 pool_kwargsrH rM rM rN connection_from_host s
z PoolManager.connection_from_hostc C s< | d j } | 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_constructorpool_keyrM rM rN rk s
z#PoolManager.connection_from_contextc
C sz | j j g | j j | } | r) | S| d } | d } | d } | j | | | d | } | | j | %s) r rm r7 rd r6 r8 r= proxyurlopen request_uriget_redirect_locationr
statusrA
isinstancer from_intremove_headers_on_redirect is_same_hostrC r iterkeysr> rE incrementr raise_on_redirect
drain_connloginfo) r\ methodrs r{ rg rt connr~ redirect_locationr} r8 headerrM rM rN r ; sH $
"
zPoolManager.urlopen)__name__
__module__ __qualname____doc__r rW r^ rc rh r_ rm rk rn ru rj r rM rM rM rN r s
c s| e Z d Z d Z d d d f d d Z d d d f d d Z d d d
Z d d Z d
f d d 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 | j f } t | } | j sg t j | j d } | j d | } | j d k r t | j | | _
| p i | _ | j
| d <| j | d | j | | S)z
Sets headers needed by proxies: specifically, the Accept and Host
headers. Only sets headers not provided by the user.
Acceptz*/*Host)r netlocupdate)r\ rs r8 headers_r rM rM rN _set_proxy_headers s
z ProxyManager._set_proxy_headersc C s8 | d k r4 | j j d k r4 t j d t d d 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 warningswarnr )r\
url_schemerM rM rN $_validate_proxy_scheme_url_selection s
z1ProxyManager._validate_proxy_scheme_url_selectionTc sx t | } | j | j | j d k rV | j d | j } | j | | | d sp " 6n