Initial commit.

app-configs/rikku/homeassistant/custom_components/mail_and_packages/diagnostics.py

@@ -0,0 +1,44 @@
+"""Provide diagnostics for Mail and Packages."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from homeassistant.components.diagnostics import async_redact_data
+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 .const import CONF_AMAZON_FWDS, COORDINATOR, DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+REDACT_KEYS = {CONF_PASSWORD, CONF_USERNAME, CONF_AMAZON_FWDS}
+
+
+async def async_get_config_entry_diagnostics(
+    hass: HomeAssistant, config_entry: ConfigEntry  # pylint: disable=unused-argument
+) -> dict[str, Any]:
+    """Return diagnostics for a config entry."""
+    diag: dict[str, Any] = {}
+    diag["config"] = config_entry.as_dict()
+    return async_redact_data(diag, REDACT_KEYS)
+
+
+async def async_get_device_diagnostics(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    device: DeviceEntry,  # pylint: disable=unused-argument
+) -> dict[str, Any]:
+    """Return diagnostics for a device."""
+    coordinator = hass.data[DOMAIN][config_entry.entry_id][COORDINATOR]
+
+    for variable in coordinator.data:
+        if "tracking" in variable or "order" in variable:
+            _LOGGER.debug("Atempting to add: %s for redaction.", variable)
+            REDACT_KEYS.add(variable)
+
+    _LOGGER.debug("Redacted keys: %s", REDACT_KEYS)
+
+    return async_redact_data(coordinator.data, REDACT_KEYS)