From c888502671c04bc7d77cfb8e590cc91d864efd5e Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Mon, 31 Mar 2025 08:41:13 -0400
Subject: [PATCH] Add quality scale summary generator (#141780)

* Add quality scale summary generator

* Remove executable bit

* Split out virtual
---
 script/quality_scale_summary.py | 89 +++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 script/quality_scale_summary.py

diff --git a/script/quality_scale_summary.py b/script/quality_scale_summary.py
new file mode 100644
index 00000000000..b93eab81451
--- /dev/null
+++ b/script/quality_scale_summary.py
@@ -0,0 +1,89 @@
+"""Generate a summary of integration quality scales.
+
+Run with python3 -m script.quality_scale_summary
+Data collected at https://docs.google.com/spreadsheets/d/1xEiwovRJyPohAv8S4ad2LAB-0A38s1HWmzHng8v-4NI
+"""
+
+import csv
+from pathlib import Path
+import sys
+
+from homeassistant.const import __version__ as current_version
+from homeassistant.util.json import load_json
+
+COMPONENTS_DIR = Path("homeassistant/components")
+
+
+def generate_quality_scale_summary() -> list[str, int]:
+    """Generate a summary of integration quality scales."""
+    quality_scales = {
+        "virtual": 0,
+        "unknown": 0,
+        "legacy": 0,
+        "internal": 0,
+        "bronze": 0,
+        "silver": 0,
+        "gold": 0,
+        "platinum": 0,
+    }
+
+    for manifest_path in COMPONENTS_DIR.glob("*/manifest.json"):
+        manifest = load_json(manifest_path)
+
+        if manifest.get("integration_type") == "virtual":
+            quality_scales["virtual"] += 1
+        elif quality_scale := manifest.get("quality_scale"):
+            quality_scales[quality_scale] += 1
+        else:
+            quality_scales["unknown"] += 1
+
+    return quality_scales
+
+
+def output_csv(quality_scales: dict[str, int], print_header: bool) -> None:
+    """Output the quality scale summary as CSV."""
+    writer = csv.writer(sys.stdout)
+    if print_header:
+        writer.writerow(
+            [
+                "Version",
+                "Total",
+                "Virtual",
+                "Unknown",
+                "Legacy",
+                "Internal",
+                "Bronze",
+                "Silver",
+                "Gold",
+                "Platinum",
+            ]
+        )
+
+    # Calculate total
+    total = sum(quality_scales.values())
+
+    # Write the summary
+    writer.writerow(
+        [
+            current_version,
+            total,
+            quality_scales["virtual"],
+            quality_scales["unknown"],
+            quality_scales["legacy"],
+            quality_scales["internal"],
+            quality_scales["bronze"],
+            quality_scales["silver"],
+            quality_scales["gold"],
+            quality_scales["platinum"],
+        ]
+    )
+
+
+def main() -> None:
+    """Run the script."""
+    quality_scales = generate_quality_scale_summary()
+    output_csv(quality_scales, "--header" in sys.argv)
+
+
+if __name__ == "__main__":
+    main()