Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-25 05:26:47 +00:00
Code Activity

Enforce RegistryEntryHider in entity registry (#73219)

Browse Source
This commit is contained in:
Erik Montnemery
2022-06-08 21:36:43 +02:00
committed by GitHub
parent 6bf219550e
commit 4435c641de
6 changed files with 60 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/helpers/entity_registry.py
View File

@@ -369,6 +369,8 @@ class EntityRegistry:
if disabled_by and not isinstance(disabled_by, RegistryEntryDisabler):
raise ValueError("disabled_by must be a RegistryEntryDisabler value")
if hidden_by and not isinstance(hidden_by, RegistryEntryHider):
raise ValueError("hidden_by must be a RegistryEntryHider value")
if (
disabled_by is None
@@ -520,6 +522,12 @@ class EntityRegistry:
and not isinstance(disabled_by, RegistryEntryDisabler)
):
raise ValueError("disabled_by must be a RegistryEntryDisabler value")
if (
hidden_by
and hidden_by is not UNDEFINED
and not isinstance(hidden_by, RegistryEntryHider)
):
raise ValueError("hidden_by must be a RegistryEntryHider value")
from .entity import EntityCategory # pylint: disable=import-outside-toplevel
@@ -729,7 +737,9 @@ class EntityRegistry:
if entity["entity_category"]
else None,
entity_id=entity["entity_id"],
hidden_by=entity["hidden_by"],
hidden_by=RegistryEntryHider(entity["hidden_by"])
if entity["hidden_by"]
else None,
icon=entity["icon"],
id=entity["id"],
name=entity["name"],