← Back
Editing: guess.cpython-38.pyc
U �ˀXP � @ 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/dist-packages/requests_toolbelt/auth/guess.py�__init__ s zGuessAuth.__init__c C s� | j d k r|jj�| j � |j |j�� |j�� }t|d�sHt � � |_ t � |j |j|j� |� |j � t�| j| j�| _| �|�}|jj|f|�}|j�|� ||_|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�kwargsZprepZ_rr r r �_handle_basic_auth_401 s z GuessAuth._handle_basic_auth_401c C sf t �| j| j�| _z| j�� 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 �dd��� }d|kr&| �||�S d|kr:| �||�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| � |�S z|j�� | _W n tk r8 Y nX |�d| j� |S �NZresponse)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) r6 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| ��||� || _|| _d | _d S r )�superr; 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|jj�| j � |j |j�� |j�� }t|d�sHt � � |_ t � |j |j|j� |� |j � t�| j| j�| _| �|�}|jj|f|�}|j�|� ||_|S r )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 �_handle_basic_auth_407^ s" � z%GuessProxyAuth._handle_basic_auth_407c C sH t j| j| jd�| _z| j�� W n tk r6 Y nX | jj|f|�S )N)r r ) r ZHTTPProxyDigestAuthr= r> r? r r' r( � handle_407r* r r r �_handle_digest_auth_407s s �z&GuessProxyAuth._handle_digest_auth_407c K s> |j �dd��� }d|kr&| �||�S d|kr:| �||�S d S )NzProxy-Authenticater, r- r. )r/ r0 r1 rB rD )r r" r# Zproxy_authenticater r r rC s zGuessProxyAuth.handle_407c sX | j d k r| � |�}z|j�� | _W n tk r8 Y nX |�d| j� tt| �� |�S r2 ) r? r r3 r r( r4 rC r< r; r6 r5 r@ r r r6 � s zGuessProxyAuth.__call__)NNNN) r7 r8 r9 r: r rB rD rC r6 � __classcell__r r r@ r r; R s � r; N) r: Zrequestsr r r, r r& r ZAuthBaser r; r r r r �<module> s I
Save File
Cancel