3
P\8 @ sH d Z ddlmZmZ ddlmZmZ G dd deZG dd deZdS ) z
requests_toolbelt.auth.handler
==============================
This holds all of the implementation details of the Authentication Handler.
)AuthBase
HTTPBasicAuth)urlparse
urlunparsec @ sT e Zd ZdZdd Zdd Zdd Zdd Zed
d Z d d
Z
dd Z dd Z dS ) AuthHandlera
The ``AuthHandler`` object takes a dictionary of domains paired with
authentication strategies and will use this to determine which credentials
to use when making a request. For example, you could do the following:
.. code-block:: python
from requests import HTTPDigestAuth
from requests_toolbelt.auth.handler import AuthHandler
import requests
auth = AuthHandler({
'https://api.github.com': ('sigmavirus24', 'fakepassword'),
'https://example.com': HTTPDigestAuth('username', 'password')
})
r = requests.get('https://api.github.com/user', auth=auth)
# =>
r = requests.get('https://example.com/some/path', auth=auth)
# =>
s = requests.Session()
s.auth = auth
r = s.get('https://api.github.com/user')
# =>
.. warning::
:class:`requests.auth.HTTPDigestAuth` is not yet thread-safe. If you
use :class:`AuthHandler` across multiple threads you should
instantiate a new AuthHandler for each thread with a new
HTTPDigestAuth instance for each thread.
c C s t || _| j d S )N)dict
strategies
_make_uniform)selfr r /usr/lib/python3.6/handler.py__init__6 s
zAuthHandler.__init__c C s | j |j}||S )N)get_strategy_forurl)r
ZrequestZauthr r r __call__: s zAuthHandler.__call__c C s dj | jS )Nz)formatr )r
r r r __repr__> s zAuthHandler.__repr__c C s6 t | jj }i | _x |D ]\}}| j|| qW d S )N)listr items add_strategy)r
Zexisting_strategieskvr r r r A s zAuthHandler._make_uniformc C s( t | }t|jj |jj ddddfS )N )r r schemelowerZnetloc)r Zparsedr r r
_key_from_urlH s
zAuthHandler._key_from_urlc C s* t |trt| }| j|}|| j|< dS )a Add a new domain and authentication strategy.
:param str domain: The domain you wish to match against. For example:
``'https://api.github.com'``
:param str strategy: The authentication strategy you wish to use for
that domain. For example: ``('username', 'password')`` or
``requests.HTTPDigestAuth('username', 'password')``
.. code-block:: python
a = AuthHandler({})
a.add_strategy('https://api.github.com', ('username', 'password'))
N)
isinstancetupler r r )r
domainZstrategykeyr r r r O s
zAuthHandler.add_strategyc C s | j |}| jj|t S )a Retrieve the authentication strategy for a specified URL.
:param str url: The full URL you will be making a request against. For
example, ``'https://api.github.com/user'``
:returns: Callable that adds authentication to a request.
.. code-block:: python
import requests
a = AuthHandler({'example.com', ('foo', 'bar')})
strategy = a.get_strategy_for('http://example.com/example')
assert isinstance(strategy, requests.auth.HTTPBasicAuth)
)r r getNullAuthStrategy)r
r r r r r r e s
z AuthHandler.get_strategy_forc C s | j |}|| jkr | j|= dS )ak Remove the domain and strategy from the collection of strategies.
:param str domain: The domain you wish remove. For example,
``'https://api.github.com'``.
.. code-block:: python
a = AuthHandler({'example.com', ('foo', 'bar')})
a.remove_strategy('example.com')
assert a.strategies == {}
N)r r )r
r r r r r remove_strategyw s
zAuthHandler.remove_strategyN)
__name__
__module__ __qualname____doc__r
r r r staticmethodr r r r" r r r r r s % r c @ s e Zd Zdd Zdd ZdS )r! c C s dS )Nzr )r
r r r r s zNullAuthStrategy.__repr__c C s |S )Nr )r
rr r r r s zNullAuthStrategy.__call__N)r# r$ r% r r r r r r r! s r! N) r& Z
requests.authr r Zrequests.compatr r r r! r r r r s z