jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-28 15:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass device manager directly in Tuya registry cleanup (#57511)

Browse Source
This commit is contained in:
Franck Nijhof 2021-10-12 05:36:46 +02:00 committed by GitHub
parent f0abd5b815
commit 71a3daa8ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/tuya/__init__.py
View File

@ -111,7 +111,7 @@ async def _init_tuya_sdk(hass: HomeAssistant, entry: ConfigEntry) -> bool:
hass.data[DOMAIN][entry.entry_id][TUYA_DEVICE_MANAGER] = device_manager hass.data[DOMAIN][entry.entry_id][TUYA_DEVICE_MANAGER] = device_manager
# Clean up device entities # Clean up device entities
await cleanup_device_registry(hass, entry) await cleanup_device_registry(hass, device_manager)
_LOGGER.debug("init support type->%s", PLATFORMS) _LOGGER.debug("init support type->%s", PLATFORMS)
@ -120,12 +120,11 @@ async def _init_tuya_sdk(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True return True
async def cleanup_device_registry(hass: HomeAssistant, entry: ConfigEntry) -> None: async def cleanup_device_registry(
hass: HomeAssistant, device_manager: TuyaDeviceManager
) -> None:
"""Remove deleted device registry entry if there are no remaining entities.""" """Remove deleted device registry entry if there are no remaining entities."""
device_registry_object = device_registry.async_get(hass) device_registry_object = device_registry.async_get(hass)
device_manager = hass.data[DOMAIN][entry.entry_id][TUYA_DEVICE_MANAGER]
for dev_id, device_entry in list(device_registry_object.devices.items()): for dev_id, device_entry in list(device_registry_object.devices.items()):
for item in device_entry.identifiers: for item in device_entry.identifiers:
if DOMAIN == item[0] and item[1] not in device_manager.device_map: if DOMAIN == item[0] and item[1] not in device_manager.device_map:
@ -171,7 +170,6 @@ class DeviceListener(TuyaDeviceListener):
def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Init DeviceListener.""" """Init DeviceListener."""
self.hass = hass self.hass = hass
self.entry = entry self.entry = entry