spec.py - mozsearch

Enable keyboard shortcuts

#!/usr/bin/env python3

import argparse

import os

from typing import Any, Optional, Text

from . import vcs

from .manifest import compute_manifest_spec_items, InvalidCacheError, Manifest, write

from .log import get_logger, enable_debug_logging

here = os.path.dirname(__file__)

wpt_root = os.path.abspath(os.path.join(here, os.pardir, os.pardir))

logger = get_logger()

def update_spec(tests_root: Text,

                manifest_path: Text,

                url_base: Text,

                cache_root: Optional[Text] = None,

                working_copy: bool = True,

                parallel: bool = True

                ) -> None:

    manifest = Manifest(tests_root, url_base)

    logger.info("Updating SPEC_MANIFEST")

    try:

        tree = vcs.get_tree(tests_root, manifest, manifest_path, cache_root,

                            working_copy, True)

        changed = manifest.update(tree, parallel, compute_manifest_spec_items)

    except InvalidCacheError:

        logger.error("Manifest cache in spec.py was invalid.")

        return

    if changed:

        write(manifest, manifest_path)

    tree.dump_caches()

def update_from_cli(**kwargs: Any) -> None:

    tests_root = kwargs["tests_root"]

    path = kwargs["path"]

    assert tests_root is not None

    update_spec(tests_root,

                path,

                kwargs["url_base"],

                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(

        "--url-base", action="store", default="/",

        help="Base url to use as the mount point for tests in this 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"], "SPEC_MANIFEST.json")

    if kwargs["verbose"]:

        enable_debug_logging()

    update_from_cli(**kwargs)