зеркало из https://github.com/mozilla/gecko-dev.git
109 строки
3.8 KiB
Python
109 строки
3.8 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 logging
|
|
import sys
|
|
import os
|
|
import tempfile
|
|
|
|
from mach.decorators import CommandArgument, CommandProvider, Command
|
|
from mozbuild.base import MachCommandBase, BinaryNotFoundException
|
|
|
|
requirements = os.path.join(os.path.dirname(__file__), "requirements", "base.txt")
|
|
|
|
|
|
@CommandProvider
|
|
class CondprofileCommandProvider(MachCommandBase):
|
|
def _init(self):
|
|
self.activate_virtualenv()
|
|
self.virtualenv_manager.install_pip_requirements(
|
|
requirements, require_hashes=False
|
|
)
|
|
|
|
@Command("fetch-condprofile", category="testing")
|
|
@CommandArgument("--target-dir", default=None, help="Target directory")
|
|
@CommandArgument("--platform", default=None, help="Platform")
|
|
@CommandArgument("--scenario", default="full", help="Scenario") # grab choices
|
|
@CommandArgument("--customization", default="default", help="Customization") # same
|
|
@CommandArgument("--task-id", default=None, help="Task ID")
|
|
@CommandArgument("--download-cache", action="store_true", default=True)
|
|
@CommandArgument(
|
|
"--repo",
|
|
default="mozilla-central",
|
|
choices=["mozilla-central", "try"],
|
|
help="Repository",
|
|
)
|
|
def fetch(
|
|
self,
|
|
target_dir,
|
|
platform,
|
|
scenario,
|
|
customization,
|
|
task_id,
|
|
download_cache,
|
|
repo,
|
|
):
|
|
self._init()
|
|
from condprof.client import get_profile
|
|
from condprof.util import get_current_platform
|
|
|
|
if platform is None:
|
|
platform = get_current_platform()
|
|
|
|
if target_dir is None:
|
|
target_dir = tempfile.mkdtemp()
|
|
|
|
get_profile(
|
|
target_dir, platform, scenario, customization, task_id, download_cache, repo
|
|
)
|
|
|
|
@Command("run-condprofile", category="testing")
|
|
@CommandArgument("archive", help="Archives Dir", type=str, default=None)
|
|
@CommandArgument("--firefox", help="Firefox Binary", type=str, default=None)
|
|
@CommandArgument("--scenario", help="Scenario to use", type=str, default="all")
|
|
@CommandArgument("--profile", help="Existing profile Dir", type=str, default=None)
|
|
@CommandArgument(
|
|
"--customization", help="Profile customization to use", type=str, default="all"
|
|
)
|
|
@CommandArgument(
|
|
"--visible", help="Don't use headless mode", action="store_true", default=False
|
|
)
|
|
@CommandArgument(
|
|
"--archives-dir", help="Archives local dir", type=str, default="/tmp/archives"
|
|
)
|
|
@CommandArgument(
|
|
"--force-new", help="Create from scratch", action="store_true", default=False
|
|
)
|
|
@CommandArgument(
|
|
"--strict",
|
|
help="Errors out immediatly on a scenario failure",
|
|
action="store_true",
|
|
default=True,
|
|
)
|
|
@CommandArgument(
|
|
"--geckodriver",
|
|
help="Path to the geckodriver binary",
|
|
type=str,
|
|
default=sys.platform.startswith("win") and "geckodriver.exe" or "geckodriver",
|
|
)
|
|
@CommandArgument("--device-name", help="Name of the device", type=str, default=None)
|
|
def run(self, **kw):
|
|
os.environ["MANUAL_MACH_RUN"] = "1"
|
|
self._init()
|
|
|
|
if kw["firefox"] is None:
|
|
try:
|
|
kw["firefox"] = self.get_binary_path()
|
|
except BinaryNotFoundException as e:
|
|
self.log(logging.ERROR, 'run-condprofile',
|
|
{'error': str(e)},
|
|
'ERROR: {error}')
|
|
self.log(logging.INFO, 'run-condprofile',
|
|
{'help': e.help()},
|
|
'{help}')
|
|
return 1
|
|
|
|
from condprof.runner import run
|
|
|
|
run(**kw)
|