3
�P\8 � @ sH d Z ddlmZmZ ddlmZmZ G dd� de�ZG dd� de�ZdS ) 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)
# => <Response [200]>
r = requests.get('https://example.com/some/path', auth=auth)
# => <Response [200]>
s = requests.Session()
s.auth = auth
r = s.get('https://api.github.com/user')
# => <Response [200]>
.. 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_for�url)r
ZrequestZauthr r r �__call__: s zAuthHandler.__call__c C s dj | j�S )Nz<AuthHandler({0!r})>)�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_strategies�k�vr r r r A s zAuthHandler._make_uniformc C s( t | �}t|jj� |jj� ddddf�S )N� )r r �scheme�lowerZnetloc)r Zparsedr r r �
_key_from_urlH s
zAuthHandler._key_from_urlc C s* t |t�rt|� }| 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)�
isinstance�tupler r r )r
�domainZstrategy�keyr 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 �get�NullAuthStrategy)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 )Nz<NullAuthStrategy>r )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 �<module> s z