B
    `d                 @   s   d dl mZ d dlZd dl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 d d	lmZ d d
lmZmZmZ erd dl	mZ d dlmZ eeZG dd deZee dS )    )absolute_importN)configparser)
BadCommandInstallationError)display_path)make_command)TempDirectory)MYPY_CHECK_RUNNING)path_to_url)VersionControl!find_path_to_setup_from_repo_rootvcs)
HiddenText)
RevOptionsc                   s   e Zd ZdZdZdZdZedd Zdd Z	d	d
 Z
dd Zdd Zedd Zedd Zedd Zedd Zedd Ze fddZ  ZS )	Mercurialhgz.hgclone)r   zhg+filezhg+httpzhg+httpszhg+sshzhg+static-httpc             C   s   | gS )N )revr   r   ?/tmp/pip-unpacked-wheel-hu9qqap1/pip/_internal/vcs/mercurial.pyget_base_rev_args'   s    zMercurial.get_base_rev_argsc          	   C   s@   t dd,}| j|j|d | jd|gd|jd W dQ R X dS )z?Export the Hg repository at the url to the destination locationexport)kind)urlarchiveF)show_stdoutcwdN)r   unpackpathrun_command)selflocationr   temp_dirr   r   r   r   +   s    zMercurial.exportc             C   sP   |  }td||t| | tddd|| | jtdd| |d d S )NzCloning hg %s%s to %sr   z
--noupdatez-qupdate)r   )
to_displayloggerinfor   r   r   to_args)r    destr   rev_optionsrev_displayr   r   r   	fetch_new5   s    
zMercurial.fetch_newc       	   
   C   s   t j|| jd}t }y>|| |dd|j t	|d}|
| W d Q R X W n6 ttjfk
r } ztd|| W d d }~X Y n X tdd| }| j||d d S )	Nhgrcpathsdefaultwz/Could not switch Mercurial repository to %s: %sr#   z-q)r   )osr   joindirnamer   RawConfigParserreadsetsecretopenwriteOSErrorNoSectionErrorr%   warningr   r'   r   )	r    r(   r   r)   repo_configconfigconfig_fileexccmd_argsr   r   r   switchD   s    
zMercurial.switchc             C   s4   | j ddg|d tdd| }| j ||d d S )Npullz-q)r   r#   )r   r   r'   )r    r(   r   r)   r@   r   r   r   r#   U   s    zMercurial.updatec             C   s2   | j ddgd|d }| |r*t|}| S )N
showconfigzpaths.defaultF)r   r   )r   strip_is_local_repositoryr
   )clsr!   r   r   r   r   get_remote_url[   s    
zMercurial.get_remote_urlc             C   s   | j ddgd|d }|S )zW
        Return the repository-local changeset revision number, as an integer.
        parentsz--template={rev}F)r   r   )r   rD   )rF   r!   current_revisionr   r   r   get_revisiond   s    zMercurial.get_revisionc             C   s   | j ddgd|d }|S )zh
        Return the changeset identification hash, as a 40-character
        hexadecimal string
        rH   z--template={node}F)r   r   )r   rD   )rF   r!   current_rev_hashr   r   r   get_requirement_revisionn   s    z"Mercurial.get_requirement_revisionc             C   s   dS )z&Always assume the versions don't matchFr   )rF   r(   namer   r   r   is_commit_id_equaly   s    zMercurial.is_commit_id_equalc             C   sB   | j dgd|d }tj|s8tjtj||}t||S )z~
        Return the path to setup.py, relative to the repo root.
        Return None if setup.py is in the repo root.
        rootF)r   r   )r   rD   r0   r   isabsabspathr1   r   )rF   r!   	repo_rootr   r   r   get_subdirectory~   s
    zMercurial.get_subdirectoryc                sx   t t| |}|r|S y| jdg|dddd}W n2 tk
rR   td| d S  tk
rd   d S X tj	
|dS )NrO   Fraise)r   r   on_returncodelog_failed_cmdzIcould not determine if %s is under hg control because hg is not availablez
)superr   get_repository_rootr   r   r%   debugr   r0   r   normpathrstrip)rF   r!   locr)	__class__r   r   rX      s"    zMercurial.get_repository_root)__name__
__module____qualname__rM   r2   	repo_nameschemesstaticmethodr   r   r+   rA   r#   classmethodrG   rJ   rL   rN   rS   rX   __classcell__r   r   )r^   r   r      s   
	
r   )
__future__r   loggingr0   pip._vendor.six.movesr   pip._internal.exceptionsr   r   pip._internal.utils.miscr   pip._internal.utils.subprocessr   pip._internal.utils.temp_dirr   pip._internal.utils.typingr	   pip._internal.utils.urlsr
    pip._internal.vcs.versioncontrolr   r   r   r   r   	getLoggerr_   r%   r   registerr   r   r   r   <module>   s"   
 