summaryrefslogtreecommitdiffstats
path: root/taskcluster/taskgraph/transforms/try_job.py
blob: c840c6f8e6666aafcf26c4194b2f6a4dae98bd7f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# 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

from taskgraph.transforms.base import TransformSequence

transforms = TransformSequence()


@transforms.add
def set_job_try_name(config, jobs):
    """
    For a task which is governed by `-j` in try syntax, set the `job_try_name`
    attribute based on the job name.
    """
    for job in jobs:
        job.setdefault("attributes", {}).setdefault("job_try_name", job["name"])
        yield job