mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-05-09 23:05:43 +02:00
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import os
|
|
import yaml
|
|
from pathlib import Path
|
|
|
|
def load_yaml_file(path):
|
|
"""Load a YAML file if it exists, otherwise return an empty dict."""
|
|
if not path.exists():
|
|
return {}
|
|
with path.open("r", encoding="utf-8") as f:
|
|
return yaml.safe_load(f) or {}
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Generate default_applications YAML from docker roles.")
|
|
parser.add_argument("--roles-dir", default="roles", help="Path to the roles directory (default: roles)")
|
|
parser.add_argument("--output-file", default="group_vars/all/11_applications.yml", help="Path to output YAML file")
|
|
|
|
args = parser.parse_args()
|
|
cwd = Path.cwd()
|
|
roles_dir = (cwd / args.roles_dir).resolve()
|
|
output_file = (cwd / args.output_file).resolve()
|
|
|
|
output_file.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
result = {"default_applications": {}}
|
|
|
|
for role_dir in sorted(roles_dir.iterdir()):
|
|
role_name = role_dir.name
|
|
vars_main = role_dir / "vars" / "main.yml"
|
|
config_file = role_dir / "vars" / "configuration.yml"
|
|
|
|
if not vars_main.exists():
|
|
print(f"[!] Skipping {role_name}: vars/main.yml missing")
|
|
continue
|
|
|
|
vars_data = load_yaml_file(vars_main)
|
|
application_id = vars_data.get("application_id")
|
|
|
|
if not application_id:
|
|
print(f"[!] Skipping {role_name}: application_id not defined in vars/main.yml")
|
|
continue
|
|
|
|
if not config_file.exists():
|
|
print(f"[!] Skipping {role_name}: vars/configuration.yml missing")
|
|
continue
|
|
|
|
config_data = load_yaml_file(config_file)
|
|
if config_data:
|
|
result["default_applications"][application_id] = config_data
|
|
|
|
with output_file.open("w", encoding="utf-8") as f:
|
|
yaml.dump(result, f, sort_keys=False)
|
|
|
|
print(f"✅ Generated: {output_file.relative_to(cwd)}")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|