forked from mirrors/gecko-dev
67 lines
1.9 KiB
Python
67 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
|