79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
|
from homeassistant.config_entries import ConfigEntry
|
||
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
from homeassistant.helpers.device_registry import DeviceEntry
|
||
|
|
|
||
|
|
from .core.const import DOMAIN, PRIVATE_KEYS, source_hash
|
||
|
|
from .core.ewelink import XRegistry
|
||
|
|
|
||
|
|
|
||
|
|
async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: ConfigEntry):
|
||
|
|
try:
|
||
|
|
if XRegistry.config:
|
||
|
|
config = XRegistry.config.copy()
|
||
|
|
for k in (CONF_USERNAME, CONF_PASSWORD):
|
||
|
|
if config.get(k):
|
||
|
|
config[k] = "***"
|
||
|
|
if config.get("devices"):
|
||
|
|
for device in config["devices"].values():
|
||
|
|
if device.get("devicekey"):
|
||
|
|
device["devicekey"] = "***"
|
||
|
|
else:
|
||
|
|
config = None
|
||
|
|
except Exception as e:
|
||
|
|
config = f"{type(e).__name__}: {e}"
|
||
|
|
|
||
|
|
options = {k: len(v) if k == "homes" else v for k, v in entry.options.items()}
|
||
|
|
|
||
|
|
registry: XRegistry = hass.data[DOMAIN][entry.entry_id]
|
||
|
|
try:
|
||
|
|
devices = {
|
||
|
|
did: {
|
||
|
|
"uiid": device["extra"]["uiid"],
|
||
|
|
"params": {
|
||
|
|
k: "***" if k in PRIVATE_KEYS else v
|
||
|
|
for k, v in device["params"].items()
|
||
|
|
},
|
||
|
|
"model": device.get("productModel"),
|
||
|
|
"online": device.get("online"),
|
||
|
|
"local": device.get("local"),
|
||
|
|
"localtype": device.get("localtype"),
|
||
|
|
"host": device.get("host"),
|
||
|
|
}
|
||
|
|
if "params" in device
|
||
|
|
else {
|
||
|
|
"localtype": device.get("localtype"),
|
||
|
|
}
|
||
|
|
for did, device in registry.devices.items()
|
||
|
|
}
|
||
|
|
except Exception as e:
|
||
|
|
devices = f"{type(e).__name__}: {e}"
|
||
|
|
|
||
|
|
try:
|
||
|
|
errors = [
|
||
|
|
entry.to_dict()
|
||
|
|
for key, entry in hass.data["system_log"].records.items()
|
||
|
|
if DOMAIN in key
|
||
|
|
]
|
||
|
|
except Exception as e:
|
||
|
|
errors = f"{type(e).__name__}: {e}"
|
||
|
|
|
||
|
|
return {
|
||
|
|
"version": source_hash(),
|
||
|
|
"cloud_auth": registry.cloud.auth is not None,
|
||
|
|
"config": config,
|
||
|
|
"options": options,
|
||
|
|
"errors": errors,
|
||
|
|
"devices": devices,
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
async def async_get_device_diagnostics(
|
||
|
|
hass: HomeAssistant, entry: ConfigEntry, device: DeviceEntry
|
||
|
|
):
|
||
|
|
did = next(i[1] for i in device.identifiers if i[0] == DOMAIN)
|
||
|
|
info = await async_get_config_entry_diagnostics(hass, entry)
|
||
|
|
info["device"] = info.pop("devices").get(did, {})
|
||
|
|
info["device"]["deviceid"] = did
|
||
|
|
return info
|