core/homeassistant/components/fronius/diagnostics.py

47 lines
1.5 KiB
Python

"""Diagnostics support for Fronius."""
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant
from . import FroniusConfigEntry
TO_REDACT = {"unique_id", "unique_identifier", "serial"}
async def async_get_config_entry_diagnostics(
hass: HomeAssistant, config_entry: FroniusConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
diag: dict[str, Any] = {}
solar_net = config_entry.runtime_data
fronius = solar_net.fronius
diag["config_entry"] = config_entry.as_dict()
diag["inverter_info"] = await fronius.inverter_info()
diag["coordinators"] = {"inverters": {}}
for inv in solar_net.inverter_coordinators:
diag["coordinators"]["inverters"] |= inv.data
diag["coordinators"]["logger"] = (
solar_net.logger_coordinator.data if solar_net.logger_coordinator else None
)
diag["coordinators"]["meter"] = (
solar_net.meter_coordinator.data if solar_net.meter_coordinator else None
)
diag["coordinators"]["ohmpilot"] = (
solar_net.ohmpilot_coordinator.data if solar_net.ohmpilot_coordinator else None
)
diag["coordinators"]["power_flow"] = (
solar_net.power_flow_coordinator.data
if solar_net.power_flow_coordinator
else None
)
diag["coordinators"]["storage"] = (
solar_net.storage_coordinator.data if solar_net.storage_coordinator else None
)
return async_redact_data(diag, TO_REDACT)