U
��.eo+ �
@ s d Z ddlZddlmZmZ ddlmZ ddlmZ ddl
m Z m Z m
Z
mZmZ ddlmZ er�ddlmZ dd lmZmZmZ dd
lmZ eeeef Ze� e �Z z ddl Z W nL e k
r� dZ Y n6 e!k
r� Z" ze �#d ee"�� dZ W 5 dZ"["X Y nX d d
� Z$G dd� de�Z%dS )z�Network Authentication Helpers
Contains interface (MultiDomainBasicAuth) and associated glue code for
providing credentials in the context of network requests.
� N)�AuthBase�
HTTPBasicAuth)�get_netrc_auth)�parse)�ask� ask_input� ask_password�remove_auth_from_url�split_auth_netloc_from_url)�MYPY_CHECK_RUNNING)�Values)�Dict�Optional�Tuple)�AuthInfo�*Keyring is skipped due to an exception: %sc
C s� | rt s dS z�z
t j}W n tk
r, Y n4X t�d| � || |�}|dk rZ|j|jfW S W dS |r�t�d| � t �| |�}|r�||fW S W n2 tk
r� } zt� dt
|�� W 5 d}~X Y nX dS )z3Return the tuple auth for a given url from keyring.Nz'Getting credentials from keyring for %sz$Getting password from keyring for %sr ) �keyring�get_credential�AttributeError�logger�debug�username�passwordZ get_password� Exception�warning�str)�urlr r Zcredr �exc� r �>/usr/lib/python3.8/site-packages/pip/_internal/network/auth.py�get_keyring_auth. s,
�r c @ s` e Zd Zddd�Zdd� Zddd�Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dS )�MultiDomainBasicAuthTNc C s || _ || _i | _d | _d S )N)� prompting�
index_urls� passwords�_credentials_to_save)�selfr" r# r r r �__init__M s z MultiDomainBasicAuth.__init__c C sB |r
| j sdS | j D ](}t|��d�d }|�|�r| S qdS )a� Return the original index URL matching the requested URL.
Cached or dynamically generated credentials may work against
the original index URL rather than just the netloc.
The provided url should have had its username and password
removed already. If the original index url had credentials then
they will be included in the return value.
Returns None if no matching index was found, or if --no-index
was specified by the user.
N�/)r# r �rstrip�
startswith)r& r �u�prefixr r r �_get_index_urlY s
z#MultiDomainBasicAuth._get_index_urlc C s� t |�\}}}|\}}|dk r6|dk r6t�d|� |S | �|�} | rft | �}
|
rf|
\} } } t�d| � | r�| d dk r�| \}}|dk r�|dk r�t�d|� | S |r�t|�}
|
r�t�d|� |
S |r�t| |�p�t||�}|r�t�d|� |S ||fS )z2Find and return credentials for the specified URL.Nz Found credentials in url for %szFound index url %sr z%Found credentials in index url for %sz!Found credentials in netrc for %sz#Found credentials in keyring for %s)r
r r r- r r )r& � original_urlZ allow_netrcZ
allow_keyringr �netlocZurl_user_passwordr r Z index_urlZ
index_info�_Zindex_url_user_passwordZ
netrc_authZkr_authr r r �_get_new_credentialsn s> �
� z)MultiDomainBasicAuth._get_new_credentialsc C s� t |�\}}}| j�|d�\}}|dkr>|dkr>| �|�\}}|dk sN|dk rl|pTd}|p\d}||f| j|< |dk r||dk s�|dkr�|dks�td�|���|||fS )a_ Return the credentials to use for the provided URL.
If allowed, netrc and keyring may be used to obtain the
correct credentials.
Returns (url_without_credentials, username, password). Note
that even if the original URL contains credentials, this
function may return a different username and password.
�NNN� z'Could not load credentials from url: {})r
r$ �getr1 �AssertionError�format)r& r. r r/ r0 r r r r r �_get_url_and_credentials� s&
�����z-MultiDomainBasicAuth._get_url_and_credentialsc C sH | � |j�\}}}||_|d k r6|d k r6t||�|�}|�d| j� |S )N�response)r7 r r �
register_hook�
handle_401)r& �reqr r r r r r �__call__� s z MultiDomainBasicAuth.__call__c C sF t d| �}|sdS t||�}|r4|d |d dfS td�}||dfS )Nz
User for %s: r2 r � Fz
Password: T)r r r )r& r/ r Zauthr r r r �_prompt_for_password� s
z)MultiDomainBasicAuth._prompt_for_passwordc C s t sdS tdddg�dkS )NFz#Save credentials to keyring [y/N]: �y�n)r r )r& r r r � _should_save_password_to_keyring� s z5MultiDomainBasicAuth._should_save_password_to_keyringc K s� |j dkr|S | js|S t�|j�}| �|j�\}}}d | _|d k rv|d k rv||f| j|j< |rv| � � rv|j||f| _|j
|j � � t
|p�d|p�d�|j�}|�d| j� | jr�|�d| j� |jj|f|�}|j�|� |S )N� r3 r8 )� status_coder" � urllib_parseZurlparser r> r/ r% r$ rA Zcontent�rawZ release_connr �requestr9 � warn_on_401�save_credentialsZ
connection�send�history�append) r& �resp�kwargsZparsedr r Zsaver; Znew_respr r r r: � s(
z MultiDomainBasicAuth.handle_401c K s |j dkrt�d|jj� dS )z6Response callback to warn about incorrect credentials.rB z)401 Error, Credentials not correct for %sN)rC r r rF r )r& rL rM r r r rG s
�z MultiDomainBasicAuth.warn_on_401c K sn t dk std��t sdS | j}d| _|rj|jdk rjzt�d� t j|� W n tk
rh t�d� Y nX dS )z1Response callback to save credentials on success.Nz'should never reach here without keyringi� z Saving credentials to keyringzFailed to save credentials) r r5 r% rC r �infoZ set_passwordr Z exception)r& rL rM Zcredsr r r rH s
z%MultiDomainBasicAuth.save_credentials)TN)TT)
�__name__�
__module__� __qualname__r' r- r1 r7 r< r> rA r: rG rH r r r r r! K s
�
2( -r! )&�__doc__ZloggingZpip._vendor.requests.authr r Zpip._vendor.requests.utilsr Z pip._vendor.six.moves.urllibr rD Zpip._internal.utils.miscr r r r r
Zpip._internal.utils.typingr Zoptparser �typingr
r r Z pip._internal.vcs.versioncontrolr r Z CredentialsZ getLoggerrO r r � ImportErrorr r r r r! r r r r � s0
�