Current File : //snap/core/17212/usr/lib/python3/dist-packages/urllib3/__pycache__/poolmanager.cpython-35.pyc
Rg<( � @ 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 s8 t j | | � | | _ t | d d d � �| _ d S)NZ dispose_funcc S s
| j � S)N)�close)�p� r �5/usr/lib/python3/dist-packages/urllib3/poolmanager.py�<lambda>C s z&PoolManager.__init__.<locals>.<lambda>)r
�__init__�connection_pool_kwr �pools)�self� num_pools�headersr" r r r r! ? s zPoolManager.__init__c C s | S)Nr )r$ r r r � __enter__E s zPoolManager.__enter__c C s | j � d S)NF)�clear)r$ �exc_typeZexc_valZexc_tbr r r �__exit__H 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" �copy� SSL_KEYWORDS�pop)r$ �scheme�host�portZpool_cls�kwargs�kwr r r � _new_poolM 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( ^ 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# �lockr4 )r$ r0 r1 r/ Zpool_keyZpoolr r r �connection_from_hostg 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.
r1 r/ )r r8 r0 r1 r/ )r$ �url�ur r r �connection_from_url� s z PoolManager.connection_from_urlTc
K s9 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 � } t | t � sHt 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.
r1 r/ FZassert_same_host�redirectr& Nr i/ ZGETZbody�retries�responseZ_poolzRedirecting %s -> %s) r r8 r0 r1 r/ r& r, �proxy�urlopenZ request_uriZget_redirect_locationr Zstatusr Z_prepare_for_method_changer6 �
isinstancer Zfrom_intZremove_headers_on_redirectZ is_same_host�list�sixZiterkeys�lowerr. Z incrementr Zraise_on_redirect�log�info) r$ �methodr9 r< r3 r: Zconnr> Zredirect_locationr= r& �headerr r r r@ � sH $
"
zPoolManager.urlopen)
�__name__�
__module__� __qualname__�__doc__r? r! r' r* r4 r( r8 r; r@ 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:%ir5 r1 r r �_proxyZ_proxy_headers)zhttpr )rA r r/ r0 r1 r r r6 �_replacer r? �
proxy_headers�superr r! )r$ � proxy_urlr% r& rO r" r? r1 )� __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 )rP r r8 r? r0 r1 r/ )r$ r0 r1 r/ )rR r r r8 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$ r9 r& Zheaders_rS 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 r/ r6 r& rU rP r r@ )r$ rG r9 r< r3 r: r& )rR r r r@ s
zProxyManager.urlopen)rI rJ rK rL r! r8 rU r@ r r )rR r r � s c K s t d | | � S)NrQ )r )r9 r3 r r r r ' s )r r r r r r )"Z
__future__r ZloggingZ urllib.parser � ImportErrorZurlparse� _collectionsr r Zconnectionpoolr r r �
exceptionsr
r r rC Zrequestr
Zutil.urlr Z
util.retryr �__all__r+ Z getLoggerrI rE r- r r r r r r r �<module> s,
�W
Mini Shell