kopia lustrzana https://github.com/jupyterhub/repo2docker
75 wiersze
2.6 KiB
Python
75 wiersze
2.6 KiB
Python
import subprocess
|
|
|
|
from .base import ContentProvider, ContentProviderException
|
|
from ..utils import execute_cmd, R2dState
|
|
|
|
args_enabling_topic = ["--config", "extensions.topic="]
|
|
|
|
|
|
class Mercurial(ContentProvider):
|
|
"""Provide contents of a remote Mercurial repository."""
|
|
|
|
def detect(self, source, ref=None, extra_args=None):
|
|
if "github.com/" in source or source.endswith(".git"):
|
|
return None
|
|
try:
|
|
subprocess.check_output(
|
|
["hg", "identify", source, "--config", "extensions.hggit=!"]
|
|
+ args_enabling_topic,
|
|
stderr=subprocess.DEVNULL,
|
|
)
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
return {"repo": source, "ref": ref}
|
|
|
|
def fetch(self, spec, output_dir, yield_output=False):
|
|
repo = spec["repo"]
|
|
ref = spec.get("ref", None)
|
|
|
|
# make a clone of the remote repository
|
|
try:
|
|
cmd = [
|
|
"hg",
|
|
"clone",
|
|
repo,
|
|
output_dir,
|
|
"--config",
|
|
"phases.publish=False",
|
|
] + args_enabling_topic
|
|
if ref is not None:
|
|
# don't update so the clone will include an empty working
|
|
# directory, the given ref will be updated out later
|
|
cmd.extend(["--noupdate"])
|
|
yield from execute_cmd(cmd, capture=yield_output)
|
|
|
|
except subprocess.CalledProcessError as error:
|
|
msg = f"Failed to clone repository from {repo}"
|
|
if ref is not None:
|
|
msg += f" (ref {ref})"
|
|
msg += "."
|
|
raise ContentProviderException(msg) from error
|
|
|
|
# check out the specific ref given by the user
|
|
if ref is not None:
|
|
try:
|
|
yield from execute_cmd(
|
|
["hg", "update", "--clean", ref] + args_enabling_topic,
|
|
cwd=output_dir,
|
|
capture=yield_output,
|
|
)
|
|
except subprocess.CalledProcessError:
|
|
self.log.error(
|
|
f"Failed to update to ref {ref}", extra=dict(phase=R2dState.FAILED)
|
|
)
|
|
raise ValueError(f"Failed to update to ref {ref}")
|
|
|
|
cmd = ["hg", "identify", "-i"] + args_enabling_topic
|
|
sha1 = subprocess.Popen(cmd, stdout=subprocess.PIPE, cwd=output_dir)
|
|
self._node_id = sha1.stdout.read().decode().strip()
|
|
|
|
@property
|
|
def content_id(self):
|
|
"""A unique ID to represent the version of the content."""
|
|
return self._node_id
|