Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 21:06:19 +00:00
Code Activity

Make entity registry disabled_by an enum (#60424)

Browse Source
This commit is contained in:
Ville Skyttä
2021-12-02 19:02:29 +02:00
committed by GitHub
parent d6dff403c9
commit 4107063a5a
3 changed files with 73 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/helpers/entity_platform.py
View File

@@ -42,7 +42,7 @@ from . import (
service,
)
from .device_registry import DeviceRegistry
from .entity_registry import DISABLED_DEVICE, DISABLED_INTEGRATION, EntityRegistry
from .entity_registry import EntityRegistry, RegistryEntryDisabler
from .event import async_call_later, async_track_time_interval
from .typing import ConfigType, DiscoveryInfoType
@@ -502,9 +502,9 @@ class EntityPlatform:
except RequiredParameterMissing:
pass
disabled_by: str | None = None
disabled_by: RegistryEntryDisabler | None = None
if not entity.entity_registry_enabled_default:
disabled_by = DISABLED_INTEGRATION
disabled_by = RegistryEntryDisabler.INTEGRATION
entry = entity_registry.async_get_or_create(
self.domain,
@@ -526,7 +526,7 @@ class EntityPlatform:
if device and device.disabled and not entry.disabled:
entry = entity_registry.async_update_entity(
entry.entity_id, disabled_by=DISABLED_DEVICE
entry.entity_id, disabled_by=RegistryEntryDisabler.DEVICE
)
entity.registry_entry = entry