Source code
Revision control
Copy as Markdown
Other Tools
# 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
"""
Transform the beetmover task into an actual task description.
"""
from taskgraph.transforms.base import TransformSequence
from taskgraph.util.dependencies import get_primary_dependency
from taskgraph.util.keyed_by import evaluate_keyed_by
from taskgraph.util.schema import Schema, optionally_keyed_by
from voluptuous import Optional, Required
from gecko_taskgraph.transforms.beetmover import craft_release_properties
from gecko_taskgraph.transforms.task import task_description_schema
from gecko_taskgraph.util.attributes import copy_attributes_from_dependent_job
from gecko_taskgraph.util.scriptworker import (
generate_beetmover_artifact_map,
generate_beetmover_upstream_artifacts,
)
beetmover_description_schema = Schema(
{
Required("label"): str,
Required("description"): str,
Required("dependencies"): task_description_schema["dependencies"],
Required("if-dependencies"): task_description_schema["if-dependencies"],
Optional("treeherder"): task_description_schema["treeherder"],
Required("run-on-projects"): task_description_schema["run-on-projects"],
Optional("attributes"): task_description_schema["attributes"],
Optional("task-from"): task_description_schema["task-from"],
Required("worker-type"): task_description_schema["worker-type"],
Required("scopes"): optionally_keyed_by("project", [str]),
}
)
transforms = TransformSequence()
@transforms.add
def remove_name(config, tasks):
for job in tasks:
if "name" in job:
del job["name"]
yield job
transforms.add_validate(beetmover_description_schema)
@transforms.add
def make_task_description(config, tasks):
for task in tasks:
dep_task = get_primary_dependency(config, task)
assert dep_task
attributes = copy_attributes_from_dependent_job(dep_task)
attributes.update(task.get("attributes", {}))
treeherder = task.get("treeherder", {})
dep_th_platform = (
dep_task.task.get("extra", {})
.get("treeherder", {})
.get("machine", {})
.get("platform", "")
)
treeherder.setdefault("platform", f"{dep_th_platform}/opt")
task["description"] = task["description"].format(
build_platform=attributes.get("build_platform"),
build_type=attributes.get("build_type"),
)
task["scopes"] = evaluate_keyed_by(
task["scopes"],
"beetmover-integration",
{"project": config.params.get("project")},
)
if task.get("locale"):
attributes["locale"] = task["locale"]
task["attributes"] = attributes
yield task
@transforms.add
def make_task_worker(config, jobs):
for job in jobs:
locale = job["attributes"].get("locale")
platform = job["attributes"]["build_platform"]
job["worker"] = {
"release-properties": craft_release_properties(config, job),
"upstream-artifacts": generate_beetmover_upstream_artifacts(
config, job, platform, locale
),
"artifact-map": generate_beetmover_artifact_map(
config, job, platform=platform, locale=locale
),
}
yield job