Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions custom_components/aguaiot/diagnostics.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
from homeassistant.core import HomeAssistant

from .const import DOMAIN

TO_REDACT = {
CONF_PASSWORD,
Expand All @@ -16,10 +15,11 @@


async def async_get_config_entry_diagnostics(
hass: HomeAssistant, entry: ConfigEntry
hass: HomeAssistant, config_entry: ConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
agua = hass.data[DOMAIN][entry.entry_id]["agua"]
coordinator = config_entry.runtime_data
agua = coordinator.agua

devices = {}
for device in agua.devices:
Expand All @@ -28,6 +28,6 @@ async def async_get_config_entry_diagnostics(
devices[device.name][reg] = device.get_register(reg)

return {
"entry": async_redact_data(entry.as_dict(), TO_REDACT),
"entry": async_redact_data(config_entry.as_dict(), TO_REDACT),
"devices": devices,
}