gecko-dev/taskcluster/taskgraph/transforms/release_deps.py
Ricky Stewart 02a7b4ebdf Bug 1654103: Standardize on Black for Python code in mozilla-central.
Allow-list all Python code in tree for use with the black linter, and re-format all code in-tree accordingly.

To produce this patch I did all of the following:

1. Make changes to tools/lint/black.yml to remove include: stanza and update list of source extensions.

2. Run ./mach lint --linter black --fix

3. Make some ad-hoc manual updates to python/mozbuild/mozbuild/test/configure/test_configure.py -- it has some hard-coded line numbers that the reformat breaks.

4. Make some ad-hoc manual updates to `testing/marionette/client/setup.py`, `testing/marionette/harness/setup.py`, and `testing/firefox-ui/harness/setup.py`, which have hard-coded regexes that break after the reformat.

5. Add a set of exclusions to black.yml. These will be deleted in a follow-up bug (1672023).

# ignore-this-changeset

Differential Revision: https://phabricator.services.mozilla.com/D94045
2020-10-26 18:34:53 +00:00

63 lines
2.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/.
"""
Add dependencies to release tasks.
"""
from __future__ import absolute_import, print_function, unicode_literals
from taskgraph.transforms.base import TransformSequence
PHASES = ["build", "promote", "push", "ship"]
transforms = TransformSequence()
@transforms.add
def add_dependencies(config, jobs):
for job in jobs:
dependencies = {}
# Add any kind_dependencies_tasks with matching product as dependencies
product = job.get("shipping-product")
phase = job.get("shipping-phase")
if product is None:
continue
required_signoffs = set(
job.setdefault("attributes", {}).get("required_signoffs", [])
)
for dep_task in config.kind_dependencies_tasks.values():
# Weed out unwanted tasks.
# XXX we have run-on-projects which specifies the on-push behavior;
# we need another attribute that specifies release promotion,
# possibly which action(s) each task belongs in.
# We can only depend on tasks in the current or previous phases
dep_phase = dep_task.attributes.get("shipping_phase")
if dep_phase and PHASES.index(dep_phase) > PHASES.index(phase):
continue
if dep_task.attributes.get("build_platform") and job.get(
"attributes", {}
).get("build_platform"):
if (
dep_task.attributes["build_platform"]
!= job["attributes"]["build_platform"]
):
continue
# Add matching product tasks to deps
if (
dep_task.task.get("shipping-product") == product
or dep_task.attributes.get("shipping_product") == product
):
dependencies[dep_task.label] = dep_task.label
required_signoffs.update(
dep_task.attributes.get("required_signoffs", [])
)
job.setdefault("dependencies", {}).update(dependencies)
if required_signoffs:
job["attributes"]["required_signoffs"] = sorted(required_signoffs)
yield job