python

custom_components/ttlock/diagnostics.py

@@ -0,0 +1,40 @@
+"""Diagnostics support for Tractive."""
+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 .const import DOMAIN, TT_LOCKS
+
+TO_REDACT = {
+    "token",
+    "lockKey",
+    "aesKeyStr",
+    "adminPwd",
+    "deletePwd",
+    "noKeyPwd",
+    "lockData",
+    "webhook_url",
+}
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, config_entry: ConfigEntry
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+
+    diagnostics_data = async_redact_data(
+        {
+            "config_entry": config_entry.as_dict(),
+            "locks": [
+                coordinator.as_dict()
+                for coordinator in hass.data[DOMAIN][config_entry.entry_id][TT_LOCKS]
+            ],
+        },
+        TO_REDACT,
+    )
+
+    return diagnostics_data