1 files changed, 39 insertions, 0 deletions
diff --git a/taskcluster/docker/snap-coreXX-build/parse.py b/taskcluster/docker/snap-coreXX-build/parse.py
new file mode 100644
index 0000000000..82e0e2f4a6
--- /dev/null
+++ b/taskcluster/docker/snap-coreXX-build/parse.py
@@ -0,0 +1,39 @@
+#!/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 sys
+
+import yaml
+
+
+def has_pkg_section(p, section):
+    has_section = section in p.keys()
+    if has_section:
+        for pkg in p[section]:
+            yield pkg
+
+
+def iter_pkgs(part, all_pkgs):
+    for pkg in has_pkg_section(part, "build-packages"):
+        if pkg not in all_pkgs:
+            all_pkgs.append(pkg)
+    for pkg in has_pkg_section(part, "stage-packages"):
+        if pkg not in all_pkgs:
+            all_pkgs.append(pkg)
+
+
+def parse(yaml_file):
+    all_pkgs = []
+    with open(yaml_file, "r") as inp:
+        snap = yaml.safe_load(inp)
+        parts = snap["parts"]
+        for p in parts:
+            iter_pkgs(parts[p], all_pkgs)
+    return " ".join(all_pkgs)
+
+
+if __name__ == "__main__":
+    print(parse(sys.argv[1]))