Use index.yaml to generate directory index in list_priorities. (#234)

* Use index.yaml to generate directory index in list_priorities. As suggested by Steven!, we should parse the index.yaml to find the list of distributions to parse, rather than having a hard-coded list. Signed-off-by: Chris Lalancette <[email protected]> Co-authored-by: Scott K Logan <[email protected]>
ros2 · Aug 11, 2022 · 7659414 · 7659414
1 parent f6f5f36
commit 7659414
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/list_priorities.py b/list_priorities.py
@@ -7,13 +7,23 @@
 import yamlinclude
 
 
-DIRECTORIES = ['foxy', 'galactic', 'humble', 'rolling']
-
 def main():
     base_path = os.path.dirname(__file__)
+
+    # First parse the index.yaml to find the list of distributions that we
+    # should parse.
+
+    yamlinclude.YamlIncludeConstructor.add_to_loader_class(loader_class=yaml.FullLoader, base_dir=base_path)
+    with open(os.path.join(base_path, 'index.yaml')) as infp:
+        index_data = yaml.load(infp, Loader=yaml.FullLoader)
+
+    directories = list(index_data['distributions'].keys())
+
+    del yaml.FullLoader.yaml_constructors[yamlinclude.YamlIncludeConstructor.DEFAULT_TAG_NAME]
+
     priorities = {}
 
-    for directory in DIRECTORIES:
+    for directory in directories:
         current_path = os.path.join(base_path, directory)
         yamlinclude.YamlIncludeConstructor.add_to_loader_class(loader_class=yaml.FullLoader, base_dir=current_path)