File: //usr/local/lib/python3.10/dist-packages/requests_toolbelt/auth/__pycache__/guess.cpython-310.pyc
o
���gP � @ 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
�G/usr/local/lib/python3.10/dist-packages/requests_toolbelt/auth/guess.py�__init__ s
zGuessAuth.__init__c C s� | j d ur
|jj�| j � |j |j�� |j�� }t|d�s$t �
� |_ t � |j |j|j� |�
|j � t�| j| j�| _| �|�}|jj|fi |��}|j�|� ||_|S �N�_cookies)r �request�body�seek�content�raw� release_conn�copy�hasattrr �RequestsCookieJarr �extract_cookies_to_jar�prepare_cookiesr �
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 sh t �| j| j�| _z| j�� W n ty Y nw t| jd�r*| jjd u r*d| j_| jj |fi |��S )N�
num_401_callsr )
� auth_compat�HTTPDigestAuthr 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 �dd��� }d|v r| �||�S d|v r | �||�S dS )z/Resends a request with auth headers, if needed.zwww-authenticate� �basic�digestN)�headers�get�lowerr' r/ )r r# r$ �www_authenticater
r
r r- : s �zGuessAuth.handle_401c C sJ | j d ur
| � |�S z|j�� | _W n ty Y nw |�d| 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- r< r
r
r
r r s r c sJ e Zd ZdZ d
� 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| ��||� || _|| _d | _d S r )�superrA r �proxy_username�proxy_password�
proxy_auth)r r r
rC rD �� __class__r
r r W s
zGuessProxyAuth.__init__c C s� | j d ur
|jj�| j � |j |j�� |j�� }t|d�s$t �
� |_ t � |j |j|j� |�
|j � t�| j| j�| _| �|�}|jj|fi |��}|j�|� ||_|S r )r r r r r r r r r r r r r r r �
HTTPProxyAuthrC rD rE r r r r! r" r
r
r �_handle_basic_auth_407^ s"
�
z%GuessProxyAuth._handle_basic_auth_407c C sJ t j| j| jd�| _z| j�� W n ty Y nw | jj|fi |��S )N)r r
) r �HTTPProxyDigestAuthrC rD rE r r+ r, �
handle_407r. r
r
r �_handle_digest_auth_407s s � �z&GuessProxyAuth._handle_digest_auth_407c K s> |j �dd��� }d|v r| �||�S d|v r | �||�S d S )NzProxy-Authenticater0 r1 r2 )r3 r4 r5 rI rL )r r# r$ �proxy_authenticater
r
r rK s �zGuessProxyAuth.handle_407c sV | j d ur
| � |�}z|j�� | _W n ty Y nw |�d| j� tt| �� |�S r7 )
rE r r9 r r, r: rK rB rA r< r; rF r
r r< � s
�zGuessProxyAuth.__call__)NNNN)
r= r> r? r@ r rI rL rK r< �
__classcell__r
r
rF r rA R s � rA N) r@ �requestsr r r0 r r) r �AuthBaser rA r
r
r
r �<module> s I