1db68327f9
* Enable Ruff PTH for the script directory * Address review comments * Fix translations script * Update script/hassfest/config_flow.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> --------- Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
"""Generate ssdp file."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections import defaultdict
|
|
|
|
from .model import Config, Integration
|
|
from .serializer import format_python_namespace
|
|
|
|
|
|
def generate_and_validate(integrations: dict[str, Integration]) -> str:
|
|
"""Validate and generate ssdp data."""
|
|
|
|
data = defaultdict(list)
|
|
|
|
for domain in sorted(integrations):
|
|
ssdp = integrations[domain].manifest.get("ssdp")
|
|
|
|
if not ssdp:
|
|
continue
|
|
|
|
for matcher in ssdp:
|
|
data[domain].append(matcher)
|
|
|
|
return format_python_namespace({"SSDP": data})
|
|
|
|
|
|
def validate(integrations: dict[str, Integration], config: Config) -> None:
|
|
"""Validate ssdp file."""
|
|
ssdp_path = config.root / "homeassistant/generated/ssdp.py"
|
|
config.cache["ssdp"] = content = generate_and_validate(integrations)
|
|
|
|
if config.specific_integrations:
|
|
return
|
|
|
|
if ssdp_path.read_text() != content:
|
|
config.add_error(
|
|
"ssdp",
|
|
"File ssdp.py is not up to date. Run python3 -m script.hassfest",
|
|
fixable=True,
|
|
)
|
|
|
|
|
|
def generate(integrations: dict[str, Integration], config: Config) -> None:
|
|
"""Generate ssdp file."""
|
|
ssdp_path = config.root / "homeassistant/generated/ssdp.py"
|
|
ssdp_path.write_text(f"{config.cache['ssdp']}")
|