U
    «ÃT_9\  ã                   @   s|  d dl mZ d dlZd dlZ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 d dlmZmZ d dlmZmZm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 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„ Z3dd„ Z4e4dd„ ƒZ5dd„ Z6dd„ Z7dd„ Z8G dd„ de9ƒZ:G dd„ de9ƒZ;G dd„ de9ƒZ<dS )é    )Úabsolute_importN)Úpkg_resources)ÚUninstallationError)Úbin_pyÚbin_user)ÚWINDOWSÚcache_from_sourceÚuses_pycache)Ú
indent_log)	ÚFakeFileÚaskÚdist_in_usersiteÚdist_is_localÚegg_link_pathÚis_localÚnormalize_pathÚrenamesÚrmtree)ÚAdjacentTempDirectoryÚTempDirectory)ÚMYPY_CHECK_RUNNING)	ÚAnyÚCallableÚDictÚIterableÚIteratorÚListÚOptionalÚSetÚTuple)ÚDistributionc                 C   sl   t | ƒrt}nt}tj ||¡}|g}trh| |d ¡ | |d ¡ |rZ| |d ¡ n| |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
    z.exez.exe.manifestz-script.pywz
-script.py)r   r   r   ÚosÚpathÚjoinr   Úappend)ÚdistÚscript_nameÚis_guiÚbin_dirÚexe_nameÚpaths_to_remove© r+   úC/tmp/pip-unpacked-wheel-8u7dkzi1/pip/_internal/req/req_uninstall.pyÚ_script_names'   s    r-   c                    s   t  ˆ ¡‡ fdd„ƒ}|S )Nc                  ?   s2   t ƒ }ˆ | |ŽD ]}||kr| |¡ |V  qd S ©N)ÚsetÚadd)ÚargsÚkwÚseenÚitem©Úfnr+   r,   Úunique?   s
    
z_unique.<locals>.unique)Ú	functoolsÚwraps)r6   r7   r+   r5   r,   Ú_unique=   s    r:   c                 c   s’   t  t|  d¡ƒ¡}|D ]t}tj | j|d ¡}|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].
    ÚRECORDr   ú.pyNéýÿÿÿú.pycú.pyo)
ÚcsvÚreaderr   Úget_metadata_linesr!   r"   r#   ÚlocationÚendswithÚsplit)r%   ÚrÚrowr"   Údnr6   Úbaser+   r+   r,   Úuninstallation_pathsJ   s    
rJ   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>l   s   þzcompact.<locals>.<genexpr>)r!   r"   rT   r/   ÚsortedrP   Úanyr0   )ÚpathsÚshort_pathsÚshould_skipr+   rS   r,   Úcompactb   s    ýr[   c           	         sø   t dd„ | D ƒƒ}t|ƒ}ttdd„ | ¡ D ƒƒtd�}tƒ }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   | ]}t j |¡|fV  qd S r.   )r!   r"   Únormcase©rQ   Úpr+   r+   r,   rU   }   s     z&compress_for_rename.<locals>.<genexpr>c                 s   s   | ]}t j |¡d  V  qdS )r   N)r!   r"   rE   r]   r+   r+   r,   rU      s   ÿrK   c                  W   s   t j t jj| Ž ¡S r.   )r!   r"   r\   r#   )Úar+   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   ÿc                 3   s   | ]}ˆˆˆ |ƒV  qd S r.   r+   )rQ   Úd©Údirnamer`   rb   r+   r,   rU   �   s   ÿc                 3   s   | ]}ˆˆˆ |ƒV  qd S r.   r+   )rQ   Úfrd   r+   r,   rU   ’   s   ÿ)Údictr/   rV   ÚvaluesrP   rW   r!   ÚwalkÚupdateÚdifference_updater0   rT   ÚmapÚ__getitem__)	rX   Úcase_mapÚ	remainingÚ	uncheckedÚ	wildcardsÚ	all_filesÚall_subdirsÚsubdirsÚfilesr+   rd   r,   Úcompress_for_renamev   s6    
ÿÿÿÿÿ

rv   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.
    r>   z__init__.pyú
.dist-infoc                 S   s   h | ]}t j |d ¡’qS )rM   )r!   r"   r#   )rQ   Úfolderr+   r+   r,   Ú	<setcomp>Ê   s    z.compress_for_output_listing.<locals>.<setcomp>)r/   rD   r0   r!   r"   re   rl   r\   r[   ri   r#   Úisfile)rX   Úwill_removeÚ	will_skipÚfoldersru   r"   Ú_normcased_filesrx   ÚdirpathÚ_ÚdirfilesÚfnameÚfile_r+   r+   r,   Úcompress_for_output_listingž   s4    

ÿÿr„   c                   @   sL   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zdd„ Z	e
dd„ ƒZdS )Ú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 tk
r*   tdd�}Y nX || jtj |¡< |jS )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©Ú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\   re   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 )z�Stashes 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‡   r$   Ú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"   rz   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_rollbackN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__rŠ   r�   r–   rœ   rŸ   r§   Úpropertyr©   r+   r+   r+   r,   r…   Ñ   s   	r…   c                   @   s^   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zddd„Zdd„ Z	dd„ Z
dd„ Zedd„ ƒZdS )Ú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/   rX   Ú_refuseÚpthr%   r…   Ú_moved_paths)r‰   r%   r+   r+   r,   rŠ   C  s
    zUninstallPathSet.__init__c                 C   s   t |ƒS )zs
        Return True if the given path is one we are permitted to
        remove/modify, False otherwise.

        )r   )r‰   r"   r+   r+   r,   Ú
_permittedK  s    zUninstallPathSet._permittedc                 C   sˆ   t j |¡\}}t j t|ƒt j |¡¡}t j |¡s:d S |  |¡rR| j 	|¡ n| j
 	|¡ t j |¡d dkr„tr„|  	t|ƒ¡ d S )Né   r<   )r!   r"   rE   r#   r   r\   Úexistsr³   rX   r0   r°   Úsplitextr	   r   )r‰   r"   r”   Útailr+   r+   r,   r0   T  s    
zUninstallPathSet.addc                 C   sL   t |ƒ}|  |¡r<|| jkr*t|ƒ| j|< | j|  |¡ n| j |¡ d S r.   )r   r³   r±   ÚUninstallPthEntriesr0   r°   )r‰   Úpth_fileÚentryr+   r+   r,   Úadd_pthh  s    

zUninstallPathSet.add_pthFc              	   C   sº   | j st d| jj¡ dS | jjd | jj }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 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)rX   r¡   r¢   r%   Úproject_nameÚversionr
   Ú_allowed_to_proceedr²   rv   rV   r[   rœ   r£   r±   rh   Úremove)r‰   Úauto_confirmÚverboseÚdist_name_versionÚmovedÚ
for_renamer"   r±   r+   r+   r,   rÀ   r  s&    þÿ


zUninstallPathSet.removec                 C   sn   dd„ }|st | jƒ\}}nt| jƒ}tƒ }|d|ƒ |d|ƒ |d| jƒ |r`|dt| jƒƒ tddƒd	kS )
zIDisplay which files would be deleted and prompt for confirmation
        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   r[   )ÚmsgrX   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ÚnrÈ   )r„   rX   r/   r°   rv   r   )r‰   rÂ   rÇ   r{   r|   r+   r+   r,   r¿   ’  s    



z$UninstallPathSet._allowed_to_proceedc                 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Î  t |jƒ}t|ƒs.t d|j|tj¡ | |ƒS |dd„ t 	d¡t 	d¡hD ƒkrht d|j|¡ | |ƒS | |ƒ}t
|ƒ}d t |j¡¡}|joœtj |j¡}t|jdd ƒ}|�r²|j d	¡�r²|j |¡�s²| |j¡ | d
¡�r| d
¡ ¡ D ]&}tj tj |j|¡¡}	| |	¡ qôn�| d¡�rä| d¡�rB| d¡‰ ng ‰ ‡ fdd„| d¡ ¡ D ƒD ]J}
tj |j|
¡}	| |	¡ | |	d ¡ | |	d ¡ | |	d ¡ �qb�n2|�rÌtd |j¡ƒ‚�n|j d¡�r"| |j¡ tj |j¡d }tj tj |j¡d¡}| |d| ¡ nÂ|�rR|j d¡�rRt |ƒD ]}	| |	¡ �q>n’|�rÔt!|dƒ�}tj "| #¡  $¡ ¡}W 5 Q R X ||jk�s¤t%d||j|jf ƒ‚| |¡ tj tj |¡d¡}| ||j¡ nt &d||j¡ | d¡�rT| 'd¡�rT| (d¡D ]L}t)|ƒ�rt*}nt+}| tj ||¡¡ t,�r| tj ||¡d ¡ �qg }|j-dd�}| .¡ D ]}| /t0||dƒ¡ �ql|j-d d�}| .¡ D ]}| /t0||d!ƒ¡ �qš|D ]}| |¡ �q¸|S )"Nz1Not uninstalling %s at %s, outside environment %sc                 S   s   h | ]}|r|’qS r+   r+   r]   r+   r+   r,   ry   Ñ  s    þz-UninstallPathSet.from_dist.<locals>.<setcomp>ÚstdlibÚ
platstdlibz<Not uninstalling %s at %s, as it is in the standard library.z{}.egg-infor"   z	.egg-infozinstalled-files.txtztop_level.txtznamespace_packages.txtc                    s   g | ]}|r|ˆ kr|‘qS r+   r+   r]   ©Ú
namespacesr+   r,   Ú
<listcomp>ù  s    þz.UninstallPathSet.from_dist.<locals>.<listcomp>r<   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´   zeasy-install.pthz./rw   rF   z;Egg-link %s does not match installed location of %s (at %s)z)Not sure how to uninstall: %s - Check: %sÚscriptsz.batÚconsole_scripts)ÚgroupFÚgui_scriptsT)1r   rC   r   r¡   r¢   rL   ÚsysÚprefixÚ	sysconfigÚget_pathr   Úformatr   Úto_filenamer½   Úegg_infor!   r"   rµ   ÚgetattrÚ	_providerrD   r0   Úhas_metadataÚget_metadataÚ
splitlinesÚnormpathr#   r   rE   re   r»   rJ   Úopenr\   ÚreadlineÚstripÚAssertionErrorr£   Úmetadata_isdirÚmetadata_listdirr   r   r   r   Úget_entry_mapÚkeysÚextendr-   )Úclsr%   Ú	dist_pathr*   Údevelop_egg_linkÚdevelop_egg_link_egg_infoÚegg_info_existsÚdistutils_egg_infoÚinstalled_filer"   Útop_level_pkgÚeasy_install_eggÚeasy_install_pthÚfhÚlink_pointerÚscriptr(   Ú_scripts_to_removerÐ   ÚnamerÒ   Úsr+   rÌ   r,   Ú	from_distÄ  sÊ    
üÿ
ý
ÿ
ÿÿÿ
þ

ýÿÿÿÿ
ÿ þ
zUninstallPathSet.from_distN)FF)rª   r«   r¬   r­   rŠ   r³   r0   r»   rÀ   r¿   r§   rŸ   Úclassmethodrù   r+   r+   r+   r,   r¯   @  s   	

 r¯   c                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
r¸   c                 C   s0   t j |¡std| ƒ‚|| _tƒ | _d | _d S )Nz.Cannot remove entries from nonexistent file %s)r!   r"   rz   r   Úfiler/   ÚentriesÚ_saved_lines)r‰   r¹   r+   r+   r,   rŠ   I  s    ÿzUninstallPthEntries.__init__c                 C   s<   t j |¡}tr,t j |¡d s,| dd¡}| j |¡ d S )Nr   ú\ú/)r!   r"   r\   r   Ú
splitdriveÚreplacerü   r0   )r‰   rº   r+   r+   r,   r0   S  s    
zUninstallPthEntries.addc              	   C   sê   t  d| j¡ 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:Úrbc                 s   s   | ]}d |kV  qdS )s   
Nr+   )rQ   Úliner+   r+   r,   rU   j  s     z-UninstallPthEntries.remove.<locals>.<genexpr>z
Ú
éÿÿÿÿzutf-8zRemoving entry: %sÚwb)r¡   r£   rû   rà   Ú	readlinesrý   rW   rD   Úencoderü   rÀ   Ú
ValueErrorÚ
writelines)r‰   ró   ÚlinesÚendlinerº   r+   r+   r,   rÀ   c  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§   z  s    
 ÿzUninstallPthEntries.rollbackN)rª   r«   r¬   rŠ   r0   rÀ   r§   r+   r+   r+   r,   r¸   H  s   
r¸   )=Ú
__future__r   r@   r8   Úloggingr!   rÓ   rÕ   Úpip._vendorr   Úpip._internal.exceptionsr   Úpip._internal.locationsr   r   Úpip._internal.utils.compatr   r   r	   Úpip._internal.utils.loggingr
   Úpip._internal.utils.miscr   r   r   r   r   r   r   r   r   Úpip._internal.utils.temp_dirr   r   Úpip._internal.utils.typingr   Útypingr   r   r   r   r   r   r   r   r   Zpip._vendor.pkg_resourcesr    Ú	getLoggerrª   r¡   r-   r:   rJ   r[   rv   r„   Úobjectr…   r¯   r¸   r+   r+   r+   r,   Ú<module>   s<   ,,

(3o  
