U
    L‹Ðdf`  ã                   @   sÖ  d dl Z d dlZd dlZd dlZd dlmZ 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 d dlmZmZ d dlmZ d dlmZ d dlmZ d d	lmZmZ d d
lmZm Z m!Z!m"Z" d dl#m$Z$m%Z% d dl&m'Z' ee(ƒZ)e*e*e+e
e*ddf dœdd„Z,ede
eddf f ede
eddf f dœdd„Z-e-ee
e*ddf dœdd„ƒZ.ee* ee* dœdd„Z/ee* ee* dœdd„Z0ee* eee* ee* f dœdd„Z1G dd„ dƒZ2G d d!„ d!ƒZ3G d"d#„ d#ƒZ4dS )$é    N)Úcache_from_source)	ÚAnyÚCallableÚDictÚ	GeneratorÚIterableÚListÚOptionalÚSetÚTuple)ÚUninstallationError)Úget_bin_prefixÚget_bin_user)ÚBaseDistribution)ÚWINDOWS)Úegg_link_path_from_location)Ú	getLoggerÚ
indent_log)ÚaskÚnormalize_pathÚrenamesÚrmtree)ÚAdjacentTempDirectoryÚTempDirectory)Úrunning_under_virtualenv)Úbin_dirÚscript_nameÚis_guiÚreturnc                 c   sV   t j | |¡}|V  tsdS |› dV  |› dV  |rF|› dV  n|› dV  dS )z‘Create the fully qualified name of the files created by
    {console,gui}_scripts for the given ``dist``.
    Returns the list of file names
    Nz.exez.exe.manifestz-script.pywz
-script.py)ÚosÚpathÚjoinr   )r   r   r   Úexe_name© r#   ú@/tmp/pip-install-netrr6me/pip/pip/_internal/req/req_uninstall.pyÚ_script_names   s    r%   .)Úfnr   c                    s.   t  ˆ ¡ttttd d f dœ‡ fdd„ƒ}|S )N)ÚargsÚkwr   c                  ?   s2   t ƒ }ˆ | |ŽD ]}||kr| |¡ |V  qd S ©N)ÚsetÚadd)r'   r(   ÚseenÚitem©r&   r#   r$   Úunique+   s
    
z_unique.<locals>.unique)Ú	functoolsÚwrapsr   r   )r&   r/   r#   r.   r$   Ú_unique(   s    "r2   ©Údistr   c                 c   sò   | j }|dk	stdƒ‚|  ¡ }|dkrzdj| d}| j}|rD|dkrdd | j| j¡}|d |¡7 }n|d |¡7 }t|ƒ‚|D ]n}tj	 
||¡}|V  | d	¡r~tj	 |¡\}}	|	dd
… }
tj	 
||
d ¡}|V  tj	 
||
d ¡}|V  q~dS )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].

    If RECORD is not found, raises UninstallationError,
    with possible information from the INSTALLER file.

    https://packaging.python.org/specifications/recording-installed-packages/
    Nznot installedz/Cannot uninstall {dist}, RECORD file not found.)r4   Úpipz{}=={}zZ You might be able to recover from this via: 'pip install --force-reinstall --no-deps {}'.z' Hint: The package was installed by {}.ú.pyéýÿÿÿú.pycú.pyo)ÚlocationÚAssertionErrorÚiter_declared_entriesÚformatÚ	installerÚraw_nameÚversionr   r   r    r!   ÚendswithÚsplit)r4   r:   ÚentriesÚmsgr>   ÚdepÚentryr    Údnr&   Úbaser#   r#   r$   Úuninstallation_paths6   s2    ÿÿ
rI   )Úpathsr   c                    sJ   t jj‰tƒ }t| tdD ]*‰ t‡ ‡fdd„|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 ¡ ˆ¡ƒ ˆkV  qdS )Ú*N)Ú
startswithÚrstripÚlen)Ú.0Ú	shortpath©r    Úsepr#   r$   Ú	<genexpr>k   s   þzcompact.<locals>.<genexpr>)r   r    rT   r*   ÚsortedrP   Úanyr+   )rJ   Úshort_pathsÚshould_skipr#   rS   r$   Úcompactb   s    ýrZ   c           	         sø   dd„ | D ƒ}t |ƒ}tdd„ | ¡ D ƒtd}t ƒ }ttdœdd„‰|D ]˜‰t‡fd	d
„|D ƒƒrfqJt ƒ }t ƒ }t ˆ¡D ]B\‰ }}| ‡ ‡‡fdd
„|D ƒ¡ | ‡ ‡‡fdd
„|D ƒ¡ q||| sJ| 	|¡ | 
ˆtj ¡ qJt t|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   i | ]}t j |¡|“qS r#   )r   r    Únormcase©rQ   Úpr#   r#   r$   Ú
<dictcomp>{   s     
 z'compress_for_rename.<locals>.<dictcomp>c                 S   s   h | ]}t j |¡d  ’qS )r   )r   r    rB   r\   r#   r#   r$   Ú	<setcomp>}   s     z&compress_for_rename.<locals>.<setcomp>rK   )Úar   c                  W   s   t j t jj| Ž ¡S r)   )r   r    r[   r!   )r`   r#   r#   r$   Ú	norm_join€   s    z&compress_for_rename.<locals>.norm_joinc                 3   s    | ]}t j ˆ ¡ |¡V  qd S r)   )r   r    r[   rN   )rQ   Úw)Úrootr#   r$   rU   „   s     z&compress_for_rename.<locals>.<genexpr>c                 3   s   | ]}ˆˆˆ |ƒV  qd S r)   r#   )rQ   Úd©Údirnamera   rc   r#   r$   rU   ‹   s     c                 3   s   | ]}ˆˆˆ |ƒV  qd S r)   r#   )rQ   Úfre   r#   r$   rU   Œ   s     )r*   rV   ÚvaluesrP   ÚstrrW   r   ÚwalkÚupdateÚdifference_updater+   rT   ÚmapÚ__getitem__)	rJ   Úcase_mapÚ	remainingÚ	uncheckedÚ	wildcardsÚ	all_filesÚall_subdirsÚsubdirsÚfilesr#   re   r$   Úcompress_for_renameu   s"    
rw   c                 C   sü   t | ƒ}t ƒ }t ƒ }t ƒ }|D ]>}| d¡r.q| d¡s@d|krR| tj |¡¡ | |¡ qt ttjj|ƒƒ}t|ƒ}|D ]d}t 	|¡D ]T\}}	}
|
D ]D}| d¡r¨q˜tj 
||¡}tj |¡r˜tj |¡|kr˜| |¡ q˜qŠq||dd„ |D ƒB }||fS )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.
    r8   z__init__.pyz
.dist-infoc                 S   s   h | ]}t j |d ¡’qS )rM   )r   r    r!   )rQ   Úfolderr#   r#   r$   r_   Ä   s     z.compress_for_output_listing.<locals>.<setcomp>)r*   rA   r+   r   r    rf   rm   r[   rZ   rj   r!   Úisfile)rJ   Úwill_removeÚ	will_skipÚfoldersrv   r    Ú_normcased_filesrx   ÚdirpathÚ_ÚdirfilesÚfnameÚfile_r#   r#   r$   Úcompress_for_output_listing—   s2    


ÿþrƒ   c                   @   s|   e Zd ZdZddœdd„Zeedœdd„Zeedœd	d
„Zeedœdd„Zddœdd„Z	ddœdd„Z
eedœdd„ƒZdS )ÚStashedUninstallPathSetzWA set of file rename operations to stash files while
    tentatively uninstalling them.N©r   c                 C   s   i | _ g | _d S r)   )Ú
_save_dirsÚ_moves©Úselfr#   r#   r$   Ú__init__Í   s    z StashedUninstallPathSet.__init__©r    r   c                 C   sD   zt |ƒ}W n tk
r*   tdd}Y nX || jtj |¡< |jS )zStashes a directory.

        Directories are stashed adjacent to their original location if
        possible, or else moved/copied into the user's temp dir.Ú	uninstall©Úkind)r   ÚOSErrorr   r†   r   r    r[   )r‰   r    Úsave_dirr#   r#   r$   Ú_get_directory_stashÕ   s    z,StashedUninstallPathSet._get_directory_stashc                 C   sº   t j |¡}t j |¡d }}d}||krfz| j| }W q†W n tk
rP   Y nX t j |¡| }}q"t j |¡}tdd}|| j|< t j ||¡}|r´|t jjkr´t j 	|j|¡S |jS )z‚Stashes a file.

        If no root has been provided, one will be created for the directory
        in the user's temp directory.NrŒ   r   )
r   r    r[   rf   r†   ÚKeyErrorr   ÚrelpathÚcurdirr!   )r‰   r    ÚheadÚold_headr   r“   r#   r#   r$   Ú_get_file_stashã   s"    


z'StashedUninstallPathSet._get_file_stashc                 C   sl   t j |¡ot 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    ÚisdirÚislinkr‘   r—   r‡   ÚappendÚrmdirr   )r‰   r    Úpath_is_dirÚnew_pathr#   r#   r$   Ústashþ   s    


zStashedUninstallPathSet.stashc                 C   s,   | j  ¡ D ]\}}| ¡  q
g | _i | _ dS )z0Commits the uninstall by removing stashed files.N)r†   ÚitemsÚcleanupr‡   )r‰   r   r   r#   r#   r$   Úcommit  s    
zStashedUninstallPathSet.commitc                 C   sÆ   | j D ]}tjd|žŽ  q| j D ]˜\}}zTt d||¡ tj |¡sPtj |¡r\t |¡ ntj 	|¡rpt
|ƒ t||ƒ W q  tk
r¶ } zt d|¡ t d|¡ W 5 d}~X Y q X q |  ¡  dS )z2Undoes the uninstall by moving stashed files back.úMoving to %s
 from %szReplacing %s from %szFailed to restore %szException: %sN)r¢   )r‡   ÚloggerÚinfoÚdebugr   r    ry   r™   Úunlinkr˜   r   r   r   Úerrorr¡   )r‰   r]   r   r    Úexr#   r#   r$   Úrollback  s    
 z StashedUninstallPathSet.rollbackc                 C   s
   t | jƒS r)   )Úboolr‡   rˆ   r#   r#   r$   Úcan_rollback-  s    z$StashedUninstallPathSet.can_rollback)Ú__name__Ú
__module__Ú__qualname__Ú__doc__rŠ   ri   r‘   r—   rž   r¡   r©   Úpropertyrª   r«   r#   r#   r#   r$   r„   É   s   r„   c                   @   s¦   e Zd ZdZeddœdd„Zeedœdd„Zeddœd	d
„Z	eeddœdd„Z
deeddœdd„Zeedœdd„Zddœdd„Zddœdd„Zeed dœdd„ƒZdS )ÚUninstallPathSetzMA set of file paths to be removed in the uninstallation of a
    requirement.Nr3   c                 C   s6   t ƒ | _t ƒ | _i | _|| _tƒ | _t ¡ t	ƒ| _
d S r)   )r*   Ú_pathsÚ_refuseÚ_pthÚ_distr„   Ú_moved_pathsr0   Ú	lru_cacher   Ú_normalize_path_cached)r‰   r4   r#   r#   r$   rŠ   6  s    zUninstallPathSet.__init__r‹   c                 C   s   t ƒ s
dS | |  tj¡¡S )zs
        Return True if the given path is one we are permitted to
        remove/modify, False otherwise.

        T)r   rN   r¸   ÚsysÚprefix)r‰   r    r#   r#   r$   Ú
_permittedA  s    zUninstallPathSet._permittedc                 C   s†   t j |¡\}}t j |  |¡t j |¡¡}t j |¡s<d S |  |¡rT| j 	|¡ n| j
 	|¡ t j |¡d dkr‚|  	t|ƒ¡ d S )Né   r6   )r   r    rB   r!   r¸   r[   Úexistsr»   r²   r+   r³   Úsplitextr   )r‰   r    r•   Útailr#   r#   r$   r+   L  s    
zUninstallPathSet.add)Úpth_filerF   r   c                 C   sN   |   |¡}|  |¡r>|| jkr,t|ƒ| j|< | j|  |¡ n| j |¡ d S r)   )r¸   r»   r´   ÚUninstallPthEntriesr+   r³   )r‰   rÀ   rF   r#   r#   r$   Úadd_pth_  s    


zUninstallPathSet.add_pthF)Úauto_confirmÚverboser   c              	   C   s¼   | j st d| jj¡ dS | jj› d| jj› }t d|¡ tƒ p |sR|  |¡r®| j}t	| j ƒ}t
t|ƒƒD ]}| |¡ t d|¡ qn| j ¡ D ]}| ¡  q”t d|¡ W 5 Q R X dS )z\Remove paths in ``self._paths`` with confirmation (unless
        ``auto_confirm`` is True).z7Can't uninstall '%s'. No files were found to uninstall.Nú-zUninstalling %s:zRemoving file or directory %szSuccessfully uninstalled %s)r²   r£   r¤   rµ   r?   r@   r   Ú_allowed_to_proceedr¶   rw   rV   rZ   rž   rÄ   r´   rh   Úremove)r‰   rÃ   rÄ   Údist_name_versionÚmovedÚ
for_renamer    Úpthr#   r#   r$   rÇ   h  s$    þ


zUninstallPathSet.remove)rÄ   r   c                 C   s|   t tt  ddœdd„}|s*t| jƒ\}}nt| jƒ}tƒ }|d|ƒ |d|ƒ |d| jƒ |rn|dt| jƒƒ td	d
ƒdkS )z@Display which files would be deleted and prompt for confirmationN)rD   rJ   r   c              	   S   sD   |sd S t  | ¡ tƒ " tt|ƒƒD ]}t  |¡ q&W 5 Q R X d S r)   )r£   r¤   r   rV   rZ   )rD   rJ   r    r#   r#   r$   Ú_displayˆ  s    
z6UninstallPathSet._allowed_to_proceed.<locals>._displayzWould remove:z+Would not remove (might be manually added):z%Would not remove (outside of prefix):zWill actually move:zProceed (Y/n)? )ÚyÚnÚ rÎ   )ri   r   rƒ   r²   r*   r³   rw   r   )r‰   rÄ   rÌ   rz   r{   r#   r#   r$   rÆ   …  s    	


z$UninstallPathSet._allowed_to_proceedr…   c                 C   sR   | j jst d| jj¡ dS t d| jj¡ | j  ¡  | j 	¡ D ]}| ¡  q@dS )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´   rh   )r‰   rË   r#   r#   r$   r©   ¡  s    þ
zUninstallPathSet.rollbackc                 C   s   | j  ¡  dS )z?Remove temporary save dir: rollback will no longer be possible.N)r¶   r¡   rˆ   r#   r#   r$   r¡   ®  s    zUninstallPathSet.commitc              
      s²  |j }|j}|d kr*t d|j¡ | |ƒS t|ƒ}|jsTt d|j|tj¡ | |ƒS |dd„ t	 
d¡t	 
d¡hD ƒkrŽt d|j|¡ | |ƒS | |ƒ}t|jƒ}|joÌ|d k	oÌtj |¡oÌ| |j› d¡ }|rÊ|d k	ræ| |¡ | ¡ }|d k	r|D ]}	| tj ||	¡¡ qün®| d	¡rüz| d
¡}
W n tk
rL   g ‰ Y nX |
jdd‰ ‡ fdd„| d	¡ ¡ D ƒD ]N}tj ||¡}| |¡ | |› d¡ | |› d¡ | |› d¡ qvn2|jrætd |j¡ƒ‚n|jr0| |¡ tj |¡d }tj tj |¡d¡}|  |d| ¡ nÌ|j!rTt"|ƒD ]}| |¡ q@n¨|rît#|ƒ$}tj $| %¡  &¡ ¡}| '|¡}W 5 Q R X tj (||¡sÀt)d|› d|› d|j› d|› d	ƒ‚| |¡ tj tj |¡d¡}|  ||¡ nt *d||¡ |j+rt,ƒ }nt-ƒ }zH| .¡ D ]:}| tj ||¡¡ t/r| tj ||› d¡¡ qW n tt0fk
rt   Y nX t1t2t3t2d d f dœdd„}|||ƒD ]}| |¡ qœ|S ) Nz-Not uninstalling %s since it is not installedz1Not uninstalling %s at %s, outside environment %sc                 S   s   h | ]}|r|’qS r#   r#   r\   r#   r#   r$   r_   Ç  s   þz-UninstallPathSet.from_dist.<locals>.<setcomp>ÚstdlibÚ
platstdlibz<Not uninstalling %s at %s, as it is in the standard library.z	.egg-infoztop_level.txtznamespace_packages.txtF)Úkeependsc                    s   g | ]}|r|ˆ kr|‘qS r#   r#   r\   ©Ú
namespacesr#   r$   Ú
<listcomp>õ  s    þz.UninstallPathSet.from_dist.<locals>.<listcomp>r6   r8   r9   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.r¼   zeasy-install.pthz./z	Egg-link z (to z') does not match installed location of z (at ú)z)Not sure how to uninstall: %s - Check: %sz.bat)r4   r   r   c                 s   sP   |   ¡ D ]B}|jdkr,t||jdƒE d H  q|jdkrt||jdƒE d H  qd S )NÚconsole_scriptsFÚgui_scriptsT)Úiter_entry_pointsÚgroupr%   Úname)r4   r   Zentry_pointr#   r#   r$   Úiter_scripts_to_removeC  s
    

z:UninstallPathSet.from_dist.<locals>.iter_scripts_to_remove)4r:   Úinfo_locationr£   r¤   Úcanonical_namer   Úlocalr¹   rº   Ú	sysconfigÚget_pathr   r?   Z"installed_with_setuptools_egg_infor   r    r½   rA   Zsetuptools_filenamer+   r<   r!   Úis_fileÚ	read_textÚFileNotFoundErrorÚ
splitlinesZinstalled_by_distutilsr   r=   Zinstalled_as_eggrB   rf   rÂ   Zinstalled_with_dist_inforI   Úopenr[   ÚreadlineÚstripr¸   Úsamefiler;   r¥   Zin_usersiter   r   Ziter_distutils_script_namesr   ÚNotADirectoryErrorr   ri   r   )Úclsr4   Údist_locationrÝ   Znormalized_dist_locationÚpaths_to_removeÚdevelop_egg_linkZsetuptools_flat_installationZinstalled_filesÚinstalled_fileZnamespace_packagesÚtop_level_pkgr    Úeasy_install_eggÚeasy_install_pthÚfhÚlink_pointerZnormalized_link_pointerr   ÚscriptrÜ   Úsr#   rÓ   r$   Ú	from_dist²  sâ    þüþý
ÿ
þú



þ

ýÿ

þ
ÿ ÿý

 ÿý"ý
zUninstallPathSet.from_dist)FF)r¬   r­   r®   r¯   r   rŠ   ri   rª   r»   r+   rÂ   rÇ   rÆ   r©   r¡   Úclassmethodr÷   r#   r#   r#   r$   r±   2  s   	r±   c                   @   sH   e Zd Zeddœdd„Zeddœdd„Zddœd	d
„Zedœdd„ZdS )rÁ   N)rÀ   r   c                 C   s   || _ tƒ | _d | _d S r)   )Úfiler*   rC   Ú_saved_lines)r‰   rÀ   r#   r#   r$   rŠ   T  s    zUninstallPthEntries.__init__)rF   r   c                 C   s<   t j |¡}tr,t j |¡d s,| dd¡}| j |¡ d S )Nr   ú\ú/)r   r    r[   r   Ú
splitdriveÚreplacerC   r+   )r‰   rF   r#   r#   r$   r+   Y  s    
zUninstallPthEntries.addr…   c              	   C   s
  t  d| j¡ tj | j¡s.t  d| j¡ d S t| jdƒ}| ¡ }|| _	W 5 Q R X t
dd„ |D ƒƒrld}nd}|rž|d  | d	¡¡sž|d | d	¡ |d< | jD ]>}z$t  d
|¡ | ||  d	¡¡ W q¤ tk
rà   Y q¤X q¤t| jdƒ}| |¡ W 5 Q R X d S )NzRemoving pth entries from %s:z.Cannot remove entries from nonexistent file %sÚrbc                 s   s   | ]}d |kV  qdS )s   
Nr#   )rQ   Úliner#   r#   r$   rU   s  s     z-UninstallPthEntries.remove.<locals>.<genexpr>z
Ú
éÿÿÿÿzutf-8zRemoving entry: %sÚwb)r£   rÄ   rù   r   r    ry   Úwarningræ   Ú	readlinesrú   rW   rA   ÚencoderC   rÇ   Ú
ValueErrorÚ
writelines)r‰   ró   ÚlinesÚendlinerF   r#   r#   r$   rÇ   h  s(    
zUninstallPthEntries.removec              	   C   sR   | j d krt d| j¡ dS t d| j¡ t| jdƒ}| | j ¡ W 5 Q R X dS )Nz.Cannot roll back changes to %s, none were madeFz!Rolling %s back to previous stater  T)rú   r£   r§   rù   r¥   ræ   r  )r‰   ró   r#   r#   r$   r©   ƒ  s    
zUninstallPthEntries.rollback)	r¬   r­   r®   ri   rŠ   r+   rÇ   rª   r©   r#   r#   r#   r$   rÁ   S  s   rÁ   )5r0   r   r¹   rà   Úimportlib.utilr   Útypingr   r   r   r   r   r   r	   r
   r   Úpip._internal.exceptionsr   Úpip._internal.locationsr   r   Zpip._internal.metadatar   Úpip._internal.utils.compatr   Zpip._internal.utils.egg_linkr   Úpip._internal.utils.loggingr   r   Úpip._internal.utils.miscr   r   r   r   Úpip._internal.utils.temp_dirr   r   Úpip._internal.utils.virtualenvr   r¬   r£   ri   rª   r%   r2   rI   rZ   rw   rƒ   r„   r±   rÁ   r#   r#   r#   r$   Ú<module>   sB   ,  þþ+"$2i  #