core/homeassistant/components/discovergy/diagnostics.py

"""Diagnostics support for discovergy."""

from __future__ import annotations

from dataclasses import asdict
from typing import Any

from homeassistant.components.diagnostics import async_redact_data
from homeassistant.core import HomeAssistant

from .coordinator import DiscovergyConfigEntry

TO_REDACT_METER = {
    "serial_number",
    "location",
    "full_serial_number",
    "printed_full_serial_number",
    "administration_number",
}


async def async_get_config_entry_diagnostics(
    hass: HomeAssistant, entry: DiscovergyConfigEntry
) -> dict[str, Any]:
    """Return diagnostics for a config entry."""
    flattened_meter: list[dict] = []
    last_readings: dict[str, dict] = {}

    for coordinator in entry.runtime_data:
        # make a dict of meter data and redact some data
        flattened_meter.append(
            async_redact_data(asdict(coordinator.meter), TO_REDACT_METER)
        )

        # get last reading for meter and make a dict of it
        last_readings[coordinator.meter.meter_id] = asdict(coordinator.data)

    return {
        "meters": flattened_meter,
        "readings": last_readings,
    }