55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""Support for Bang & Olufsen diagnostics."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from homeassistant.components.event import DOMAIN as EVENT_DOMAIN
|
|
from homeassistant.components.media_player import DOMAIN as MEDIA_PLAYER_DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import BangOlufsenConfigEntry
|
|
from .const import DEVICE_BUTTONS, DOMAIN
|
|
|
|
|
|
async def async_get_config_entry_diagnostics(
|
|
hass: HomeAssistant, config_entry: BangOlufsenConfigEntry
|
|
) -> dict[str, Any]:
|
|
"""Return diagnostics for a config entry."""
|
|
|
|
data: dict = {
|
|
"config_entry": config_entry.as_dict(),
|
|
"websocket_connected": config_entry.runtime_data.client.websocket_connected,
|
|
}
|
|
|
|
if TYPE_CHECKING:
|
|
assert config_entry.unique_id
|
|
|
|
entity_registry = er.async_get(hass)
|
|
|
|
# Add media_player entity's state
|
|
if entity_id := entity_registry.async_get_entity_id(
|
|
MEDIA_PLAYER_DOMAIN, DOMAIN, config_entry.unique_id
|
|
):
|
|
if state := hass.states.get(entity_id):
|
|
state_dict = dict(state.as_dict())
|
|
|
|
# Remove context as it is not relevant
|
|
state_dict.pop("context")
|
|
data["media_player"] = state_dict
|
|
|
|
# Add button Event entity states (if enabled)
|
|
for device_button in DEVICE_BUTTONS:
|
|
if entity_id := entity_registry.async_get_entity_id(
|
|
EVENT_DOMAIN, DOMAIN, f"{config_entry.unique_id}_{device_button}"
|
|
):
|
|
if state := hass.states.get(entity_id):
|
|
state_dict = dict(state.as_dict())
|
|
|
|
# Remove context as it is not relevant
|
|
state_dict.pop("context")
|
|
data[f"{device_button}_event"] = state_dict
|
|
|
|
return data
|