a
    sh#^                     @  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mZmZ d dl	m
Z
 d dlmZmZ d dlmZ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(dddddddZ)dddddZ*e*dddddZ+dddd d!Z,dddd"d#Z-dd$dd%d&Z.G d'd( d(Z/G d)d* d*Z0G d+d, d,Z1dS )-    )annotationsN)	GeneratorIterable)cache_from_source)AnyCallable)LegacyDistutilsInstallUninstallMissingRecord)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strboolGenerator[str, None, None])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 )zCreate 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#   [/var/www/html/assistant/venv/lib/python3.9/site-packages/pip/_internal/req/req_uninstall.py_script_names   s    r%   z)Callable[..., Generator[Any, None, None]])fnr   c                   s$   t  dddd fdd}|S )Nr   zGenerator[Any, None, None])argskwr   c                  ?  s6   t  } | i |D ]}||vr|| |V  qd S N)setadd)r'   r(   seenitemr&   r#   r$   unique.   s
    
z_unique.<locals>.unique)	functoolswraps)r&   r/   r#   r.   r$   _unique+   s    r2   r   distr   c                 c  s   | j }|dusJ d|  }|du r0t| d|D ]n}tj||}|V  |dr4tj|\}}|dd }tj||d }|V  tj||d }|V  q4d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 an error,
    with possible information from the INSTALLER file.

    https://packaging.python.org/specifications/recording-installed-packages/
    Nznot installeddistribution.py.pyc.pyo)locationiter_declared_entriesr	   r   r    r!   endswithsplit)r4   r;   entriesentryr    dnr&   baser#   r#   r$   uninstallation_paths9   s    

rC   Iterable[str]zset[str])pathsr   c                   sJ   t jjt }t| tdD ]* t fdd|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  kV  qdS )*N)
startswithrstriplen).0	shortpathr    sepr#   r$   	<genexpr>d   s   zcompact.<locals>.<genexpr>)r   r    rO   r*   sortedrK   anyr+   )rE   short_pathsshould_skipr#   rN   r$   compact[   s    rU   c           	        s   dd | D }t |}tdd | D td}t  }dd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 )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   i | ]}t j||qS r#   )r   r    normcaserL   pr#   r#   r$   
<dictcomp>t       z'compress_for_rename.<locals>.<dictcomp>c                 S  s   h | ]}t j|d  qS )r   )r   r    r>   rW   r#   r#   r$   	<setcomp>v   rZ   z&compress_for_rename.<locals>.<setcomp>rF   r   )ar   c                  W  s   t jt jj|  S r)   )r   r    rV   r!   )r\   r#   r#   r$   	norm_joiny   s    z&compress_for_rename.<locals>.norm_joinc                 3  s    | ]}t j |V  qd S r)   )r   r    rV   rI   )rL   w)rootr#   r$   rP   }   rZ   z&compress_for_rename.<locals>.<genexpr>c                 3  s   | ]} |V  qd S r)   r#   )rL   ddirnamer]   r_   r#   r$   rP      rZ   c                 3  s   | ]} |V  qd S r)   r#   )rL   fra   r#   r$   rP      rZ   )r*   rQ   valuesrK   rR   r   walkupdatedifference_updater+   rO   map__getitem__)	rE   case_map	remaining	unchecked	wildcards	all_filesall_subdirssubdirsfilesr#   ra   r$   compress_for_renamen   s"    
rr   ztuple[set[str], set[str]]c                 C  s   t | }t  }t  }t  }|D ]>}|dr.q|ds@d|v rR|tj| || qt ttjj|}t|}|D ]d}t	|D ]T\}}	}
|
D ]D}|drqtj
||}tj|rtj||vr|| qqq||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.
    r9   z__init__.pyz
.dist-infoc                 S  s   h | ]}t j|d qS )rH   )r   r    r!   )rL   folderr#   r#   r$   r[      rZ   z.compress_for_output_listing.<locals>.<setcomp>)r*   r=   r+   r   r    rb   rh   rV   rU   re   r!   isfile)rE   will_remove	will_skipfoldersrq   r    _normcased_filesrs   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ddddd	Zdddd
dZdddddZddddZddddZ	e
ddddZdS )StashedUninstallPathSetzWA set of file rename operations to stash files while
    tentatively uninstalling them.Noner   c                 C  s   i | _ g | _d S r)   )
_save_dirs_movesselfr#   r#   r$   __init__   s    z StashedUninstallPathSet.__init__r   r    r   c                 C  sB   zt |}W n ty(   tdd}Y n0 || 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    rV   )r   r    save_dirr#   r#   r$   _get_directory_stash   s    z,StashedUninstallPathSet._get_directory_stashc                 C  s   t j|}t j|d }}d}||krdz| j| }W qW n tyN   Y n0 t j|| }}q"t j|}tdd}|| j|< t j||}|r|t jjkrt j	|j|S |jS )zStashes 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    rV   rb   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   rd   cleanupr   )r   r   r#   r#   r$   commit  s    
zStashedUninstallPathSet.commitc                 C  s   | j D ]}tjdg|R   q| j D ]\}}zTtd|| tj|sTtj|r`t| ntj	|rtt
| t|| W q$ ty } z$td| td| W Y d}~q$d}~0 0 q$|   dS )z2Undoes the uninstall by moving stashed files back.zMoving to %s
 from %szReplacing %s from %szFailed to restore %szException: %sN)r   loggerinfodebugr   r    rt   r   unlinkr   r   r   r   errorr   )r   rX   r   r    exr#   r#   r$   rollback  s    
$z StashedUninstallPathSet.rollbackr   c                 C  s
   t | jS r)   )r   r   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dddZddd	d
dZddd	ddZddddddZd ddddddZdddddZ	ddddZ
ddddZedd dddZdS )!UninstallPathSetzMA set of file paths to be removed in the uninstallation of a
    requirement.r   r   r3   c                 C  s4   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   .  s    zUninstallPathSet.__init__r   r   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   rI   r   sysprefix)r   r    r#   r#   r$   
_permitted9  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   r7   )r   r    r>   r!   r   rV   existsr   r   r+   r   splitextr   )r   r    r   tailr#   r#   r$   r+   D  s    
zUninstallPathSet.add)pth_filer@   r   c                 C  sN   |  |}| |r>|| jvr,t|| j|< | j| | n| j| d S r)   )r   r   r   UninstallPthEntriesr+   r   )r   r   r@   r#   r#   r$   add_pthW  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 z |sR| |r| j}t	| j }t
t|D ]}|| td| qn| j D ]}|  qtd| W d   n1 s0    Y  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   raw_nameZraw_versionr   _allowed_to_proceedr   rr   rQ   rU   r   r   r   rd   remove)r   r   r   dist_name_versionmoved
for_renamer    pthr#   r#   r$   r   `  s$    


zUninstallPathSet.remove)r   r   c                 C  sx   dddddd}|s&t | j\}}nt| j}t }|d| |d| |d	| j |rj|d
t| j tdddkS )z@Display which files would be deleted and prompt for confirmationr   rD   r   )msgrE   r   c                 S  sX   |sd S t |  t , tt|D ]}t | q&W d    n1 sJ0    Y  d S r)   )r   r   r   rQ   rU   )r   rE   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   )r~   r   r*   r   rr   r   )r   r   r   ru   rv   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   rd   )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 u r*td|j | |S t|}|jsTtd|j|tj | |S |dd t	
dt	
dhD v rtd|j| | |S | |}t|j}|jo|d uotj|o||j d }|r|d ur|| | }|d ur|D ]}	|tj||	 qn|d	r
z|d
}
W n tyJ   g  Y n0 |
jdd  fdd|d	 D D ]N}tj||}|| || d || d || d qtnB|jrt|dn,|jr(|| tj|d }tjtj|d}||d|  n|j rLt!|D ]}|| q8n|rt"|.}tj#|$ % }|&|}W d    n1 s0    Y  tj'||sJ 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t.fy   Y n0 ddd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#   rW   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| vr|qS r#   r#   rW   
namespacesr#   r$   
<listcomp>  s   z.UninstallPathSet.from_dist.<locals>.<listcomp>r7   r9   r:   r5   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.batr   r   r   )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   entry_pointr#   r#   r$   iter_scripts_to_remove8  s
    

z:UninstallPathSet.from_dist.<locals>.iter_scripts_to_remove)/r;   info_locationr   r   canonical_namer   localr   r   	sysconfigget_pathr   r   Z"installed_with_setuptools_egg_infor   r    r   r=   Zsetuptools_filenamer+   r<   r!   is_file	read_textFileNotFoundError
splitlinesZinstalled_by_distutilsr   Zinstalled_as_eggr>   rb   r   Zinstalled_with_dist_inforC   openrV   readlinestripr   samefiler   Zin_usersiter   r
   Ziter_distutils_script_namesr   NotADirectoryError)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_distN)FF)r   r   r   r   r   r   r+   r   r   r   r   r   classmethodr   r#   r#   r#   r$   r   *  s   	r   c                   @  sH   e Zd ZdddddZdddddZdd	d
dZdd	ddZdS )r   r   r   )r   r   c                 C  s   || _ t | _d | _d S r)   )filer*   r?   _saved_lines)r   r   r#   r#   r$   r   I  s    zUninstallPthEntries.__init__)r@   r   c                 C  s<   t j|}tr,t j|d s,|dd}| j| d S )Nr   \/)r   r    rV   r   
splitdrivereplacer?   r+   )r   r@   r#   r#   r$   r+   N  s    
zUninstallPthEntries.addr   c              	   C  s2  t d| j tj| js.t d| j d S t| jd}| }|| _	W d    n1 s^0    Y  t
dd |D rd}nd}|r|d |d	s|d |d	 |d< | jD ]<}z$t d
| ||| d	 W q ty   Y q0 qt| jd}|| W d    n1 s$0    Y  d S )NzRemoving pth entries from %s:z.Cannot remove entries from nonexistent file %srbc                 s  s   | ]}d |v V  qdS )s   
Nr#   )rL   liner#   r#   r$   rP   h  rZ   z-UninstallPthEntries.remove.<locals>.<genexpr>z

zutf-8zRemoving entry: %swb)r   r   r   r   r    rt   warningr   	readlinesr   rR   r=   encoder?   r   
ValueError
writelines)r   r   linesendliner@   r#   r#   r$   r   ]  s(    $
zUninstallPthEntries.remover   c                 C  sf   | j d u rtd| j dS td| j t| jd}|| j  W d    n1 sX0    Y  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   x  s    
*zUninstallPthEntries.rollbackN)r   r   r   r   r+   r   r   r#   r#   r#   r$   r   H  s   r   )2
__future__r   r0   r   r   r   collections.abcr   r   importlib.utilr   typingr   r   pip._internal.exceptionsr   r	   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   r%   r2   rC   rU   rr   r~   r   r   r   r#   r#   r#   r$   <module>   s:   !"1i   