U
��.e�c � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d d lmZ d d
l m Z d d l m Z d d l!m"Z"m#Z#m$Z$m%Z% d d
l&m'Z'm(Z(m)Z)m*Z* d dl+m,Z, d dl-m.Z. d dl/m0Z0m1Z1 e�r8d dlm2Z3 n d dlm3Z3 e.�r�d dl4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ> d dl?m@Z@ edddddd d d d d d!d"d#d$gZCe�DeE�ZFd%d&� ZGd'd(� ZHd)d"� ZId*d � ZJed+d,d-�d�d/d��ZKd0d1� ZLd2d3� ZMd4d� ZNd�d6d�ZOd7d8� ZPd9d:� ZQd;d� ZRdd?� ZTd@d � ZUdAd � ZVejWfdBdC�ZXd�dEd �ZYdFd� ZZdGd � Z[dHdI� Z\dJdK� Z]dLdM� Z^dNdO� Z_dPdQ� Z`dRdS� ZadDe)dDd.d.dfdTdU�ZbdVdW� ZcdXdY� ZddZd[� Zed\d]� ZfG d^d_� d_eg�ZhG d`da� dae3�Ziejjdbdc� �Zkddd!� Zldedf� ZmG dgdh� dheg�Znd�did#�Zodjdk� Zpdldm� Zqdndo� Zrd�dqdr�Zsdsdt� Ztdudv� Zudwdx� Zvdydz� Zwd{d|� Zxd}d~� Zydd�� Zzd�d$� Z{d�d�� Z|G d�d�� d�eg�Z}d�d�� Z~d�d�� Zd�d�� Z�d�d�� Z�dS )�� )�absolute_importN��deque)�
pkg_resources)�retry)�PY2� text_type)�input)�parse)�unquote)� __version__)� CommandError)�distutils_scheme�get_major_minor_version�
site_packages� user_site)�WINDOWS�
expanduser� stdlib_pkgs�str_to_display)�write_delete_marker_file)�MYPY_CHECK_RUNNING)�running_under_virtualenv�virtualenv_no_global)�BytesIO)�StringIO)
�Any�AnyStr� Container�Iterable�List�Optional�Text�Tuple�Union�cast)� Distributionc C s |S �N� )Ztype_�valuer( r( � } z|jtjkr.� W 5 d}~X Y nX dS )z os.path.makedirs without EEXIST.N)r9 �makedirs�OSError�errnoZEEXIST)r: �er( r( r* r6 q s
c
C sP z0t j�tjd �} | dkr(dtj W S | W S W n tttfk
rJ Y nX dS )Nr )z __main__.pyz-cz %s -m pip�pip) r9 r: �basename�sys�argv�
executable�AttributeError� TypeError�
IndexError)�progr( r( r* r4 { s
i� i� )Zstop_max_delayZ
wait_fixedFc C s t j| |td� d S )N)�
ignore_errors�onerror)�shutilr+ �rmtree_errorhandler)�dirrR r( r( r* r+ � s �c C sX zt �|�jtj@ }W n ttfk
r2 Y dS X |rRt �|tj� | |� dS � dS )z�On Windows, the files in .svn are read-only, so when rmtree() tries to
remove them, an exception is thrown. We catch that here, remove the
read-only attribute, and hopefully continue without problems.N)r9 �stat�st_mode�S_IWRITE�IOErrorrF �chmod)�funcr: �exc_infoZhas_attr_readonlyr( r( r* rU � s rU c C sd | dkr dS t | t�r| S z| �t�� d�}W n0 tk
r^ trRtd�| ��}nt | �}Y nX |S )z�
Convert a bytes (or text) path to text (unicode in Python 2) for display
and logging purposes.
This function should never error out. Also, this function is mainly needed
for Python 2 since in Python 3 str paths are already text.
N�strictzb{!r})
�
isinstancer �decoderK �getfilesystemencoding�UnicodeDecodeErrorr r r? �ascii)r: r, r( r( r* �path_to_display� s
rd c C st t j�t j�| ��} tjd dkrB| �t�� d�} | �t� � d�} | �
t � � t jj �rpd| t
t � � �d� } | S )zTGives the display value for a given path, making it relative to cwd
if possible.r � �replace�.N)r9 r: �normcaser> rK � version_infor` ra �encode�getdefaultencoding�
startswith�getcwd�seprC �r: r( r( r* r, � s �.bakc C s6 d}|}t j�| | �r.|d7 }|t|� }q| | S )z\Figure out the name of a directory to back up the given dir to
(adding .bak, .bak2, etc)� )r9 r: �exists�str)rV �ext�n� extensionr( r( r* r- � s c C s2 t j�dd��� D ]}||kr| S qt| |�S )NZPIP_EXISTS_ACTION� )r9 �environ�get�splitr. )�message�options�actionr( r( r* �ask_path_exists� s
r~ c C s t j�d�rtd| ��dS )z&Raise an error if no input is allowed.Z PIP_NO_INPUTz7No input was expected ($PIP_NO_INPUT set); question: %sN)r9 rx ry � Exception�r{ r( r( r* �_check_no_input� s ��r� c C sF t | � t| �}|�� �� }||kr rh )r: Zresolve_symlinksr( r( r* r2 3 s
c C s@ t �| �\}}|�� �d�r8|dd� | }|dd� }||fS )z,Like os.path.splitext, but take off .tar tooz.tar�����N)� posixpathr/ r� �endswith)r: �basert r( r( r* r/ A s
c C sz t j�|�\}}|r.|r.t j�|�s.t �|� t�| |� t j�| �\}}|rv|rvzt �|� W n tk
rt Y nX dS )z7Like os.renames(), but handles renaming across devices.N) r9 r: rz rr rE rT Zmove�
removedirsrF )�old�new�head�tailr( r( r* r3 K s
c C s t � s
dS | �ttj��S )z�
Return True if path is within sys.prefix, if we're running in a virtualenv.
If we're not in a virtualenv, all paths are considered "local."
Caution: this function assumes the head of path has been normalized
with normalize_path.
T)r rl r2 rK �prefixro r( r( r* �is_local] s
r� c C s t t| ��S )z�
Return True if given Distribution object is installed locally
(i.e. within current virtualenv).
Always True if we're not in a virtualenv.
)r� �
dist_location��distr( r( r* �
dist_is_locall s r� c C s t | ��tt��S )zF
Return True if given Distribution is installed in user site.
)r� rl r2 r r� r( r( r* �dist_in_usersitex s r� c C s t | ��tt��S )z[
Return True if given Distribution is installed in
sysconfig.get_python_lib().
)r� rl r2 r r� r( r( r* �dist_in_site_packages� s r� c C s, t t| ��}|�t td�d �d�d ��S )zf
Return True if given Distribution is installed in
path matching distutils_scheme layout.
rw Zpurelib�pythonr )r2 r� rl r rz )r� Z norm_pathr( r( r* �dist_in_install_path� s �r� c C s6 t jD ]*}tj�|| jd �}tj�|�r dS qdS )zC
Return True if given Distribution is an editable install.
� .egg-linkTF)rK r: r9 r; � project_namer� )r� Z path_item�egg_linkr( r( r* �dist_is_editable� s
r� c s� |rt �|�}nt j}| r t�ndd� �|r6dd� � ndd� � |rLdd� �ndd� �|r^t�nd d
� �� ����fd d �|D �S )
a^
Return a list of installed Distribution objects.
If ``local_only`` is True (default), only return installations
local to the current virtualenv, if in a virtualenv.
``skip`` argument is an iterable of lower-case project names to
ignore; defaults to stdlib_pkgs
If ``include_editables`` is False, don't report editables.
If ``editables_only`` is True , only report editables.
If ``user_only`` is True , only report installations in the user
site directory.
If ``paths`` is set, only report the distributions present at the
specified list of locations.
c S s dS �NTr( ��dr( r( r* �
local_test� s z/get_installed_distributions..local_testc S s dS r� r( r� r( r( r* �
editable_test� s z2get_installed_distributions..editable_testc S s
t | � S r' �r� r� r( r( r* r� � s c S s t | �S r' r� r� r( r( r* �editables_only_test� s z8get_installed_distributions..editables_only_testc S s dS r� r( r� r( r( r* r� � s c S s dS r� r( r� r( r( r* � user_test� s z.get_installed_distributions..user_testc s: g | ]2}�|�r|j �kr� |�r�|�r�|�r|�qS r( )�key)�.0r� �r� r� r� �skipr� r( r* �
� s
�z/get_installed_distributions..)r �
WorkingSet� working_setr� r� )Z
local_onlyr� Zinclude_editablesZeditables_onlyZ user_only�pathsr� r( r� r* �get_installed_distributions� s
r� c C sx g }t � r*|�t� t� sBtrB|�t� ntr8|�t� |�t� |D ],}tj�|| j�d }tj� |�rF| S qFdS )a
Return the path for the .egg-link file if it exists, otherwise, None.
There's 3 scenarios:
1) not in a virtualenv
try to find in site.USER_SITE, then site_packages
2) in a no-global virtualenv
try to find in site_packages
3) in a yes-global virtualenv
try to find in site_packages, then site.USER_SITE
(don't look in global location)
For #1 and #3, there could be odd cases, where there's an egg-link in 2
locations.
This method will just return the first one found.
r� N)
r �appendr r r r9 r: r; r� r� )r� ZsitesZsiteZegglinkr( r( r* �
egg_link_path� s
r� c C s t | �}|rt|�S t| j�S )aO
Get the site-packages location of this distribution. Generally
this is dist.location, except in the case of develop-installed
packages, where dist.location is the source code location, and we
want to know where the egg-link file is.
The returned location is normalized (in particular, with symlinks removed).
)r� r2 �location)r� r� r( r( r* r� s
r� c G s t j| f|�� d S r' )�logger�info)�msg�argsr( r( r* � write_output s r� c C s t �| � t| � d S r' )r9 rE r )Z build_dirr( r( r* �_make_build_dir s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FakeFilezQWrap a list of lines in an object with readline() to make
ConfigParser happy.c C s dd� |D �| _ d S )Nc s s | ]
}|V qd S r' r( )r� �lr( r( r* � & s z$FakeFile.__init__..��_gen)�self�linesr( r( r* �__init__% s zFakeFile.__init__c C sP z4zt | j�W W S tk
r0 | j� � Y W S X W n tk
rJ Y dS X d S )Nrw )�nextr� � NameError�
StopIteration�r� r( r( r* �readline( s zFakeFile.readlinec C s | j S r' r� r� r( r( r* �__iter__1 s zFakeFile.__iter__N)�__name__�
__module__� __qualname__�__doc__r� r� r� r( r( r( r* r� " s r� c @ s$ e Zd Zedd� �Zedd� �ZdS )�
StreamWrapperc C s || _ | � S r' )� orig_stream)�clsr� r( r( r* � from_stream7 s zStreamWrapper.from_streamc C s | j jS r' )r� �encodingr� r( r( r* r� = s zStreamWrapper.encodingN)r� r� r� � classmethodr� �propertyr� r( r( r( r* r� 5 s
r� c
c s@ t t| �}tt| t�|�� zt t| �V W 5 tt| |� X dS )z�Return a context manager used by captured_stdout/stdin/stderr
that temporarily replaces the sys stream *stream_name* with a StringIO.
Taken from Lib/support/__init__.py in the CPython repo.
N)�getattrrK �setattrr� r� )Z stream_nameZ orig_stdoutr( r( r* �captured_outputB s
r� c C s t d�S )z�Capture the output of sys.stdout:
with captured_stdout() as stdout:
print('hello')
self.assertEqual(stdout.getvalue(), 'hello
')
Taken from Lib/support/__init__.py in the CPython repo.
�stdout�r� r( r( r( r* r5 Q s c C s t d�S )z
See captured_stdout().
�stderrr� r( r( r( r* �captured_stderr] s r� c @ s e Zd ZdZdd� Zdd� ZdS )�cached_propertyz�A property that is only computed once per instance and then replaces
itself with an ordinary attribute. Deleting the attribute resets the
property.
Source: https://github.com/bottlepy/bottle/blob/0.11.5/bottle.py#L175
c C s t |d�| _|| _d S )Nr� )r� r� r\ )r� r\ r( r( r* r� l s zcached_property.__init__c C s( |d kr | S | � |� }|j| j j< |S r' )r\ �__dict__r� )r� �objr� r) r( r( r* �__get__p s zcached_property.__get__N)r� r� r� r� r� r� r( r( r( r* r� d s r� c C s4 t j�| �}|dkr t �� }|�|�}|r0|jS dS )zCGet the installed version of dist_name avoiding pkg_resources cacheN)r Z Requirementr
r� �find�version)Z dist_namer� Zreqr� r( r( r* r7 x s
c C s t | dd� dS )z Consume an iterable at C speed.r )�maxlenNr )�iteratorr( r( r* �consume� s r� c O s@ t t| tt| ���f|�}dd� |�� D �}||d< tdd|�S )Nc S s i | ]\}}||�qS r( r( )r� r� r) r( r( r* �
� s zenum..Zreverse_mapping�Enumr( )�dict�zip�rangerC �items�type)Z
sequentialZnamedZenums�reverser( r( r* �enum� s r� c C s* |dkr | S d| kr d� | �} d� | |�S )z.
Build a netloc from a host-port pair
N�:�[{}]z{}:{})r? )Zhost�portr( r( r* � build_netloc� s
r� �httpsc C s4 | � d�dkr(d| kr(d| kr(d�| �} d�|| �S )z)
Build a full URL from a netloc.
r� re �@�[r� z{}://{})�countr? )�netloc�schemer( r( r* �build_url_from_netloc� s
r� c C s t | �}t�|�}|j|jfS )z2
Return the host-port pair from a netloc.
)r� � urllib_parseZurlparseZhostnamer� )r� �urlZparsedr( r( r* � parse_netloc� s
r c C sX d| kr| dfS | � dd�\}} d|kr6|�dd�}n|df}tdd� |D ��}| |fS )zp
Parse out and remove the auth information from a netloc.
Returns: (netloc, (username, password)).
r� )NNrq r� Nc s s" | ]}|d krd nt |�V qd S r' )�urllib_unquote)r� �xr( r( r* r� � s z)split_auth_from_netloc..)�rsplitrz �tuple)r� �authZ user_passr( r( r* �split_auth_from_netloc� s �r c C sL t | �\} \}}|dkr | S |dkr.d}d}nt�|�}d}dj||| d�S )z�
Replace the sensitive data in a netloc with "****", if it exists.
For example:
- "user:[email protected]" returns "user:****@example.com"
- "[email protected]" returns "****@example.com"
N�****rw z:****z{user}{password}@{netloc})�user�passwordr� )r r� Zquoter? )r� r r
r( r( r* �
redact_netloc� s
�r c C s@ t �| �}||j�}|j|d |j|j|jf}t �|�}||fS )aR Transform and replace netloc in a url.
transform_netloc is a function taking the netloc and returning a
tuple. The first element of this tuple is the new netloc. The
entire tuple is returned.
Returns a tuple containing the transformed url as item 0 and the
original tuple returned by transform_netloc as item 1.
r )r� Zurlsplitr� r� r: ZqueryZfragmentZ
urlunsplit)r Ztransform_netlocZpurlZ netloc_tupleZ
url_piecesZsurlr( r( r* �_transform_url� s
�
r c C s t | �S r' )r �r� r( r( r* � _get_netloc s r c C s
t | �fS r' )r r
r( r( r* �_redact_netloc s r c C s t | t�\}\}}|||fS )z�
Parse a url into separate netloc, auth, and url with no auth.
Returns: (url_without_auth, netloc, (username, password))
�r r )r Zurl_without_authr� r r( r( r* �split_auth_netloc_from_url s r c C s t | t�d S )z7Return a copy of url with 'username:password@' removed.r r �r r( r( r* r8 s c C s t | t�d S )z.Replace the password in a given url with ****.r )r r r r( r( r* �redact_auth_from_url s r c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
HiddenTextc C s || _ || _d S r' )�secret�redacted)r� r r r( r( r* r� " s zHiddenText.__init__c C s d� t| ��S )Nz)r? rs r� r( r( r* �__repr__+ s zHiddenText.__repr__c C s | j S r' �r r� r( r( r* �__str__/ s zHiddenText.__str__c C s t | �t |�krdS | j|jkS )NF)r� r �r� �otherr( r( r* �__eq__4 s zHiddenText.__eq__c C s
| |k S r' r( r r( r( r* �__ne__? s zHiddenText.__ne__N)r� r� r� r� r r r r r( r( r( r* r ! s
r c C s t | dd�S )Nr r )r )r) r( r( r* �
hide_valueD s r c C s t | �}t| |d�S )Nr )r r )r r r( r( r* �hide_urlI s r c C s� t � }dD ]P}|�dj|d�� |�djtjd |d�� |�djtjdd� d |i�� q
| oxtoxtj�tj d �|k}|r�tj
d
d gtj d d� }t d
�d� |����dS )z�Protection of pip.exe from modification on Windows
On Windows, any operation modifying pip should be run as:
python -m pip ...
)rw z.exezpip{ext})rt z
pip{}{ext}r z
pip{}.{}{ext}Nre rt z-mrI rq z3To modify pip, please run the following command:
{}� )
�set�addr? rK ri r r9 r: rJ rL rM r
r; )Z
modifying_pipZ pip_namesrt Zshould_show_use_python_msgZ new_commandr( r( r* �(protect_pip_from_modification_on_windowsO s, "�� � ���r# c C s t jdk ot j�� S )z!Is this console interactive?
N)rK �stdin�isattyr( r( r( r* �is_console_interactivem s r& )F)rp )T)N)r� )�Z
__future__r �
contextlibrG r� �ioZloggingr9 r� rT rW rK � collectionsr Z pip._vendorr Zpip._vendor.retryingr Zpip._vendor.sixr r Zpip._vendor.six.movesr Z pip._vendor.six.moves.urllibr
r� Z"pip._vendor.six.moves.urllib.parser r rI r Zpip._internal.exceptionsr
Zpip._internal.locationsr r r r Zpip._internal.utils.compatr r r r Z pip._internal.utils.marker_filesr Zpip._internal.utils.typingr Z pip._internal.utils.virtualenvr r r r �typingr r r r r r! r" r# r$ r% Zpip._vendor.pkg_resourcesr& �intrB �__all__Z getLoggerr� r� r@ rD r6 r4 r+ rU rd r, r- r~ r� r. r� r� r0 r1 �DEFAULT_BUFFER_SIZEr� r2 r/ r3 r� r� r� r� r� r� r� r� r� r� r� �objectr� r� �contextmanagerr� r5 r� r� r7 r� r� r� r� r r r r r r r r8 r r r r r# r& r( r( r( r* � s� 0 �
"
�
E$
#