diagnostics.py

Go to the documentation of this file.

"""Support for Bang & Olufsen diagnostics."""
 
from __future__ import annotations
 
from typing import TYPE_CHECKING, Any
 
from homeassistant.components.media_player import DOMAIN as MEDIA_PLAYER_DOMAIN
from homeassistant.core import HomeAssistant
import homeassistant.helpers.entity_registry as er
 
from . import BangOlufsenConfigEntry
from .const import 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
 
    # Add media_player entity's state
    entity_registry = er.async_get(hass)
    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
 
    return data