forked from mirrors/gecko-dev
d7e248fb83
MozReview-Commit-ID: Iaet3uja3vG --HG-- extra : rebase_source : a2a7a5b449361c9ec1e89d34feda11a9f64f5d3b
74 lines
2.7 KiB
Python
74 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
|
|
|
|
from taskgraph.transforms.base import TransformSequence
|
|
from taskgraph.transforms.task import task_description_schema
|
|
from taskgraph.util.schema import resolve_keyed_by, Schema, validate_schema
|
|
from taskgraph.util.scriptworker import get_release_config
|
|
|
|
from voluptuous import Required
|
|
|
|
|
|
transforms = TransformSequence()
|
|
|
|
# Voluptuous uses marker objects as dictionary *keys*, but they are not
|
|
# comparable, so we cast all of the keys back to regular strings
|
|
task_description_schema = {str(k): v for k, v in task_description_schema.schema.iteritems()}
|
|
|
|
google_play_description_schema = Schema({
|
|
Required('name'): basestring,
|
|
Required('label'): task_description_schema['label'],
|
|
Required('description'): task_description_schema['description'],
|
|
Required('job-from'): task_description_schema['job-from'],
|
|
Required('attributes'): task_description_schema['attributes'],
|
|
Required('treeherder'): task_description_schema['treeherder'],
|
|
Required('run-on-projects'): task_description_schema['run-on-projects'],
|
|
Required('shipping-phase'): task_description_schema['shipping-phase'],
|
|
Required('shipping-product'): task_description_schema['shipping-product'],
|
|
Required('worker-type'): task_description_schema['worker-type'],
|
|
Required('worker'): object,
|
|
})
|
|
|
|
|
|
@transforms.add
|
|
def validate_jobs_schema(config, jobs):
|
|
for job in jobs:
|
|
job['label'] = job['name']
|
|
validate_schema(
|
|
google_play_description_schema, job,
|
|
"In GooglePlayStrings ({!r} kind) task for {!r}:".format(config.kind, job['label'])
|
|
)
|
|
yield job
|
|
|
|
|
|
@transforms.add
|
|
def set_worker_data(config, jobs):
|
|
for job in jobs:
|
|
worker = job['worker']
|
|
|
|
env = worker.setdefault('env', {})
|
|
resolve_keyed_by(
|
|
env, 'PACKAGE_NAME', item_name=job['name'],
|
|
project=config.params['project']
|
|
)
|
|
|
|
resolve_keyed_by(
|
|
env, 'MAJOR_VERSION_NUMBER', item_name=job['name'],
|
|
project=config.params['project']
|
|
)
|
|
if env.get('MAJOR_VERSION_NUMBER'):
|
|
release_config = get_release_config(config)
|
|
major_version = release_config['version'].split('.')[0]
|
|
env['MAJOR_VERSION_NUMBER'] = env['MAJOR_VERSION_NUMBER'].format(
|
|
major_version=major_version
|
|
)
|
|
else:
|
|
del env['MAJOR_VERSION_NUMBER']
|
|
|
|
yield job
|