1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#!/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()
|
