Small cleanups for august (#49493)

This commit is contained in:
J. Nick Koston
2021-04-24 23:32:34 -10:00
committed by GitHub
parent 153d6e891e
commit e2837f08e8
5 changed files with 123 additions and 26 deletions

View File

@@ -45,7 +45,7 @@ SENSOR_TYPES_BATTERY = {
async def async_setup_entry(hass, config_entry, async_add_entities):
"""Set up the August sensors."""
data = hass.data[DOMAIN][config_entry.entry_id][DATA_AUGUST]
devices = []
entities = []
migrate_unique_id_devices = []
operation_sensors = []
batteries = {
@@ -72,7 +72,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
"Adding battery sensor for %s",
device.device_name,
)
devices.append(AugustBatterySensor(data, "device_battery", device, device))
entities.append(AugustBatterySensor(data, "device_battery", device, device))
for device in batteries["linked_keypad_battery"]:
detail = data.get_device_detail(device.device_id)
@@ -90,15 +90,15 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
keypad_battery_sensor = AugustBatterySensor(
data, "linked_keypad_battery", detail.keypad, device
)
devices.append(keypad_battery_sensor)
entities.append(keypad_battery_sensor)
migrate_unique_id_devices.append(keypad_battery_sensor)
for device in operation_sensors:
devices.append(AugustOperatorSensor(data, device))
entities.append(AugustOperatorSensor(data, device))
await _async_migrate_old_unique_ids(hass, migrate_unique_id_devices)
async_add_entities(devices, True)
async_add_entities(entities)
async def _async_migrate_old_unique_ids(hass, devices):