|
21 | 21 | import time |
22 | 22 |
|
23 | 23 | REPOSITORY_DIR = "/var/lib/repository" |
24 | | -ELEMENTS_DIR = "genesis-elements" |
| 24 | +ELEMENTS_DIRS = [ |
| 25 | + "genesis-elements", |
| 26 | + "exordos-elements", |
| 27 | +] |
25 | 28 | INVENTORY = "*/inventory.json" |
26 | 29 |
|
27 | 30 |
|
@@ -67,22 +70,23 @@ def write_merged_inventory(merged_data, output_path): |
67 | 70 |
|
68 | 71 |
|
69 | 72 | def main(): |
70 | | - root_dir = os.path.join(REPOSITORY_DIR, ELEMENTS_DIR) |
71 | | - output_file = os.path.join(root_dir, "inventory.json") |
| 73 | + for elements_dir in ELEMENTS_DIRS: |
| 74 | + root_dir = os.path.join(REPOSITORY_DIR, elements_dir) |
| 75 | + output_file = os.path.join(root_dir, "inventory.json") |
72 | 76 |
|
73 | | - print("Searching for inventory.json files...") |
74 | | - inventory_files = find_inventory_files(root_dir) |
| 77 | + print("Searching for inventory.json files...") |
| 78 | + inventory_files = find_inventory_files(root_dir) |
75 | 79 |
|
76 | | - if not inventory_files: |
77 | | - print("No inventory.json files found.") |
78 | | - return |
| 80 | + if not inventory_files: |
| 81 | + print("No inventory.json files found.") |
| 82 | + return |
79 | 83 |
|
80 | | - print(f"Found {len(inventory_files)} inventory.json files.") |
| 84 | + print(f"Found {len(inventory_files)} inventory.json files.") |
81 | 85 |
|
82 | | - merged_data = merge_inventories(inventory_files) |
83 | | - print(f"Merged {len(merged_data['elements'])} items.") |
| 86 | + merged_data = merge_inventories(inventory_files) |
| 87 | + print(f"Merged {len(merged_data['elements'])} items.") |
84 | 88 |
|
85 | | - write_merged_inventory(merged_data, output_file) |
| 89 | + write_merged_inventory(merged_data, output_file) |
86 | 90 |
|
87 | 91 |
|
88 | 92 | if __name__ == "__main__": |
|
0 commit comments