8
8
from homeassistant .const import (ATTR_BATTERY_LEVEL , CONF_DEVICES , CONF_MAC ,
9
9
CONF_NAME , CONF_TEMPERATURE_UNIT ,
10
10
CONF_UNIQUE_ID , UnitOfMass , UnitOfTemperature )
11
+ from homeassistant .helpers import device_registry , entity_registry
11
12
from homeassistant .helpers .event import async_call_later
12
13
from homeassistant .helpers .typing import StateType
13
14
from homeassistant .util import dt
@@ -140,8 +141,8 @@ async def async_add_sensor(key, device_model, firmware, auto_sensors, manufactur
140
141
141
142
# setup sensors of configured devices on startup when device model is available in registry
142
143
if self .config [CONF_DEVICES ]:
143
- dev_registry = hass . helpers . device_registry .async_get (hass )
144
- ent_registry = hass . helpers . entity_registry .async_get (hass )
144
+ dev_registry = device_registry .async_get (hass )
145
+ ent_registry = entity_registry .async_get (hass )
145
146
for device in self .config [CONF_DEVICES ]:
146
147
# get device_model and firmware from device registry to setup sensor
147
148
key = dict_get_or (device )
@@ -158,7 +159,7 @@ async def async_add_sensor(key, device_model, firmware, auto_sensors, manufactur
158
159
firmware = RENAMED_FIRMWARE_DICT .get (firmware , firmware )
159
160
manufacturer = RENAMED_MANUFACTURER_DICT .get (manufacturer , manufacturer )
160
161
# get all entities for this device
161
- entity_list = hass . helpers . entity_registry .async_entries_for_device (
162
+ entity_list = entity_registry .async_entries_for_device (
162
163
registry = ent_registry , device_id = device_id , include_disabled_entities = False
163
164
)
164
165
# find the measurement key for each entity
0 commit comments