mirror/custom_vesync
1
0
Fork 0
mirror of https://github.com/micahqcade/custom_vesync.git synced 2025-03-13 08:10:06 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve diagnostics (#43)

Browse Source 
* Improve diagnostics

* fix style
This commit is contained in:
Vincent Le Bourlot 2022-08-03 11:06:21 +02:00 committed by GitHub
parent dc222cb03c
commit 95a8c624b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
custom_components/vesync/diagnostics.py
View File

@ -3,11 +3,15 @@ from __future__ import annotations
from typing import Any
from homeassistant.components.diagnostics import async_redact_data
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from .common import is_humidifier
from .const import DOMAIN
TO_REDACT = {"cid", "uuid", "mac_id"}
def _if_has_attr_else_none(obj, attr):
return getattr(obj, attr) if hasattr(obj, attr) else None
@ -18,14 +22,12 @@ async def async_get_config_entry_diagnostics(
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
data = hass.data[DOMAIN][entry.entry_id]
devices = {"fans": [], "outlets": [], "switches": [], "bulbs": []}
devices = {}
for type in ["fans", "outlets", "switches", "bulbs"]:
for d in data["manager"]._dev_list[type]:
devices[type].append(
{
"config_dict": _if_has_attr_else_none(d, "config_dict") or {},
"config": _if_has_attr_else_none(d, "config") or {},
"details": _if_has_attr_else_none(d, "details") or {},
}
)
return devices
t = "humidifier" if is_humidifier(d.device_type) else type
devices = {
**devices,
**{t: [{k: v for k, v in d.__dict__.items() if k != "manager"}]},
}
return async_redact_data(devices, TO_REDACT)