gecko-dev/taskcluster/taskgraph/transforms/upstream_artifact_task.py
Aki Sasaki 19175da396 Bug 1562412 - [3-task notarization] fix ./mach try release r=Callek
- backs out 71e376fd296d
- fixes bug 1618162.

Differential Revision: https://phabricator.services.mozilla.com/D64459

--HG--
extra : moz-landing-system : lando
2020-02-26 21:51:11 +00:00

29 lines
1 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/.
"""
Find upstream artifact task.
"""
from __future__ import absolute_import, print_function, unicode_literals
from taskgraph.transforms.base import TransformSequence
transforms = TransformSequence()
@transforms.add
def find_upstream_artifact_task(config, jobs):
for job in jobs:
dep_job = None
if job.get('dependent-tasks'):
dep_labels = [l for l in job['dependent-tasks'].keys()]
for label in dep_labels:
if 'notarization-part-1' in label:
assert dep_job is None, "Can't determine whether " \
"{} or {} is dep_job!".format(dep_job.label, label)
dep_job = job['dependent-tasks'][label]
if dep_job is not None:
job['upstream-artifact-task'] = dep_job
yield job