group_by.py - mozsearch

Enable keyboard shortcuts

# 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/.

from taskgraph.util.dependencies import group_by

@group_by("component")

def component_grouping(config, tasks):

    groups = {}

    for task in tasks:

        component = task.attributes.get("component")

        if component == "all":

            continue

        build_type = task.attributes.get("build-type")

        groups.setdefault((component, build_type), []).append(task)

    tasks_for_all_components = [

        task

        for task in tasks

        if task.attributes.get("component") == "all"

        # We just want to depend on the task that waits on all chunks. This way

        # we have a single dependency for that kind

        and task.attributes.get("is_final_chunked_task", True)

    for (_, build_type), tasks in groups.items():

        tasks.extend(

                task

                for task in tasks_for_all_components

                if task.attributes.get("build-type") == build_type

    return groups.values()

@group_by('build-type')

def build_type_grouping(config, tasks):

    groups = {}

    for task in tasks:

        # We just want to depend on the task that waits on all chunks. This way

        # we have a single dependency for that kind

        if not task.attributes.get("is_final_chunked_task", True):

            continue

        build_type = task.attributes.get('build-type')

        groups.setdefault(build_type, []).append(task)

    return groups.values()