Revision control
Copy as Markdown
# 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
from taskgraph.transforms.base import TransformSequence
from taskgraph.transforms.task import payload_builders
from taskgraph.util.schema import Schema
from voluptuous import Extra, Optional, Required
transforms = TransformSequence()
# Defined in `mozilla_taskgraph.worker_types`.
worker_schema = payload_builders["scriptworker-bitrise"].schema["bitrise"]
bitrise_schema = Schema({
Required("bitrise"): {
Required("workflows"): worker_schema["workflows"],
Optional("artifact_prefix"): worker_schema["artifact_prefix"],
},
Extra: object,
})
transforms.add_validate(bitrise_schema)
@transforms.add
def set_bitrise_app(config, tasks):
for task in tasks:
task["bitrise"]["app"] = config.params["project"]
yield task
@transforms.add
def add_worker(config, tasks):
for task in tasks:
worker = task.setdefault("worker", {})
worker["bitrise"] = task.pop("bitrise")
yield task