mirror of https://github.com/home-assistant/core
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
"""Generate usb file."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from .model import Config, Integration
|
|
from .serializer import format_python_namespace
|
|
|
|
|
|
def generate_and_validate(integrations: dict[str, Integration]) -> str:
|
|
"""Validate and generate usb data."""
|
|
match_list = []
|
|
|
|
for domain in sorted(integrations):
|
|
match_types = integrations[domain].manifest.get("usb", [])
|
|
|
|
if not match_types:
|
|
continue
|
|
|
|
match_list.extend(
|
|
{
|
|
"domain": domain,
|
|
**{k: v for k, v in entry.items() if k != "known_devices"},
|
|
}
|
|
for entry in match_types
|
|
)
|
|
|
|
return format_python_namespace({"USB": match_list})
|
|
|
|
|
|
def validate(integrations: dict[str, Integration], config: Config) -> None:
|
|
"""Validate usb file."""
|
|
usb_path = config.root / "homeassistant/generated/usb.py"
|
|
config.cache["usb"] = content = generate_and_validate(integrations)
|
|
|
|
if config.specific_integrations:
|
|
return
|
|
|
|
if usb_path.read_text() != content:
|
|
config.add_error(
|
|
"usb",
|
|
"File usb.py is not up to date. Run python3 -m script.hassfest",
|
|
fixable=True,
|
|
)
|
|
|
|
|
|
def generate(integrations: dict[str, Integration], config: Config) -> None:
|
|
"""Generate usb file."""
|
|
usb_path = config.root / "homeassistant/generated/usb.py"
|
|
usb_path.write_text(f"{config.cache['usb']}")
|