ada64d6ce5
This creates a new "job-from" field that contains the relative filename the job was defined in. The filename is relative to 'config.path'. If the task came from the 'jobs' key defined in kind.yml, this field will be set to 'kind.yml'. MozReview-Commit-ID: 9e1tEb6XuZT
77 lines
2.7 KiB
Python
77 lines
2.7 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 push-apk kind into an actual task description.
|
|
"""
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import functools
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
from taskgraph.util.schema import Schema
|
|
from taskgraph.util.scriptworker import get_push_apk_scope, get_push_apk_track, \
|
|
get_push_apk_dry_run_option, get_push_apk_rollout_percentage
|
|
from taskgraph.util.push_apk import fill_labels_tranform, validate_jobs_schema_transform_partial, \
|
|
validate_dependent_tasks_transform, delete_non_required_fields_transform, generate_dependencies
|
|
|
|
from voluptuous import Required
|
|
|
|
|
|
transforms = TransformSequence()
|
|
|
|
push_apk_description_schema = Schema({
|
|
# the dependent task (object) for this beetmover job, used to inform beetmover.
|
|
Required('dependent-tasks'): object,
|
|
Required('name'): basestring,
|
|
Required('label'): basestring,
|
|
Required('description'): basestring,
|
|
Required('job-from'): basestring,
|
|
Required('attributes'): object,
|
|
Required('treeherder'): object,
|
|
Required('run-on-projects'): list,
|
|
Required('worker-type'): basestring,
|
|
Required('worker'): object,
|
|
Required('scopes'): None,
|
|
Required('deadline-after'): basestring,
|
|
})
|
|
|
|
validate_jobs_schema_transform = functools.partial(
|
|
validate_jobs_schema_transform_partial,
|
|
push_apk_description_schema,
|
|
'PushApk'
|
|
)
|
|
|
|
transforms.add(fill_labels_tranform)
|
|
transforms.add(validate_jobs_schema_transform)
|
|
transforms.add(validate_dependent_tasks_transform)
|
|
|
|
|
|
@transforms.add
|
|
def make_task_description(config, jobs):
|
|
for job in jobs:
|
|
job['dependencies'] = generate_dependencies(job['dependent-tasks'])
|
|
job['worker']['upstream-artifacts'] = generate_upstream_artifacts(job['dependencies'])
|
|
job['worker']['google-play-track'] = get_push_apk_track(config)
|
|
job['worker']['dry-run'] = get_push_apk_dry_run_option(config)
|
|
|
|
rollout_percentage = get_push_apk_rollout_percentage(config)
|
|
if rollout_percentage is not None:
|
|
job['worker']['rollout-percentage'] = rollout_percentage
|
|
|
|
job['scopes'] = [get_push_apk_scope(config)]
|
|
|
|
yield job
|
|
|
|
|
|
transforms.add(delete_non_required_fields_transform)
|
|
|
|
|
|
def generate_upstream_artifacts(dependencies):
|
|
return [{
|
|
'taskId': {'task-reference': '<{}>'.format(task_kind)},
|
|
'taskType': 'signing',
|
|
'paths': ['public/build/target.apk'],
|
|
} for task_kind in dependencies.keys() if 'breakpoint' not in task_kind]
|