jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-25 01:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve AsusWRT diagnostics (#69826)

Browse Source
This commit is contained in:
ollo69 2022-04-18 21:20:01 +02:00 committed by GitHub
parent ce1f074ca9
commit a27f61cf07
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
homeassistant/components/asuswrt/diagnostics.py
View File

@ -3,9 +3,16 @@ from __future__ import annotations
from typing import Any
import attr
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.const import (
ATTR_CONNECTIONS,
ATTR_IDENTIFIERS,
CONF_PASSWORD,
CONF_USERNAME,
)
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
@ -13,6 +20,7 @@ from .const import DATA_ASUSWRT, DOMAIN
from .router import AsusWrtRouter
TO_REDACT = {CONF_PASSWORD, CONF_USERNAME}
TO_REDACT_DEV = {ATTR_CONNECTIONS, ATTR_IDENTIFIERS}
async def async_get_config_entry_diagnostics(
@ -33,11 +41,7 @@ async def async_get_config_entry_diagnostics(
return data
data["device"] = {
"name": hass_device.name,
"name_by_user": hass_device.name_by_user,
"disabled": hass_device.disabled,
"disabled_by": hass_device.disabled_by,
"device_info": async_redact_data(dict(router.device_info), {"identifiers"}),
**async_redact_data(attr.asdict(hass_device), TO_REDACT_DEV),
"entities": {},
"tracked_devices": [],
}
@ -59,16 +63,12 @@ async def async_get_config_entry_diagnostics(
state_dict.pop("context", None)
data["device"]["entities"][entity_entry.entity_id] = {
"name": entity_entry.name,
"original_name": entity_entry.original_name,
"disabled": entity_entry.disabled,
"disabled_by": entity_entry.disabled_by,
"entity_category": entity_entry.entity_category,
"device_class": entity_entry.device_class,
"original_device_class": entity_entry.original_device_class,
"icon": entity_entry.icon,
"original_icon": entity_entry.original_icon,
"unit_of_measurement": entity_entry.unit_of_measurement,
**async_redact_data(
attr.asdict(
entity_entry, filter=lambda attr, value: attr.name != "entity_id"
),
TO_REDACT,
),
"state": state_dict,
}