home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-06 14:15:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Ensure diagnostics redaction can handle lists of lists

Browse Source
This commit is contained in:
Aaron Bach
2022-01-28 22:14:15 -07:00
parent 5e62ff95b9
commit 74c9ed7450
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
homeassistant/components/diagnostics/util.py
View File

@@ -2,6 +2,7 @@
from __future__ import annotations
from collections.abc import Iterable, Mapping
from copy import deepcopy
from typing import Any
from homeassistant.core import callback
@@ -10,19 +11,23 @@ from .const import REDACTED
@callback
def async_redact_data(data: Mapping, to_redact: Iterable[Any]) -> dict[str, Any]:
def async_redact_data(data: Mapping, to_redact: Iterable[Any]) -> Mapping[Any, Any]:
"""Redact sensitive data in a dict."""
if not isinstance(data, (Mapping, list)):
return data
redacted = {**data}
redacted = deepcopy(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]
if isinstance(redacted, dict):
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]
elif isinstance(redacted, list):
for idx, value in enumerate(redacted):
redacted[idx] = async_redact_data(value, to_redact)
return redacted