1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
# 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 attr
from six import text_type
from ..config import GraphConfig
from ..parameters import Parameters
from ..util.schema import Schema, validate_schema
@attr.s(frozen=True)
class TransformConfig(object):
"""
A container for configuration affecting transforms. The `config` argument
to transforms is an instance of this class.
"""
# the name of the current kind
kind = attr.ib()
# the path to the kind configuration directory
path = attr.ib(type=text_type)
# the parsed contents of kind.yml
config = attr.ib(type=dict)
# the parameters for this task-graph generation run
params = attr.ib(type=Parameters)
# a list of all the tasks associated with the kind dependencies of the
# current kind
kind_dependencies_tasks = attr.ib()
# Global configuration of the taskgraph
graph_config = attr.ib(type=GraphConfig)
# whether to write out artifacts for the decision task
write_artifacts = attr.ib(type=bool)
@attr.s()
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.
"""
_transforms = attr.ib(factory=list)
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 add(self, func):
self._transforms.append(func)
return func
def add_validate(self, schema):
self.add(ValidateSchema(schema))
@attr.s
class ValidateSchema(object):
schema = attr.ib(type=Schema)
def __call__(self, config, tasks):
for task in tasks:
if "name" in task:
error = "In {kind} kind task {name!r}:".format(
kind=config.kind, name=task["name"]
)
elif "label" in task:
error = "In job {label!r}:".format(label=task["label"])
elif "primary-dependency" in task:
error = "In {kind} kind task for {dependency!r}:".format(
kind=config.kind, dependency=task["primary-dependency"].label
)
else:
error = "In unknown task:"
validate_schema(self.schema, task, error)
yield task
|
