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
|
# 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/.
import logging
from .transform import loader as transform_loader
logger = logging.getLogger(__name__)
DEFAULT_TRANSFORMS = [
"taskgraph.transforms.run:transforms",
"taskgraph.transforms.task:transforms",
]
def loader(kind, path, config, params, loaded_tasks):
"""
This default loader builds on the `transform` loader by providing sensible
default transforms that the majority of simple tasks will need.
Specifically, `run` and `task` transforms will be appended to the end of the
list of transforms in the kind being loaded.
"""
transform_refs = config.setdefault("transforms", [])
for t in DEFAULT_TRANSFORMS:
if t in config.get("transforms", ()):
raise KeyError(
f"Transform {t} is already present in the loader's default transforms; it must not be defined in the kind"
)
transform_refs.extend(DEFAULT_TRANSFORMS)
return transform_loader(kind, path, config, params, loaded_tasks)
|
