gecko-dev/taskcluster/taskgraph/transforms/beetmover_geckoview.py

158 строки
5.2 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 six import text_type
from taskgraph.loader.single_dep import schema
from taskgraph.transforms.base import TransformSequence
from taskgraph.transforms.beetmover import (
craft_release_properties as beetmover_craft_release_properties,
)
from taskgraph.util.attributes import copy_attributes_from_dependent_job
from taskgraph.util.declarative_artifacts import (
get_geckoview_template_vars,
get_geckoview_upstream_artifacts,
get_geckoview_artifact_id,
)
from taskgraph.util.schema import resolve_keyed_by, optionally_keyed_by
from taskgraph.util.scriptworker import generate_beetmover_artifact_map
from taskgraph.transforms.task import task_description_schema
from voluptuous import Required, Optional
beetmover_description_schema = schema.extend(
{
Optional("label"): text_type,
Optional("treeherder"): task_description_schema["treeherder"],
Required("run-on-projects"): task_description_schema["run-on-projects"],
Required("run-on-hg-branches"): task_description_schema["run-on-hg-branches"],
Optional("bucket-scope"): optionally_keyed_by("release-level", text_type),
Optional("shipping-phase"): optionally_keyed_by(
"project", task_description_schema["shipping-phase"]
),
Optional("shipping-product"): task_description_schema["shipping-product"],
Optional("attributes"): task_description_schema["attributes"],
}
)
transforms = TransformSequence()
transforms.add_validate(beetmover_description_schema)
@transforms.add
def resolve_keys(config, jobs):
for job in jobs:
resolve_keyed_by(
job,
"run-on-hg-branches",
item_name=job["label"],
project=config.params["project"],
)
resolve_keyed_by(
job,
"shipping-phase",
item_name=job["label"],
project=config.params["project"],
)
resolve_keyed_by(
job,
"bucket-scope",
item_name=job["label"],
**{"release-level": config.params.release_level()}
)
yield job
@transforms.add
def make_task_description(config, jobs):
for job in jobs:
dep_job = job["primary-dependency"]
attributes = copy_attributes_from_dependent_job(dep_job)
attributes.update(job.get("attributes", {}))
treeherder = job.get("treeherder", {})
treeherder.setdefault("symbol", "BM-gv")
dep_th_platform = (
dep_job.task.get("extra", {})
.get("treeherder", {})
.get("machine", {})
.get("platform", "")
)
treeherder.setdefault("platform", "{}/opt".format(dep_th_platform))
treeherder.setdefault("tier", 2)
treeherder.setdefault("kind", "build")
label = job["label"]
description = (
"Beetmover submission for geckoview"
"{build_platform}/{build_type}'".format(
build_platform=attributes.get("build_platform"),
build_type=attributes.get("build_type"),
)
)
dependencies = deepcopy(dep_job.dependencies)
dependencies[dep_job.kind] = dep_job.label
if job.get("locale"):
attributes["locale"] = job["locale"]
attributes["run_on_hg_branches"] = job["run-on-hg-branches"]
task = {
"label": label,
"description": description,
"worker-type": "beetmover",
"scopes": [
job["bucket-scope"],
"project:releng:beetmover:action:push-to-maven",
],
"dependencies": dependencies,
"attributes": attributes,
"run-on-projects": job["run-on-projects"],
"treeherder": treeherder,
"shipping-phase": job["shipping-phase"],
}
yield task
@transforms.add
def make_task_worker(config, jobs):
for job in jobs:
job["worker"] = {
"artifact-map": generate_beetmover_artifact_map(
config,
job,
**get_geckoview_template_vars(
config,
job["attributes"]["build_platform"],
job["attributes"].get("update-channel"),
)
),
"implementation": "beetmover-maven",
"release-properties": craft_release_properties(config, job),
"upstream-artifacts": get_geckoview_upstream_artifacts(config, job),
}
yield job
def craft_release_properties(config, job):
release_properties = beetmover_craft_release_properties(config, job)
release_properties["artifact-id"] = get_geckoview_artifact_id(
config,
job["attributes"]["build_platform"],
job["attributes"].get("update-channel"),
)
release_properties["app-name"] = "geckoview"
return release_properties