mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-22 01:35:35 +00:00
73e5e1e2b3
MozReview-Commit-ID: CuEQyn03Anh --HG-- extra : rebase_source : 2fa6e873643723a869154c33a0f466a36cdb5172
64 lines
2.2 KiB
Python
64 lines
2.2 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/.
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
|
|
class TransformConfig(object):
|
|
"""A container for configuration affecting transforms. The `config`
|
|
argument to transforms is an instance of this class, possibly with
|
|
additional kind-specific attributes beyond those set here."""
|
|
def __init__(self, kind, path, config, params,
|
|
kind_dependencies_tasks=None, graph_config=None):
|
|
# the name of the current kind
|
|
self.kind = kind
|
|
|
|
# the path to the kind configuration directory
|
|
self.path = path
|
|
|
|
# the parsed contents of kind.yml
|
|
self.config = config
|
|
|
|
# the parameters for this task-graph generation run
|
|
self.params = params
|
|
|
|
# a list of all the tasks associated with the kind dependencies of the
|
|
# current kind
|
|
self.kind_dependencies_tasks = kind_dependencies_tasks
|
|
|
|
# Global configuration of the taskgraph
|
|
self.graph_config = graph_config or {}
|
|
|
|
|
|
class TransformSequence(object):
|
|
"""
|
|
Container for a sequence of transforms. Each transform is represented as a
|
|
callable taking (config, items) and returning a generator which will yield
|
|
transformed items. The resulting sequence has the same interface.
|
|
|
|
This is convenient to use in a file full of transforms, as it provides a
|
|
decorator, @transforms.add, that will add the decorated function to the
|
|
sequence.
|
|
"""
|
|
|
|
def __init__(self, transforms=None):
|
|
self.transforms = transforms or []
|
|
|
|
def __call__(self, config, items):
|
|
for xform in self.transforms:
|
|
items = xform(config, items)
|
|
if items is None:
|
|
raise Exception("Transform {} is not a generator".format(xform))
|
|
return items
|
|
|
|
def __repr__(self):
|
|
return '\n'.join(
|
|
['TransformSequence(['] +
|
|
[repr(x) for x in self.transforms] +
|
|
['])'])
|
|
|
|
def add(self, func):
|
|
self.transforms.append(func)
|
|
return func
|