mirror of
https://github.com/home-assistant/core.git
synced 2025-05-03 21:49:17 +00:00
cc6b0cc843
* Ensure diagnostics redaction can handle lists of lists * Code review * Update homeassistant/components/diagnostics/util.py Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io> * Code review * Typing * Revert "Typing" This reverts commit 8a57f772caa5180b609175591d81dfc473769f70. * New typing attempt * Revert "New typing attempt" This reverts commit e26e4aae69f62325fdd6af4d80c8fd1f74846e54. * Fix typing * Fix typing again * Add tests Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
34 lines
913 B
Python
34 lines
913 B
Python
"""Diagnostic utilities."""
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Iterable, Mapping
|
|
from typing import Any, TypeVar, cast
|
|
|
|
from homeassistant.core import callback
|
|
|
|
from .const import REDACTED
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
@callback
|
|
def async_redact_data(data: T, to_redact: Iterable[Any]) -> T:
|
|
"""Redact sensitive data in a dict."""
|
|
if not isinstance(data, (Mapping, list)):
|
|
return data
|
|
|
|
if isinstance(data, list):
|
|
return cast(T, [async_redact_data(val, to_redact) for val in data])
|
|
|
|
redacted = {**data}
|
|
|
|
for key, value in redacted.items():
|
|
if key in to_redact:
|
|
redacted[key] = REDACTED
|
|
elif isinstance(value, dict):
|
|
redacted[key] = async_redact_data(value, to_redact)
|
|
elif isinstance(value, list):
|
|
redacted[key] = [async_redact_data(item, to_redact) for item in value]
|
|
|
|
return cast(T, redacted)
|