#!/usr/bin/env python3
import argparse
import os
from typing import Any, Optional, TYPE_CHECKING
from . import manifest
from . import vcs
from .log import get_logger, enable_debug_logging
from .download import download_from_github
if TYPE_CHECKING:
from .manifest import Manifest # avoid cyclic import
here = os.path.dirname(__file__)
wpt_root = os.path.abspath(os.path.join(here, os.pardir, os.pardir))
logger = get_logger()
def update(tests_root: str,
manifest: "Manifest",
manifest_path: Optional[str] = None,
working_copy: bool = True,
cache_root: Optional[str] = None,
rebuild: bool = False,
parallel: bool = True
) -> bool:
logger.warning("Deprecated; use manifest.load_and_update instead")
logger.info("Updating manifest")
tree = vcs.get_tree(tests_root, manifest, manifest_path, cache_root,
working_copy, rebuild)
return manifest.update(tree, parallel)
def update_from_cli(**kwargs: Any) -> None:
tests_root = kwargs["tests_root"]
path = kwargs["path"]
assert tests_root is not None
if not kwargs["rebuild"] and kwargs["download"]:
download_from_github(path, tests_root)
manifest.load_and_update(tests_root,
path,
kwargs["url_base"],
update=True,
rebuild=kwargs["rebuild"],
cache_root=kwargs["cache_root"],
parallel=kwargs["parallel"])
def abs_path(path: str) -> str:
return os.path.abspath(os.path.expanduser(path))
def create_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser()
parser.add_argument(
"-v", "--verbose", dest="verbose", action="store_true", default=False,
help="Turn on verbose logging")
parser.add_argument(
"-p", "--path", type=abs_path, help="Path to manifest file.")
parser.add_argument(
"--tests-root", type=abs_path, default=wpt_root, help="Path to root of tests.")
parser.add_argument(
"-r", "--rebuild", action="store_true", default=False,
help="Force a full rebuild of the manifest.")
parser.add_argument(
"--url-base", action="store", default="/",
help="Base url to use as the mount point for tests in this manifest.")
parser.add_argument(
"--no-download", dest="download", action="store_false", default=True,
help="Never attempt to download the manifest.")
parser.add_argument(
"--cache-root", action="store", default=os.path.join(wpt_root, ".wptcache"),
help="Path in which to store any caches (default <tests_root>/.wptcache/)")
parser.add_argument(
"--no-parallel", dest="parallel", action="store_false", default=True,
help="Do not parallelize building the manifest")
return parser
def run(*args: Any, **kwargs: Any) -> None:
if kwargs["path"] is None:
kwargs["path"] = os.path.join(kwargs["tests_root"], "MANIFEST.json")
if kwargs["verbose"]:
enable_debug_logging()
update_from_cli(**kwargs)
def main() -> None:
opts = create_parser().parse_args()
run(**vars(opts))