3
�PfV0 � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlm Z
ddl m Z ddl
mZmZmZmZmZ ddgZeje�ZG d d
� d
e�Ze� ZG d d � d e�Zd
d� ZdS )z)Handles all VCS (version control) support� )�absolute_importN)�parse)�
BadCommand)� display_path�
backup_dir�call_subprocess�rmtree�ask_path_exists�vcs�get_src_requirementc s� e Zd Zi ZddddddgZ� fdd�Zd d
� Zed d � �Zed
d� �Z edd� �Z
dd� Z d dd�Z dd� Z
dd� Zdd� Z� ZS ) �
VcsSupportZsshZgitZhgZbzrZsftpZsvnc s: t jj| j� tt dd �r(t jj| j� tt| �j� d S )N�
uses_fragment) � urllib_parseZ uses_netloc�extend�schemes�getattrr
�superr �__init__)�self)� __class__� � /usr/lib/python3.6/__init__.pyr s zVcsSupport.__init__c C s
| j j� S )N)� _registry�__iter__)r r r r r $ s zVcsSupport.__iter__c C s t | jj� �S )N)�listr �values)r r r r �backends' s zVcsSupport.backendsc C s dd� | j D �S )Nc S s g | ]
}|j �qS r )�dirname)�.0�backendr r r �
<listcomp>- s z'VcsSupport.dirnames.<locals>.<listcomp>)r )r r r r �dirnames+ s zVcsSupport.dirnamesc C s$ g }x| j D ]}|j|j� q W |S )N)r r r )r r r r r r � all_schemes/ s zVcsSupport.all_schemesc C sF t |d�s tjd|j� d S |j| jkrB|| j|j< tjd|j� d S )N�namezCannot register VCS %szRegistered VCS backend: %s)�hasattr�logger�warning�__name__r# r �debug)r �clsr r r �register6 s
zVcsSupport.registerNc C s< || j kr| j |= n$|| j j� kr.| j |j= n
tjd� d S )Nz0Cannot unregister because no class or name given)r r r# r% r& )r r) r# r r r �
unregister> s
zVcsSupport.unregisterc C s8 x2| j j� D ]$}|j|�r tjd||j� |jS q W dS )z�
Return the name of the version control backend if found at given
location, e.g. vcs.get_backend_name('/path/to/vcs/checkout')
z Determine that %s uses VCS: %sN)r r �controls_locationr% r( r# )r �location�vc_typer r r �get_backend_nameF s
zVcsSupport.get_backend_namec C s |j � }|| jkr | j| S d S )N)�lowerr )r r# r r r � get_backendR s
zVcsSupport.get_backendc C s | j |�}|r| j|�S d S )N)r/ r1 )r r- r. r r r �get_backend_from_locationW s
z$VcsSupport.get_backend_from_location)NN)r' �
__module__� __qualname__r r r r �propertyr r! r" r* r+ r/ r1 r2 �
__classcell__r r )r r r s
r c s� e Zd ZdZdZf Zd+� fdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd,d'd(�Zed)d*� �Z� ZS )-�VersionControl� Nc s || _ tt| �j||� d S )N)�urlr r7 r )r r9 �args�kwargs)r r r r g s zVersionControl.__init__c C s" t jj|�\}}|jt jj�p |S )zy
posix absolute paths start with os.path.sep,
win32 ones start with drive (like c:\folder)
)�os�path�
splitdrive�
startswith�sep)r ZrepoZdrive�tailr r r �_is_local_repositoryk s z#VersionControl._is_local_repositoryc C s |j dd�S )N�/�_)�replace)r Zsurnamer r r �translate_egg_surnameu s z$VersionControl.translate_egg_surnamec C s t �dS )z�
Export the repository at the url to the destination location
i.e. only download the files, without vcs informations
N)�NotImplementedError)r r- r r r �exporty s zVersionControl.exportc C sz d}d| j ks t|| j ��| j jdd�d }tj|�\}}}}}d}d|kr^|jdd�\}}tj||||df�}||fS )zm
Returns the correct repository URL and revision by parsing the given
repository URL
zvSorry, '%s' is a malformed VCS url. The format is <vcs>+<protocol>://<url>, e.g. svn+http://myrepo/svn/MyApp#egg=MyApp�+� N�@r8 )r9 �AssertionError�splitr Zurlsplit�rsplitZ
urlunsplit) r Z
error_messager9 �schemeZnetlocr= ZqueryZfragZrevr r r � get_url_rev� s zVersionControl.get_url_revc C s4 |j d�j| j� s td| ��| j|�| j|�fS )zA
Returns (url, revision), where both are strings
rC zBad directory: %s)�rstrip�endswithr rL �get_url� get_revision)r r- r r r �get_info� s
zVersionControl.get_infoc C s t j|�jd�S )zi
Normalize a URL for comparison by unquoting it and removing any
trailing slash.
rC )r ZunquoterQ )r r9 r r r �
normalize_url� s z VersionControl.normalize_urlc C s | j |�| j |�kS )zV
Compare two repo URLs for identity, ignoring incidental differences.
)rV )r Zurl1Zurl2r r r � compare_urls� s zVersionControl.compare_urlsc C s t �dS )zx
Called when installing or updating an editable package, takes the
source path of the checkout.
N)rG )r �destr r r �obtain� s zVersionControl.obtainc C s t �dS )zB
Switch the repo at ``dest`` to point to ``URL``.
N)rG )r rX r9 � rev_optionsr r r �switch� s zVersionControl.switchc C s t �dS )zO
Update an already-existing repo to the given ``rev_options``.
N)rG )r rX rZ r r r �update� s zVersionControl.updatec C s t �dS )zp
Return True if the version is identical to what exists and
doesn't need to be updated.
N)rG )r rX rZ r r r �
check_version� s z VersionControl.check_versionc
C s� d}d}t jj|�r�d}t jjt jj|| j��r�| j|�}| j||�r�tjd| j j
� t |�|� | j ||�s�tj
dt |�| j |� | j||� q�tj
d� q�tjd| j| j t |�|� d}ntjd || j| j � d}|�r�tjd| j|� td|d |d �}|dk�r2tj
d| j t |�||� | j|||� n~|d k�r>nr|d
k�rftjdt |�� t|� d}nJ|d k�r�t|�} tjdt |�| � tj|| � d}n|dk�r�tjd� |S )z�
Prepare a location to receive a checkout/clone.
Return True if the location is ready for (and requires) a
checkout/clone, False otherwise.
TFz)%s in %s exists, and has correct URL (%s)zUpdating %s %s%sz$Skipping because already up-to-date.z %s %s in %s exists with URL %s�%(s)witch, (i)gnore, (w)ipe, (b)ackup �s�i�w�bz0Directory %s already exists, and is not a %s %s.�(i)gnore, (w)ipe, (b)ackup z+The plan is to install the %s repository %szWhat to do? %sr rJ zSwitching %s %s to %s%sz Deleting %szBacking up %s to %s�a�r_ r` ra rb )r^ re �r` ra rb )rc rf ���)r< r= �exists�joinr rS rW r% r( Z repo_name�titler r] �infor\ r& r# r r[ r r �shutilZmove�sys�exit)
r rX r9 rZ Z rev_displayZcheckout�promptZ existing_urlZresponseZdest_dirr r r �check_destination� s�
z VersionControl.check_destinationc C s" t jj|�rt|� | j|� dS )zq
Clean up current location and download the url repository
(and vcs infos) into location
N)r< r= rh r rY )r r- r r r �unpack s zVersionControl.unpackc C s t �dS )z�
Return a string representing the requirement needed to
redownload the files currently present in location, something
like:
{repository_url}@{revision}#egg={project_name}-{version_identifier}
N)rG )r �distr- r r r r s z"VersionControl.get_src_requirementc C s t �dS )z_
Return the url used at location
Used in get_info or check_destination
N)rG )r r- r r r rS ) s zVersionControl.get_urlc C s t �dS )z_
Return the current revision of the files at location
Used in get_info
N)rG )r r- r r r rT 0 s zVersionControl.get_revisionT�raisec C sf | j g| }yt|||||||�S tk
r` } z$|jtjkrNtd| j ��n� W Y dd}~X nX dS )z�
Run a VCS subcommand
This is simply a wrapper around call_subprocess that adds the VCS
command name, and checks that the VCS is available
zCannot find command %rN)r# r �OSError�errno�ENOENTr ) r �cmdZ show_stdout�cwdZ
on_returncodeZ command_descZ
extra_environZspinner�er r r � run_command7 s zVersionControl.run_commandc C s0 t jd|| j| j� tjj|| j�}tjj|�S )z�
Check if a location is controlled by the vcs.
It is meant to be overridden to implement smarter detection
mechanisms for specific vcs.
z Checking in %s for %s (%s)...)r% r( r r# r< r= ri rh )r) r- r= r r r r, N s z VersionControl.controls_location)N)TNrs NNN)r' r3 r4 r# r r r rB rF rH rP rU rV rW rY r[ r\ r] rp rq r rS rT rz � classmethodr, r6 r r )r r r7 a s2
U
r7 c C sZ t j|�}|rFy|� j| |�S tk
rD tjd||j� | j� S X tjd|� | j� S )NzPcannot determine version of editable source in %s (%s command not found in path)ztcannot determine version of editable source in %s (is not SVN checkout, Git clone, Mercurial clone or Bazaar branch))r
r2 r r r% r& r# Zas_requirement)rr r- Zversion_controlr r r r [ s
) �__doc__Z
__future__r ru Zloggingr< rl rm Z pip._vendor.six.moves.urllibr r Zpip.exceptionsr Z pip.utilsr r r r r �__all__Z getLoggerr' r% �objectr r
r7 r r r r r �<module> s
G {