chromium/third_party/wpt_tools/wpt/tools/wptrunner/wptrunner/config.py

# mypy: allow-untyped-defs

from configparser import ConfigParser
import os
import sys
from collections import OrderedDict
from typing import Dict, Mapping, Optional

here = os.path.dirname(__file__)


class ConfigDict(Dict[str, str]):
    def __init__(self, base_path: str, *args: str, **kwargs: str):
        self.base_path = base_path
        dict.__init__(self, *args, **kwargs)

    def get_path(self, key: str, default:Optional[str] = None) -> Optional[str]:
        if key not in self:
            return default
        path = self[key]
        os.path.expanduser(path)
        return os.path.abspath(os.path.join(self.base_path, path))


def read(config_path: str) -> Mapping[str, ConfigDict]:
    config_path = os.path.abspath(config_path)
    config_root = os.path.dirname(config_path)
    parser = ConfigParser()
    success = parser.read(config_path)
    assert config_path in success, success

    subns = {"pwd": os.path.abspath(os.path.curdir)}

    rv = OrderedDict()
    for section in parser.sections():
        rv[section] = ConfigDict(config_root)
        for key in parser.options(section):
            rv[section][key] = parser.get(section, key, raw=False, vars=subns)

    return rv


def path(argv=None):
    if argv is None:
        argv = []
    path = None

    for i, arg in enumerate(argv):
        if arg == "--config":
            if i + 1 < len(argv):
                path = argv[i + 1]
        elif arg.startswith("--config="):
            path = arg.split("=", 1)[1]
        if path is not None:
            break

    if path is None:
        if os.path.exists("wptrunner.ini"):
            path = os.path.abspath("wptrunner.ini")
        else:
            path = os.path.join(here, "..", "wptrunner.default.ini")

    return os.path.abspath(path)


def load():
    return read(path(sys.argv))