Add a function to get tasks from multiple kinds in one pass

This is especially useful when some of the kinds the caller is
interested in is a dependency of another kind.

Author: glandium

src/taskgraph/generator.py

@@ -446,19 +446,28 @@ def verify(self, name, *args, **kwargs):
             return name, args[0]
 
 
-def load_tasks_for_kind(parameters, kind, root_dir=None):
+def load_tasks_for_kinds(parameters, kinds, root_dir=None):
     """
-    Get all the tasks of a given kind.
+    Get all the tasks of a the given kinds.
 
     This function is designed to be called from outside of taskgraph.
     """
     # make parameters read-write
     parameters = dict(parameters)
-    parameters["target-kinds"] = [kind]
+    parameters["target-kinds"] = kinds
     parameters = parameters_loader(spec=None, strict=False, overrides=parameters)
     tgg = TaskGraphGenerator(root_dir=root_dir, parameters=parameters)
     return {
         task.task["metadata"]["name"]: task
         for task in tgg.full_task_set
-        if task.kind == kind
+        if task.kind in kinds
     }
+
+
+def load_tasks_for_kind(parameters, kind, root_dir=None):
+    """
+    Get all the tasks of a given kind.
+
+    This function is designed to be called from outside of taskgraph.
+    """
+    return load_tasks_for_kinds(parameters, [kind], root_dir)