Bug 1899553: Add google play action to allow manual triggering of goo…

…gle play upload r=releng-reviewers,taskgraph-reviewers,jcristau

Differential Revision: https://phabricator.services.mozilla.com/D212033

taskcluster/gecko_taskgraph/actions/googleplay.py

@@ -0,0 +1,45 @@
+# 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/.
+
+
+import logging
+
+from .registry import register_callback_action
+from .util import create_tasks, fetch_graph_and_labels
+
+logger = logging.getLogger(__name__)
+
+
+@register_callback_action(
+    name="googleplay",
+    title="Submit android apps to google play",
+    symbol="gp",
+    description="Submit android apps to google play",
+    order=150,
+    context=[],
+    available=lambda params: params["project"] == "mozilla-central",
+    schema={"type": "object", "properties": {}},
+    permission="googleplay",
+)
+def add_push_bundle(parameters, graph_config, input, task_group_id, task_id):
+    decision_task_id, full_task_graph, label_to_taskid, _ = fetch_graph_and_labels(
+        parameters, graph_config
+    )
+
+    def filter(task):
+        if task.kind != "push-bundle":
+            return False
+        return task.attributes["build-type"] in ("fenix-nightly", "focus-nightly")
+
+    to_run = [label for label, task in full_task_graph.tasks.items() if filter(task)]
+
+    create_tasks(
+        graph_config,
+        to_run,
+        full_task_graph,
+        label_to_taskid,
+        parameters,
+        decision_task_id,
+    )
+    logger.info(f"Scheduled {len(to_run)} push-bundle tasks")