updated diagnostics

This commit is contained in:
Tomaae 2022-03-26 01:26:30 +01:00
parent 5c8a8d4870
commit 0c097e2206
No known key found for this signature in database
GPG key ID: 8360BBD8A381D1C0

View file

@ -1,45 +1,10 @@
"""Diagnostics support for Mikrotik Router.""" """Diagnostics support for Mikrotik Router."""
from __future__ import annotations from __future__ import annotations
from typing import Any from typing import Any
from homeassistant.components.diagnostics import async_redact_data from homeassistant.components.diagnostics import async_redact_data
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from .const import DOMAIN, TO_REDACT
from .const import DOMAIN
TO_REDACT = {
"ip-address",
"client-ip-address",
"address",
"active-address",
"mac-address",
"active-mac-address",
"orig-mac-address",
"port-mac-address",
"client-mac-address",
"client-id",
"active-client-id",
"eeprom",
"sfp-vendor-serial",
"gateway",
"dns-server",
"wins-server",
"ntp-server",
"caps-manager",
"serial-number",
"source",
"from-addresses",
"to-addresses",
"src-address",
"dst-address",
"username",
"password",
"caller-id",
"target",
"ssid",
}
async def async_get_config_entry_diagnostics( async def async_get_config_entry_diagnostics(
@ -47,11 +12,11 @@ async def async_get_config_entry_diagnostics(
) -> dict[str, Any]: ) -> dict[str, Any]:
"""Return diagnostics for a config entry.""" """Return diagnostics for a config entry."""
controller = hass.data[DOMAIN][config_entry.entry_id] controller = hass.data[DOMAIN][config_entry.entry_id]
diag: dict[str, Any] = {}
diag["entry"]: dict[str, Any] = {}
diag["entry"]["data"] = async_redact_data(config_entry.data, TO_REDACT) return {
diag["entry"]["options"] = async_redact_data(config_entry.options, TO_REDACT) "entry": {
diag["data"] = async_redact_data(controller.data, TO_REDACT) "data": async_redact_data(config_entry.data, TO_REDACT),
"options": async_redact_data(config_entry.options, TO_REDACT),
return diag },
"data": async_redact_data(controller.data, TO_REDACT),
}