From c4f98a3084f2648749d3e4eeeade9696630d9abd Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Fri, 26 Mar 2021 01:05:19 -1000 Subject: [PATCH] Small speed up to adding entities (#48353) --- homeassistant/helpers/entity_platform.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/homeassistant/helpers/entity_platform.py b/homeassistant/helpers/entity_platform.py index 382ebf8055e..6f41c67ef01 100644 --- a/homeassistant/helpers/entity_platform.py +++ b/homeassistant/helpers/entity_platform.py @@ -18,7 +18,12 @@ from homeassistant.core import ( valid_entity_id, ) from homeassistant.exceptions import HomeAssistantError, PlatformNotReady -from homeassistant.helpers import config_validation as cv, service +from homeassistant.helpers import ( + config_validation as cv, + device_registry as dev_reg, + entity_registry as ent_reg, + service, +) from homeassistant.helpers.typing import HomeAssistantType from homeassistant.util.async_ import run_callback_threadsafe @@ -298,8 +303,8 @@ class EntityPlatform: hass = self.hass - device_registry = await hass.helpers.device_registry.async_get_registry() - entity_registry = await hass.helpers.entity_registry.async_get_registry() + device_registry = dev_reg.async_get(hass) + entity_registry = ent_reg.async_get(hass) tasks = [ self._async_add_entity( # type: ignore entity, update_before_add, entity_registry, device_registry