taskcluster/taskgraph/parameters.py
author Justin Wood <Callek@gmail.com>
Mon, 20 Jun 2016 21:06:55 -0400
changeset 327283 a1748bc8704ad72414ab3fbb7d93e7d0cf00c042
parent 322454 3fb747099412c45b769155857ba937467aa608de
child 335706 7a3a65ff58653cfa96f39c2ab4e458111602b92c
permissions -rw-r--r--
Bug 1280956 - Use in-tree linter job to flake8 test taskcluster directory. r=dustin MozReview-Commit-ID: FsWmAnnycZ2

# -*- coding: utf-8 -*-

# 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/.

from __future__ import absolute_import, print_function, unicode_literals

import json
import yaml
from mozbuild.util import ReadOnlyDict


class Parameters(ReadOnlyDict):
    """An immutable dictionary with nicer KeyError messages on failure"""
    def __getitem__(self, k):
        try:
            return super(Parameters, self).__getitem__(k)
        except KeyError:
            raise KeyError("taskgraph parameter {!r} not found".format(k))


def load_parameters_file(options):
    """
    Load parameters from the --parameters option
    """
    filename = options['parameters']
    if not filename:
        return Parameters()
    with open(filename) as f:
        if filename.endswith('.yml'):
            return Parameters(**yaml.safe_load(f))
        elif filename.endswith('.json'):
            return Parameters(**json.load(f))
        else:
            raise TypeError("Parameters file `{}` is not JSON or YAML".format(filename))