)X$3@sddlmZddlZddlZddlZyddlmZWn"ek rlddlmZYnXddl m Z ddl m Z m Z ddl mZddlmZmZmZdd lmZdd lmZdd lmZd d dgZejeZd'Zejdd(Zejdej d)Z!ejde!j eZ"d d!Z#iej$e#e!d"6ej$e#e"d#6Z%ie d"6e d#6Z&Gd$d d eZ'Gd%d d e'Z(d&dZ)dS)*)absolute_importN)urljoin)RecentlyUsedContainer)HTTPConnectionPoolHTTPSConnectionPool)port_by_scheme)LocationValueError MaxRetryErrorProxySchemeUnknown)RequestMethods) parse_url)Retry PoolManager ProxyManagerproxy_from_urlkey_file cert_file cert_reqsca_certs ssl_version ca_cert_dir BasePoolKeyschemehostport HTTPPoolKeytimeoutretriesstrictblocksource_address HTTPSPoolKeycCs_i}x$|jD]}|j|||>> 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 cKsPtj||||_t|ddd|_t|_tj|_dS)N dispose_funccSs |jS)N)close)pr)r)r*ysz&PoolManager.__init__..)r __init__connection_pool_kwrpoolspool_classes_by_schemekey_fn_by_schemecopy)self num_poolsheadersr4r)r)r*r3us    zPoolManager.__init__cCs|S)Nr))r9r)r)r* __enter__szPoolManager.__enter__cCs|jdS)NF)clear)r9exc_typeZexc_valexc_tbr)r)r*__exit__s zPoolManager.__exit__cCse|j|}|j}|dkrU|jj}x!tD]}|j|dq8Wn||||S)a Create a new :class:`ConnectionPool` based on host, port and scheme. 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. r,N)r6r4r8 SSL_KEYWORDSpop)r9rrrpool_clskwargskwr)r)r* _new_pools    zPoolManager._new_poolcCs|jjdS)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)r5r=)r9r)r)r*r=szPoolManager.clearr,cCsz|stdn|jj}|p-d|d<|sYtj|djd}n||d<||d<|j|S)z 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``. zNo host specified.r,rPrr)r r4r8rr$r%connection_from_context)r9rrrr&r)r)r*connection_from_hosts  z PoolManager.connection_from_hostcCs6|dj}|j|}||}|j|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. r)r%r7connection_from_pool_key)r9r&rZpool_key_constructorpool_keyr)r)r*rHs  z#PoolManager.connection_from_contextc Cs^|jjL|jj|}|r)|S|j|j|j|j}||j| %s)r rIrrrr;proxyurlopen request_uriget_redirect_locationrstatusr$ isinstancerZfrom_int incrementr Zraise_on_redirectloginfo) r9methodrNrRrErOconnrTredirect_locationrr)r)r*rWs8 $    "     zPoolManager.urlopen)__name__ __module__ __qualname____doc__rVr3r<r@rFr=rIrHrJrPrWr)r)r)r*rWs       csmeZdZdZdddfddZddfddZdd d Zd fd d ZS)raw 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 r.Nc st|tr.d|j|j|jf}nt|}|jsmtj|jd}|jd|}n|jdkrt |jn||_ |pi|_ |j |d<|j |dsD        W