U
��.e&\ � @ s| d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z d dl m
Z
mZ d dlmZmZmZ d dlmZ d dlmZmZmZmZmZmZm Z m Z m Z d d l m Z m!Z! d d
l"m#Z# e#�r
d d l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- d d l.m/Z/ e�0e1�Z2d
d� Z3dd� Z4e4dd� �Z5dd� Z6dd� Z7dd� Z8G dd� de9�Z:G dd � d e9�Z;G d d � d e9�Z.unique)� functools�wraps)r3 r4 r( r2 r) �_unique= s r7 c c s� t �t| �d���}|D ]t}tj�| j|d �}|V |�d�rtj� |�\}}|dd� }tj�||d �}|V tj�||d �}|V qdS )a
Yield all the uninstallation paths for dist based on RECORD-without-.py[co]
Yield paths to all the files in RECORD. For each .py file in RECORD, add
the .pyc and .pyo in the same directory.
UninstallPathSet.add() takes care of the __pycache__ .py[co].
ZRECORDr �.pyN������.pyc�.pyo)
�csv�readerr Zget_metadata_linesr! r" r# �location�endswith�split)r% �r�rowr" Zdnr3 �baser( r( r) �uninstallation_pathsJ s
rD c sJ t jj�t� }t| td�D ]*� t� �fdd�|D ��}|s|�� � q|S )z�Compact a path set to contain the minimal number of paths
necessary to contain all paths in the set. If /a/path/ and
/a/path/to/a/file.txt are both in the set, leave only the
shorter path.��keyc 3 s: | ]2}� � |�d ��o0� t|�d ����� �kV qdS )�*N)�
startswith�rstrip�len)�.0Z shortpath�r" �sepr( r) � l s �zcompact..)r! r" rM r, �sortedrJ �anyr- )�pathsZ short_pathsZ should_skipr( rL r) �compactb s � rR c s� t dd� | D ��}t|�}ttdd� |�� D ��td�}t� }dd� �|D ]��t�fdd�|D ��rfqJt� }t� }t���D ]B\� }}|�� ��fdd�|D �� |�� ��fd d�|D �� q||| sJ|� |� |�
�tj � qJtt |j
|��|B S )
z�Returns a set containing the paths that need to be renamed.
This set may include directories when the original sequence of paths
included every file on disk.
c s s | ]}t j�|�|fV qd S r+ )r! r" �normcase�rK �pr( r( r) rN } s z&compress_for_rename..c s s | ]}t j�|�d V qdS )r N)r! r" r@ rT r( r( r) rN s �rE c W s t j�t jj| � �S r+ )r! r" rS r# )�ar( r( r) � norm_join� s z&compress_for_rename..norm_joinc 3 s | ]}t j�� ��|�V qd S r+ )r! r" rS rH )rK �w)�rootr( r) rN � s �c 3 s | ]}��� |�V qd S r+ r( )rK �d��dirnamerW rY r( r) rN � s �c 3 s | ]}��� |�V qd S r+ r( )rK �fr[ r( r) rN � s �)�dictr, rO �valuesrJ rP r! �walk�update�difference_updater- rM �map� __getitem__) rQ Zcase_mapZ remainingZ uncheckedZ wildcardsZ all_filesZ all_subdirsZsubdirs�filesr( r[ r) �compress_for_renamev s6
�� ���
rf c
C s� t | �}t � }t � }t � }|D ]>}|�d�r.q |�d�s@d|krR|�tj�|�� |�|� q t ttjj|��}t|�}|D ]d}t� |�D ]T\}} }
|
D ]D} | �d�r�q�tj�
|| �} tj� | �r�tj�| �|kr�|�| � q�q�q||dd� |D �B }||fS )as Returns a tuple of 2 sets of which paths to display to user
The first set contains paths that would be deleted. Files of a package
are not added and the top-level directory of the package has a '*' added
at the end - to signify that all it's contents are removed.
The second set contains files that would have been skipped in the above
folders.
r: z __init__.py�
.dist-infoc S s h | ]}t j�|d ��qS )rG )r! r" r# )rK �folderr( r( r) � � s z.compress_for_output_listing..) r, r? r- r! r" r\ rc rS rR r` r# �isfile)
rQ � will_remove� will_skipZfoldersre r" Z_normcased_filesrh �dirpath�_ZdirfilesZfnameZfile_r( r( r) �compress_for_output_listing� s4
��ro c @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z dS )�StashedUninstallPathSetzWA set of file rename operations to stash files while
tentatively uninstalling them.c C s i | _ g | _d S r+ )�
_save_dirs�_moves��selfr( r( r) �__init__� s z StashedUninstallPathSet.__init__c C sD z t |�}W n tk
r* tdd�}Y nX || jtj�|�< |jS )z�Stashes a directory.
Directories are stashed adjacent to their original location if
possible, or else moved/copied into the user's temp dir.� uninstall�Zkind)r �OSErrorr rq r! r" rS )rt r" �save_dirr( r( r) �_get_directory_stash� s z,StashedUninstallPathSet._get_directory_stashc C s� t j�|�}t j�|�d }}d}||krfz| j| }W q�W n tk
rP Y nX t j�|�| }}q"t j�|�}tdd�}|| j|< t j�||�}|r�|t jjkr�t j� |j|�S |jS )z�Stashes a file.
If no root has been provided, one will be created for the directory
in the user's temp directory.Nrv rw )
r! r" rS r\ rq �KeyErrorr �relpath�curdirr# )rt r" �headZold_headry r| r( r( r) �_get_file_stash� s"
z'StashedUninstallPathSet._get_file_stashc C sl t j�|�ot j�|� }|r*| �|�}n
| �|�}| j�||f� |r^t j�|�r^t �|� t ||� |S )z�Stashes the directory or file and returns its new location.
Handle symlinks as files to avoid modifying the symlink targets.
)
r! r" �isdir�islinkrz r rr r$ �rmdirr )rt r" Z path_is_dir�new_pathr( r( r) �stash s
z StashedUninstallPathSet.stashc C s, | j �� D ]\}}|�� q
g | _i | _ dS )z0Commits the uninstall by removing stashed files.N)rq �itemsZcleanuprr )rt rn ry r( r( r) �commit s
z StashedUninstallPathSet.commitc C s� | j D ]}tjd|�� q| j D ]�\}}zTt�d||� tj�|�sPtj�|�r\t� |� ntj�
|�rpt |� t ||� W q t
k
r� } z t�d|� t�d|� W 5 d}~X Y q X q | �� dS )z2Undoes the uninstall by moving stashed files back.�Moving to %s
from %szReplacing %s from %szFailed to restore %sz
Exception: %sN)r� )rr �logging�info�logger�debugr! r" rj r� �unlinkr� r r rx �errorr� )rt rU r� r" Zexr( r( r) �rollback& s
z StashedUninstallPathSet.rollbackc C s
t | j�S r+ )�boolrr rs r( r( r) � can_rollback: s z$StashedUninstallPathSet.can_rollbackN) �__name__�
__module__� __qualname__�__doc__ru rz r r� r� r� �propertyr� r( r( r( r) rp � s rp c @ s^ e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
dd� Z e dd� �Z
dS )�UninstallPathSetzMA set of file paths to be removed in the uninstallation of a
requirement.c C s( t � | _t � | _i | _|| _t� | _d S r+ )r, rQ �_refuse�pthr% rp � _moved_paths)rt r% r( r( r) ru C s
zUninstallPathSet.__init__c C s t |�S )zs
Return True if the given path is one we are permitted to
remove/modify, False otherwise.
)r )rt r" r( r( r) �
_permittedK s zUninstallPathSet._permittedc C s� t j�|�\}}t j�t|�t j�|��}t j�|�s:d S | �|�rR| j� |� n | j
� |� t j� |�d dkr�t r�| � t
|�� d S )N� r8 )r! r" r@ r# r rS �existsr� rQ r- r� �splitextr r )rt r" r~ �tailr( r( r) r- T s
zUninstallPathSet.addc C sL t |�}| �|�r<|| jkr*t|�| j|< | j| �|� n | j�|� d S r+ )r r� r� �UninstallPthEntriesr- r� )rt �pth_file�entryr( r( r) �add_pthh s
zUninstallPathSet.add_pthFc C s� | j st�d| jj� dS | jjd | jj }t�d|� t� �p |sP| �|�r�| j}t | j �}t
t |��D ]}|� |� t�
d|� ql| j�� D ] }|�� q�t�d|� W 5 Q R X dS )z[Remove paths in ``self.paths`` with confirmation (unless
``auto_confirm`` is True).z7Can't uninstall '%s'. No files were found to uninstall.N�-zUninstalling %s:z Removing file or directory %szSuccessfully uninstalled %s)rQ r� r� r% � project_name�versionr
�_allowed_to_proceedr� rf rO rR r� r� r� r_ �remove)rt Z auto_confirm�verboseZdist_name_versionZmovedZ
for_renamer" r� r( r( r) r� r s& ��
zUninstallPathSet.removec C sn dd� }|s t | j�\}}nt| j�}t� }|d|� |d|� |d| j� |r`|dt| j�� tdd�d kS )
zIDisplay which files would be deleted and prompt for confirmation
c S sD |sd S t �| � t� �" tt|��D ]}t �|� q&W 5 Q R X d S r+ )r� r� r
rO rR )�msgrQ r" r( r( r) �_display� s
z6UninstallPathSet._allowed_to_proceed.._displayz
Would remove:z+Would not remove (might be manually added):z%Would not remove (outside of prefix):zWill actually move:zProceed (y/n)? )�y�nr� )ro rQ r, r� rf r )rt r� r� rk rl r( r( r) r� � s
z$UninstallPathSet._allowed_to_proceedc C sR | j js t�d| jj� dS t�d| jj� | j �� | j� � D ] }|�� q@dS )z1Rollback the changes previously made by remove().z'Can't roll back %s; was not uninstalledNz Rolling back uninstall of %s)
r� r� r� r� r% r� r� r� r� r_ )rt r� r( r( r) r� � s �
zUninstallPathSet.rollbackc C s | j �� dS )z?Remove temporary save dir: rollback will no longer be possible.N)r� r� rs r( r( r) r� � s zUninstallPathSet.commitc s� t |j�}t|�s.t�d|j|tj� | |�S |dd� t� d�t� d�hD �krht�d|j|� | |�S | |�}t
|�}d� t �
|j��}|jo�tj�|j�}t|jdd �}|�r�|j�d ��r�|j�|��s�|�|j� |�d
��r |�d
��� D ]&}tj�tj�|j|��} |�| � q�n�|�d ��r�|�d ��rB|�d �� ng � � fd
d�|�d ��� D �D ]J}
tj�|j|
�} |�| � |�| d � |�| d � |�| d � �qb�n|�r�t d� |j���n�|j�d��r |�|j� tj� |j�d } tj�tj� |j�d�} |� | d| � n�|�rP|j�d��rPt |�D ]} |�| � �q.ZstdlibZ
platstdlibz� s �z.UninstallPathSet.from_dist..r8 r: r; z�Cannot uninstall {!r}. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.z.eggr� zeasy-install.pthz./rg rA z)Not sure how to uninstall: %s - Check: %sZscriptsz.bat�console_scripts)�groupF� gui_scriptsT)0r r> r r� r� rF �sys�prefix� sysconfigZget_pathr �formatr Z to_filenamer� Zegg_infor! r" r� �getattrZ _providerr? r- Z has_metadataZ get_metadata�
splitlines�normpathr# r r@ r\ r� rD �openrS �readline�stripr� Zmetadata_isdirZmetadata_listdirr
r r r Z
get_entry_map�keys�extendr* )�clsr% Z dist_pathr' Zdevelop_egg_linkZdevelop_egg_link_egg_infoZegg_info_existsZdistutils_egg_infoZinstalled_filer" Z
top_level_pkgZeasy_install_eggZeasy_install_pth�fhZ link_pointerZscriptr& Z_scripts_to_remover� �namer� �sr( r� r) � from_dist� s�
��
�
�
� � �
�
�� �
� �
zUninstallPathSet.from_distN)FF)r� r� r� r� ru r� r- r� r� r� r� r� � classmethodr� r( r( r( r) r� @ s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r� c C s0 t j�|�std| ��|| _t� | _d | _d S )Nz.Cannot remove entries from nonexistent file %s)r! r" rj r �filer, �entries� _saved_lines)rt r� r( r( r) ru I s �z UninstallPthEntries.__init__c C s< t j�|�}tr,t j�|�d s,|�dd�}| j�|� d S )Nr �\�/)r! r" rS r �
splitdrive�replacer� r- )rt r� r( r( r) r- S s
zUninstallPthEntries.addc C s� t �d| j� t| jd��}|�� }|| _W 5 Q R X tdd� |D ��rLd}nd}|r~|d �|�d��s~|d |�d� |d< | j D ]>}z$t �d |� |�
|| �d�� W q� t k
r� Y q�X q�t| jd
��}|� |� W 5 Q R X d S ) Nz Removing pth entries from %s:�rbc s s | ]}d |kV qdS )s
Nr( )rK �liner( r( r) rN j s z-UninstallPthEntries.remove..z
�
�����zutf-8zRemoving entry: %s�wb)
r� r� r� r� � readlinesr� rP r? �encoder� r� �
ValueError�
writelines)rt r� �linesZendliner� r( r( r) r� c s"
zUninstallPthEntries.removec C sR | j d kr t�d| j� dS t�d| j� t| jd��}|�| j � W 5 Q R X dS )Nz.Cannot roll back changes to %s, none were madeFz!Rolling %s back to previous stater� T)r� r� r� r� r� r� r� )rt r� r( r( r) r� z s
�z UninstallPthEntries.rollbackN)r� r� r� ru r- r� r� r( r( r( r) r� H s
r� )=Z
__future__r r< r5 r� r! r� r� Z pip._vendorr Zpip._internal.exceptionsr Zpip._internal.locationsr r Zpip._internal.utils.compatr r r Zpip._internal.utils.loggingr
Zpip._internal.utils.miscr r r
r r r r r r Z pip._internal.utils.temp_dirr r Zpip._internal.utils.typingr �typingr r r r r r r r r Zpip._vendor.pkg_resourcesr Z getLoggerr� r� r* r7 rD rR rf ro �objectrp r� r� r( r( r( r) � s< , ,
(3o