File: //lib/python3.6/site-packages/requests_toolbelt/auth/__pycache__/guess.cpython-36.pyc
3
�P\P � @ sR d Z ddlmZ ddlmZ ddlmZmZ G dd� dej�Z G dd � d e �Z
d
S ) z-The module containing the code for GuessAuth.� )�auth)�cookies� )�_digest_auth_compat�http_proxy_digestc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� GuessAuthz7Guesses the auth type by the WWW-Authentication header.c C s || _ || _d | _d | _d S )N)�username�passwordr �pos)�selfr r � r �/usr/lib/python3.6/guess.py�__init__ s zGuessAuth.__init__c C s� | j d k r|jjj| j � |j |jj� |jj� }t|d�sHt j
� |_ t j |j |j|j� |j
|j � tj| j| j�| _| j|�}|jj|f|�}|jj|� ||_|S )N�_cookies)r
�request�body�seek�content�raw� release_conn�copy�hasattrr �RequestsCookieJarr �extract_cookies_to_jar�prepare_cookiesr Z
HTTPBasicAuthr r �
connection�send�history�append)r �r�kwargs�prep�_rr r r
�_handle_basic_auth_401 s
z GuessAuth._handle_basic_auth_401c C sf t j| j| j�| _y| jj� W n tk
r4 Y nX t| jd�rV| jjd krVd| j_| jj |f|�S )N�
num_401_callsr )
� auth_compatZHTTPDigestAuthr r r �init_per_thread_state�AttributeErrorr r$ �
handle_401)r r r r r r
�_handle_digest_auth_401' s z!GuessAuth._handle_digest_auth_401c K s> |j jdd�j� }d|kr&| j||�S d|kr:| j||�S dS )z/Resends a request with auth headers, if needed.zwww-authenticate� �basic�digestN)�headers�get�lowerr# r) )r r r Zwww_authenticater r r
r( : s
zGuessAuth.handle_401c C sL | j d k r| j |�S y|jj� | _W n tk
r8 Y nX |jd| j� |S )N�response)r r �tellr
r' �
register_hookr( )r r r r r
�__call__E s
zGuessAuth.__call__N) �__name__�
__module__� __qualname__�__doc__r r# r) r( r3 r r r r
r s r c sF e Zd ZdZd
� fdd� Zdd� Zdd� Zd d
� Z� fd d �Z� Z S )�GuessProxyAuthzZ
Guesses the auth type by WWW-Authentication and Proxy-Authentication
headers
Nc s( t t| �j||� || _|| _d | _d S )N)�superr8 r �proxy_username�proxy_password�
proxy_auth)r r r r: r; )� __class__r r
r W s zGuessProxyAuth.__init__c C s� | j d k r|jjj| j � |j |jj� |jj� }t|d�sHt j
� |_ t j |j |j|j� |j
|j � tj| j| j�| _| j|�}|jj|f|�}|jj|� ||_|S )Nr )r
r r r r r r r r r r r r r r Z
HTTPProxyAuthr: r; r< r r r r )r r r r! r" r r r
�_handle_basic_auth_407^ s
z%GuessProxyAuth._handle_basic_auth_407c C sH t j| j| jd�| _y| jj� W n tk
r6 Y nX | jj|f|�S )N)r r ) r ZHTTPProxyDigestAuthr: r; r< r r& r' �
handle_407)r r r r r r
�_handle_digest_auth_407s s z&GuessProxyAuth._handle_digest_auth_407c K s> |j jdd�j� }d|kr&| j||�S d|kr:| j||�S d S )NzProxy-Authenticater* r+ r, )r- r. r/ r> r@ )r r r Zproxy_authenticater r r
r? s
zGuessProxyAuth.handle_407c sX | j d k r| j |�}y|jj� | _W n tk
r8 Y nX |jd| j� tt| �j |�S )Nr0 )
r< r r1 r
r' r2 r? r9 r8 r3 )r r )r= r r
r3 � s
zGuessProxyAuth.__call__)NNNN)
r4 r5 r6 r7 r r> r@ r? r3 �
__classcell__r r )r= r
r8 R s r8 N) r7 Zrequestsr r r* r r% r ZAuthBaser r8 r r r r
�<module> s
I