59 lines
1.7 KiB
Python
Executable file
59 lines
1.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# 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 argparse
|
|
import os
|
|
|
|
import taskcluster
|
|
|
|
queue = taskcluster.Queue(
|
|
{
|
|
"rootUrl": os.environ.get("TASKCLUSTER_PROXY_URL", "https://taskcluster.net"),
|
|
}
|
|
)
|
|
|
|
|
|
def check_all_dependencies_are_completed(current_task_id):
|
|
print(f"Fetching task definition of {current_task_id}...")
|
|
task = queue.task(current_task_id)
|
|
dependencies_task_ids = task["dependencies"]
|
|
|
|
print(f"Fetching status of {len(dependencies_task_ids)} dependencies...")
|
|
# TODO Make this dict-comprehension async once we go Python 3
|
|
state_per_task_ids = {
|
|
task_id: queue.status(task_id)["status"]["state"]
|
|
for task_id in dependencies_task_ids
|
|
}
|
|
print("Statuses fetched.")
|
|
non_completed_tasks = {
|
|
task_id: state
|
|
for task_id, state in state_per_task_ids.items()
|
|
if state != "completed"
|
|
}
|
|
|
|
if non_completed_tasks:
|
|
raise ValueError(f"Some tasks are not completed: {non_completed_tasks}")
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description='Errors out if one of the DEPENDENCY_TASK_ID does not have the Taskcluster status "completed"'
|
|
)
|
|
|
|
parser.add_argument(
|
|
"current_task_id",
|
|
metavar="CURRENT_TASK_ID",
|
|
help="The task ID of the current running task",
|
|
)
|
|
|
|
result = parser.parse_args()
|
|
check_all_dependencies_are_completed(result.current_task_id)
|
|
print("All dependencies are completed. Reporting a green task!")
|
|
exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|