Current File : //snap/core/current/usr/lib/python3/dist-packages/urllib3/__pycache__/poolmanager.cpython-35.pyc
� \h�+ � @ s[ d d l m Z d d l Z y d d l m Z Wn" e k
rT d d l m Z Yn Xd d l m Z 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 Z d d l m Z d d
l m Z d d l m Z d d
d g Z d e d e i Z e j e � Z d Z Gd d � d e � Z Gd d
� d
e � Z d d � Z! d S)� )�absolute_importN)�urljoin� )�HTTPHeaderDict�RecentlyUsedContainer)�HTTPConnectionPool�HTTPSConnectionPool)�port_by_scheme)�LocationValueError�
MaxRetryError�ProxySchemeUnknown)�RequestMethods)� parse_url)�Retry� PoolManager� ProxyManager�proxy_from_url�http�https�key_file� cert_file� cert_reqs�ca_certs� ssl_version� ca_cert_dirc @ s� e Z d Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d d
� Z d 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 s� t j | | � d | k ru | d } t | t � su | d k } t j | d d �} | | _ | j � } | | d <| | _ t | d d d � �| _ d S)N�retriesF�redirectZ dispose_funcc S s
| j � S)N)�close)�p� r �5/usr/lib/python3/dist-packages/urllib3/poolmanager.py�<lambda>S s z&PoolManager.__init__.<locals>.<lambda>)
r
�__init__�
isinstancer �from_int�raise_on_redirect�copy�connection_pool_kwr �pools)�self� num_pools�headersr( r r& r r r! r# ? s
zPoolManager.__init__c C s | S)Nr )r* r r r! � __enter__U s zPoolManager.__enter__c C s | j � d S)NF)�clear)r* �exc_typeZexc_valZexc_tbr r r! �__exit__X s
zPoolManager.__exit__c C s_ t | } | j } | d k rO | j j � } x t D] } | j | d � q5 W| | | | � 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)�pool_classes_by_schemer( r' � SSL_KEYWORDS�pop)r* �scheme�host�portZpool_cls�kwargs�kwr r r! � _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)r) r. )r* r r r! r. n s zPoolManager.clearr c C s� | s t d � � | p d } | p3 t j | d � } | | | f } | j j �C | j j | � } | rn | S| j | | | � } | | j | <Wd QRX| 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 �P N)r
r �getr) �lockr9 )r* r5 r6 r4 Zpool_keyZpoolr r r! �connection_from_hostw s
z PoolManager.connection_from_hostc C s. t | � } | j | j d | j d | j �S)a*
Similar to :func:`urllib3.connectionpool.connection_from_url` but
doesn't pass any additional parameters to the
:class:`urllib3.connectionpool.ConnectionPool` constructor.
Additional parameters are taken from the :class:`.PoolManager`
constructor.
r6 r4 )r r= r5 r6 r4 )r* �url�ur r r! �connection_from_url� s z PoolManager.connection_from_urlTc
K s? t | � } | j | j d | j d | j �} d | d <d | d <d | k rc | j j � | d <| j d k r� | j d k r� | j | | | � } n | j | | j | � } | o� | j
� } | s� | St | | � } | j d k rd
} d | d <t
| d � j � | d <| j d | j � } t | t � sNt j | d | �} | j r�| j | � r�t t j | d � � }
x7 |
D]/ } | j � | j k r�| d j | d � q�Wy" | j | | d
| d | �} Wn" t k
r | j r�� | SYn X| | d <| | d <t j d | | f � | j | | | � 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.
r6 r4 FZassert_same_hostr r, Nr i/ ZGETZbodyr �responseZ_poolzRedirecting %s -> %s) r r= r5 r6 r4 r, r' �proxy�urlopenZ request_uriZget_redirect_locationr Zstatusr Z_prepare_for_method_changer; r r$ r r% Zremove_headers_on_redirectZ is_same_host�list�sixZiterkeys�lowerr3 Z incrementr r& �log�info) r* �methodr> r r8 r? ZconnrA Zredirect_locationr r, �headerr r r! rC � sH $
"
zPoolManager.urlopen)
�__name__�
__module__� __qualname__�__doc__rB r# r- r0 r9 r. r= r@ rC r r r r! r ! s c sm e Z d Z d Z d d d � f d d � Z d d � f d d � Z d d d
� Z d � f d d
� 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
r 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 <t t
| � j | | | � d S) Nz
%s://%s:%ir: r6 r r �_proxyZ_proxy_headers)zhttpr )r$ r r4 r5 r6 r r r; �_replacer rB �
proxy_headers�superr r# )r* � proxy_urlr+ r, rQ r( rB r6 )� __class__r r! r# � s
zProxyManager.__init__r c sV | d k r( t t | � j | | | � St t | � j | j j | j j | j j � S)Nr )rR r r= rB r5 r6 r4 )r* r5 r6 r4 )rT r r! r= s
z!ProxyManager.connection_from_hostc C sB d d i } t | � j } | r+ | | d <| r> | j | � | S)z�
Sets headers needed by proxies: specifically, the Accept and Host
headers. Only sets headers not provided by the user.
ZAcceptz*/*ZHost)r �netloc�update)r* r> r, Zheaders_rU r r r! �_set_proxy_headers s
z ProxyManager._set_proxy_headersTc sh t | � } | j d k rF | j d | j � } | j | | � | d <t t | � j | | d | | �S)z@Same as HTTP(S)ConnectionPool.urlopen, ``url`` must be absolute.r r, r )r r4 r; r, rW rR r rC )r* rI r> r r8 r? r, )rT r r! rC ) s
zProxyManager.urlopen)rK rL rM rN r# r= rW rC r r )rT r! r � s c K s t d | | � S)NrS )r )r> r8 r r r! r 7 s )r r r r r r )"Z
__future__r ZloggingZ urllib.parser � ImportErrorZurlparse� _collectionsr r Zconnectionpoolr r r �
exceptionsr
r r rE Zrequestr
Zutil.urlr Z
util.retryr �__all__r1 Z getLoggerrK rG r2 r r r r r r r! �<module> s,
�W
Mini Shell