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/ e0e1Z2d
d Z3dd Z4e4dd Z5dd Z6dd Z7dd Z8G dd de9Z:G dd d e9Z;G d d d e9Z.unique) functoolswraps)r3 r4 r( r2 r) _unique= s r7 c c s t t| d}|D ]t}tj| j|d }|V |drtj |\}}|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)
csvreaderr Zget_metadata_linesr! r" r# locationendswithsplit)r% rrowr" Zdnr3 baser( r( r) uninstallation_pathsJ s
rD c sJ t jjt }t| tdD ]* t fdd|D }|s| q|S )zCompact 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)
startswithrstriplen).0Z shortpathr" 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 ]tfdd|D rfqJt }t }tD ]B\ }}| fdd|D | fd d|D q||| sJ| | |
tj qJtt |j
||B S )
zReturns 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" normcaserK 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 jt 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 ddirnamerW rY r( r) rN s c 3 s | ]} |V qd S r+ r( )rK fr[ r( r) rN s )dictr, rO valuesrJ rP r! walkupdatedifference_updater- rM map __getitem__) rQ Zcase_mapZ remainingZ uncheckedZ wildcardsZ all_filesZ all_subdirsZsubdirsfilesr( r[ r) compress_for_renamev s6
rf c
C s t | }t }t }t }|D ]>}|dr.q |ds@d|krR|tj| || q t ttjj|}t|}|D ]d}t |D ]T\}} }
|
D ]D} | drqtj
|| } tj | rtj| |kr|| qqq||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_movesselfr( r( r) __init__ s z StashedUninstallPathSet.__init__c C sD z t |}W n tk
r* tdd}Y nX || jtj|< |jS )zStashes a directory.
Directories are stashed adjacent to their original location if
possible, or else moved/copied into the user's temp dir. uninstallZkind)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 qW n tk
rP Y nX t j|| }}q"t j|}tdd}|| j|< t j||}|r|t jjkrt j |j|S |jS )zStashes 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 relpathcurdirr# )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 )zStashes the directory or file and returns its new location.
Handle symlinks as files to avoid modifying the symlink targets.
)
r! r" isdirislinkrz r rr r$ rmdirr )rt r" Z path_is_dirnew_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 ]\}}zTtd|| tj|sPtj|r\t | ntj
|rpt | t || W q t
k
r } z td| td| 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 logginginfologgerdebugr! r" rj r unlinkr r r rx errorr )rt rU r r" Zexr( r( r) rollback& s
z StashedUninstallPathSet.rollbackc C s
t | jS 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 _refusepthr% 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 jt|t j|}t j|s:d S | |rR| j | n | j
| t j |d dkrt 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_fileentryr( r( r) add_pthh s
zUninstallPathSet.add_pthFc C s | j std| jj dS | jjd | jj }td| t p |sP| |r| j}t | j }t
t |D ]}| | t
d| ql| j D ] }| qtd| 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_nameversionr
_allowed_to_proceedr rf rO rR r r r r_ remove)rt Z auto_confirmverboseZdist_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 tddd 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)? )ynr )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 td| jj dS td| 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.td|j|tj | |S |dd t dt dhD krhtd|j| | |S | |}t
|}d t
|j}|jotj|j}t|jdd }|r|jd r|j|s||j |d
r |d
D ]&}tjtj|j|} || qn|d r|d rB|d ng fd
d|d D D ]J}
tj|j|
} || || d || d || d qbn|rt d |jn|jdr ||j tj |jd } tjtj |jd} | | d| n|rP|jdrPt |D ]} || q.ZstdlibZ
platstdlibz s z.UninstallPathSet.from_dist..r8 r: r; zCannot 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.batconsole_scripts)groupF gui_scriptsT)0r r> r r r rF sysprefix sysconfigZget_pathr formatr Z to_filenamer Zegg_infor! r" r getattrZ _providerr? r- Z has_metadataZ get_metadata
splitlinesnormpathr# r r@ r\ r rD openrS readlinestripr Zmetadata_isdirZmetadata_listdirr
r r r Z
get_entry_mapkeysextendr* )clsr% Z dist_pathr' Zdevelop_egg_linkZdevelop_egg_link_egg_infoZegg_info_existsZdistutils_egg_infoZinstalled_filer" Z
top_level_pkgZeasy_install_eggZeasy_install_pthfhZ 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
splitdrivereplacer 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 |ds~|d |d |d< | j D ]>}z$t d | |
|| d W q t k
r Y qX qt| 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: %swb)
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 td| j dS td| 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