87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
|
"""Rebuild codes_index.json by scanning all codes/{platform}/*.json files.
|
|
|
|
Run from the repo root:
|
|
python tools/build_index.py
|
|
|
|
Outputs: custom_components/infrarize/codes_index.json
|
|
"""
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
PLATFORMS = ["climate", "fan", "media_player", "light"]
|
|
|
|
_TOOLS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
REPO_ROOT = os.path.dirname(_TOOLS_DIR)
|
|
CODES_DIR = os.path.join(REPO_ROOT, "codes")
|
|
OUTPUT_FILE = os.path.join(
|
|
REPO_ROOT, "custom_components", "infrarize", "codes_index.json"
|
|
)
|
|
|
|
|
|
def build_index() -> dict:
|
|
index: dict[str, list] = {}
|
|
total = 0
|
|
errors = 0
|
|
|
|
for platform in PLATFORMS:
|
|
platform_dir = os.path.join(CODES_DIR, platform)
|
|
entries: list[dict] = []
|
|
|
|
if not os.path.isdir(platform_dir):
|
|
print(f" {platform}: directory not found, skipping")
|
|
index[platform] = entries
|
|
continue
|
|
|
|
files = sorted(f for f in os.listdir(platform_dir) if f.endswith(".json"))
|
|
|
|
for filename in files:
|
|
code_str = filename[:-5] # strip .json
|
|
try:
|
|
code = int(code_str)
|
|
except ValueError:
|
|
continue
|
|
|
|
path = os.path.join(platform_dir, filename)
|
|
try:
|
|
with open(path, encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
except Exception as exc:
|
|
print(f" ERROR {filename}: {exc}", file=sys.stderr)
|
|
errors += 1
|
|
continue
|
|
|
|
entries.append(
|
|
{
|
|
"code": code,
|
|
"manufacturer": data.get("manufacturer", "Unknown"),
|
|
"models": data.get("supportedModels", []),
|
|
"controller": data.get("supportedController", "Unknown"),
|
|
"encoding": data.get("commandsEncoding", "Unknown"),
|
|
}
|
|
)
|
|
total += 1
|
|
|
|
index[platform] = entries
|
|
print(f" {platform}: {len(entries)} devices")
|
|
|
|
return index, total, errors
|
|
|
|
|
|
def main() -> None:
|
|
print(f"Scanning {CODES_DIR} ...")
|
|
index, total, errors = build_index()
|
|
|
|
with open(OUTPUT_FILE, "w", encoding="utf-8") as f:
|
|
json.dump(index, f, indent=2)
|
|
|
|
print(f"\nDone: {total} devices indexed")
|
|
print(f"Output: {OUTPUT_FILE}")
|
|
if errors:
|
|
print(f"Warnings: {errors} files failed to parse", file=sys.stderr)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|