U
��.e� � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lmZmZmZ er�d d l mZ d d lmZ e�e�ZG d
d� de�Z e� e � dS )� )�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 Zd ZdZdZdZdZedd� �Zdd� Z d d
� Z
d d � Z d
d� Z e
dd� �Ze
dd� �Ze
dd� �Ze
dd� �Ze
dd� �Ze
� fdd��Z� ZS )� Mercurial�hgz.hg�clone)r zhg+filezhg+httpzhg+httpszhg+sshzhg+static-httpc C s | gS )N� )Zrevr r �?/usr/lib/python3.8/site-packages/pip/_internal/vcs/mercurial.py�get_base_rev_args' s zMercurial.get_base_rev_argsc C s@ t dd��,}| j|j|d� | jd|gd|jd� W 5 Q R X dS )z?Export the Hg repository at the url to the destination location�export)Zkind)�url�archiveF�� show_stdout�cwdN)r �unpack�path� run_command)�self�locationr Ztemp_dirr r r r + s �zMercurial.exportc C sP |� � }t�d||t|�� | �tddd||�� | jtdd|�� �|d� d S )NzCloning hg %s%s to %sr z
--noupdate�-q�update�r )Z
to_display�logger�infor r r �to_args)r �destr � rev_optionsZ rev_displayr r r � fetch_new5 s ��zMercurial.fetch_newc
C s� t j�|| jd�}t�� }z>|�|� |�dd|j� t |d��}|�
|� W 5 Q R X W n6 t tj fk
r� } zt
�d||� W 5 d }~X Y n X tdd|�� �}| j||d� d S ) NZhgrc�paths�default�wz/Could not switch Mercurial repository to %s: %sr" r! r# )�osr �join�dirnamer ZRawConfigParser�read�setZsecret�open�write�OSErrorZNoSectionErrorr$ Zwarningr r& r ) r r' r r( Z repo_configZconfigZ config_file�exc�cmd_argsr r r �switchD s
�zMercurial.switchc C s4 | j ddg|d� tdd|�� �}| j ||d� d S )NZpullr! r# r" )r r r& )r r' r r( r6 r r r r" U s zMercurial.updatec C s2 | j ddgd|d��� }| �|�r*t|�}|�� S )NZ
showconfigz
paths.defaultFr )r �stripZ_is_local_repositoryr
)�clsr r r r r �get_remote_url[ s �
zMercurial.get_remote_urlc C s | j ddgd|d��� }|S )zW
Return the repository-local changeset revision number, as an integer.
�parentsz--template={rev}Fr �r r8 )r9 r Zcurrent_revisionr r r � get_revisiond s �
zMercurial.get_revisionc C s | j ddgd|d��� }|S )zh
Return the changeset identification hash, as a 40-character
hexadecimal string
r; z--template={node}Fr r< )r9 r Zcurrent_rev_hashr r r �get_requirement_revisionn s �
z"Mercurial.get_requirement_revisionc C s dS )z&Always assume the versions don't matchFr )r9 r' �namer r r �is_commit_id_equaly s z Mercurial.is_commit_id_equalc C sB | j dgd|d��� }tj�|�s8tj�tj�||��}t||�S )z~
Return the path to setup.py, relative to the repo root.
Return None if setup.py is in the repo root.
�rootFr )r r8 r- r �isabs�abspathr. r )r9 r Z repo_rootr r r �get_subdirectory~ s �
zMercurial.get_subdirectoryc sP t t| ��|�rdS z | jdg|dddd� W dS ttfk
rJ Y dS X d S )NTZidentifyF�raise)r r Z
on_returncodeZlog_failed_cmd)�superr �controls_locationr r r )r9 r �� __class__r r rG � s �zMercurial.controls_location)�__name__�
__module__� __qualname__r? r/ Z repo_nameZschemes� staticmethodr r r) r7 r" � classmethodr: r= r> r@ rD rG �
__classcell__r r rH r r s,
r ) Z
__future__r Zloggingr- Zpip._vendor.six.movesr Zpip._internal.exceptionsr r Zpip._internal.utils.miscr Z pip._internal.utils.subprocessr Z pip._internal.utils.temp_dirr Zpip._internal.utils.typingr Zpip._internal.utils.urlsr
Z pip._internal.vcs.versioncontrolr r r
r r Z getLoggerrJ r$ r �registerr r r r � s
|