fune/taskcluster/taskgraph/transforms/cross_channel.py
Andrew Halberstadt d024a31639 Bug 1726573 - Run 'pyupgrade --py36-plus' on taskcluster/taskgraph directory, r=taskgraph-reviewers,bhearsum
This patch was generated via:

  $ pyupgrade --py36-plus $(find taskcluster/taskgraph -name "*.py" -type f)
  $ autoflake --remove-all-unused-imports -i $(find taskcluster/taskgraph -name "*.py" -type f)

The same set of commands are being applied to standalone taskgraph as well so
they remain in sync.

Differential Revision: https://phabricator.services.mozilla.com/D123234
2021-08-24 19:35:46 +00:00

44 lines
1.3 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/.
"""
Build a command to run `mach l10n-cross-channel`.
"""
from pipes import quote as shell_quote
from taskgraph.transforms.base import TransformSequence
from taskgraph.util.schema import resolve_keyed_by
transforms = TransformSequence()
@transforms.add
def resolve_keys(config, jobs):
for job in jobs:
for item in ["ssh-key-secret", "run.actions"]:
resolve_keyed_by(job, item, item, **{"level": str(config.params["level"])})
yield job
@transforms.add
def build_command(config, jobs):
for job in jobs:
command = [
"l10n-cross-channel",
"-o",
"/builds/worker/artifacts/outgoing.diff",
"--attempts",
"5",
]
ssh_key_secret = job.pop("ssh-key-secret")
if ssh_key_secret:
command.extend(["--ssh-secret", ssh_key_secret])
job.setdefault("scopes", []).append(f"secrets:get:{ssh_key_secret}")
command.extend(job["run"].pop("actions", []))
job.setdefault("run", {}).update(
{"using": "mach", "mach": " ".join(map(shell_quote, command))}
)
yield job