testing/web-platform/vcs.py
author Jared Wein <jwein@mozilla.com>
Tue, 14 May 2019 20:05:47 +0000
changeset 532675 3cec13612ee64b570b834e0ec7b68806cc759d77
parent 477579 424703a5d9e1796527cefff1992fed1b321de19b
permissions -rw-r--r--
Bug 1549809 - Stub out LoginFilter custom element. r=MattN,Pike Differential Revision: https://phabricator.services.mozilla.com/D30753

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
        # 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
        # TODO: Test on windows
        return True