gecko-dev/testing/web-platform/vcs.py

68 lines
1.9 KiB
Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import os
import subprocess
class Mercurial(object):
def __init__(self, repo_root):
self.root = os.path.abspath(repo_root)
self.hg = Mercurial.get_func(repo_root)
@staticmethod
def get_func(repo_root):
def hg(cmd, *args):
full_cmd = ["hg", cmd] + list(args)
return subprocess.check_output(full_cmd, cwd=repo_root)
# TODO: Test on Windows.
return hg
@staticmethod
def is_hg_repo(repo_root):
try:
with open(os.devnull, "w") as devnull:
subprocess.check_call(
["hg", "root"], cwd=repo_root, stdout=devnull, stderr=devnull
)
except subprocess.CalledProcessError:
return False
except OSError:
return False
# TODO: Test on windows
return True
class Git(object):
def __init__(self, repo_root, url_base):
self.root = os.path.abspath(repo_root)
self.git = Git.get_func(repo_root)
@staticmethod
def get_func(repo_root):
def git(cmd, *args):
full_cmd = ["git", cmd] + list(args)
return subprocess.check_output(full_cmd, cwd=repo_root)
# TODO: Test on Windows.
return git
@staticmethod
def is_git_repo(repo_root):
try:
with open(os.devnull, "w") as devnull:
subprocess.check_call(
["git", "rev-parse", "--show-cdup"],
cwd=repo_root,
stdout=devnull,
stderr=devnull,
)
except subprocess.CalledProcessError:
return False
except OSError:
return False
# TODO: Test on windows
return True