forked from mirrors/gecko-dev
7b13402d7e
Differential Revision: https://phabricator.services.mozilla.com/D1286 --HG-- extra : rebase_source : 3cb1080e5a94e3e9e5b5a3f6909b0e8d636de09c extra : source : 4a182c3447fa01c75fa8a231e067bebd83683da9
61 lines
2.3 KiB
Python
61 lines
2.3 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/.
|
|
"""
|
|
Transform the beetmover task into an actual task description.
|
|
"""
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
from copy import deepcopy
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
from taskgraph.util.schema import resolve_keyed_by
|
|
from taskgraph.util.taskcluster import get_taskcluster_artifact_prefix
|
|
from taskgraph.util.treeherder import add_suffix
|
|
|
|
transforms = TransformSequence()
|
|
|
|
|
|
@transforms.add
|
|
def add_command(config, tasks):
|
|
for task in tasks:
|
|
total_chunks = task["extra"]["chunks"]
|
|
|
|
for this_chunk in range(1, total_chunks+1):
|
|
chunked = deepcopy(task)
|
|
chunked["treeherder"]["symbol"] = add_suffix(
|
|
chunked["treeherder"]["symbol"], this_chunk)
|
|
chunked["label"] = "release-update-verify-{}-{}/{}".format(
|
|
chunked["name"], this_chunk, total_chunks
|
|
)
|
|
if not chunked["worker"].get("env"):
|
|
chunked["worker"]["env"] = {}
|
|
chunked["worker"]["command"] = [
|
|
"/bin/bash",
|
|
"-c",
|
|
"hg clone $BUILD_TOOLS_REPO tools && " +
|
|
"tools/scripts/release/updates/chunked-verify.sh " +
|
|
"UNUSED UNUSED {} {}".format(
|
|
total_chunks,
|
|
this_chunk,
|
|
)
|
|
]
|
|
for thing in ("CHANNEL", "VERIFY_CONFIG", "BUILD_TOOLS_REPO"):
|
|
thing = "worker.env.{}".format(thing)
|
|
resolve_keyed_by(chunked, thing, thing, **config.params)
|
|
|
|
update_verify_config = None
|
|
for upstream in chunked.get("dependencies", {}).keys():
|
|
if 'update-verify-config' in upstream:
|
|
update_verify_config = "{}update-verify.cfg".format(
|
|
get_taskcluster_artifact_prefix(task, "<{}>".format(upstream))
|
|
)
|
|
if not update_verify_config:
|
|
raise Exception("Couldn't find upate verify config")
|
|
|
|
chunked["worker"]["env"]["TASKCLUSTER_VERIFY_CONFIG"] = {
|
|
"task-reference": update_verify_config
|
|
}
|
|
|
|
yield chunked
|