github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-08-02 23:40:32 +02:00
Code Releases Activity

Use generators for async_add_entities in Abode (#76569)

Browse Source
This commit is contained in:
Robert Hillis 2022-08-10 18:09:05 -04:00 committed by GitHub
parent 52fd63acbc
commit ca3033b84c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 35 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/abode/binary_sensor.py
View File

@ -30,12 +30,10 @@ async def async_setup_entry(
CONST.TYPE_OPENING,
]
entities = []
for device in data.abode.get_devices(generic_type=device_types):
entities.append(AbodeBinarySensor(data, device))
async_add_entities(entities)
async_add_entities(
AbodeBinarySensor(data, device)
for device in data.abode.get_devices(generic_type=device_types)
)
class AbodeBinarySensor(AbodeDevice, BinarySensorEntity):

9
homeassistant/components/abode/camera.py
View File

@ -28,12 +28,11 @@ async def async_setup_entry(
) -> None:
"""Set up Abode camera devices."""
data: AbodeSystem = hass.data[DOMAIN]
entities = []
for device in data.abode.get_devices(generic_type=CONST.TYPE_CAMERA):
entities.append(AbodeCamera(data, device, TIMELINE.CAPTURE_IMAGE))
async_add_entities(entities)
async_add_entities(
AbodeCamera(data, device, TIMELINE.CAPTURE_IMAGE)
for device in data.abode.get_devices(generic_type=CONST.TYPE_CAMERA)
)
class AbodeCamera(AbodeDevice, Camera):

10
homeassistant/components/abode/cover.py
View File

@ -19,12 +19,10 @@ async def async_setup_entry(
"""Set up Abode cover devices."""
data: AbodeSystem = hass.data[DOMAIN]
entities = []
for device in data.abode.get_devices(generic_type=CONST.TYPE_COVER):
entities.append(AbodeCover(data, device))
async_add_entities(entities)
async_add_entities(
AbodeCover(data, device)
for device in data.abode.get_devices(generic_type=CONST.TYPE_COVER)
)
class AbodeCover(AbodeDevice, CoverEntity):

10
homeassistant/components/abode/light.py
View File

@ -32,12 +32,10 @@ async def async_setup_entry(
"""Set up Abode light devices."""
data: AbodeSystem = hass.data[DOMAIN]
entities = []
for device in data.abode.get_devices(generic_type=CONST.TYPE_LIGHT):
entities.append(AbodeLight(data, device))
async_add_entities(entities)
async_add_entities(
AbodeLight(data, device)
for device in data.abode.get_devices(generic_type=CONST.TYPE_LIGHT)
)
class AbodeLight(AbodeDevice, LightEntity):

10
homeassistant/components/abode/lock.py
View File

@ -19,12 +19,10 @@ async def async_setup_entry(
"""Set up Abode lock devices."""
data: AbodeSystem = hass.data[DOMAIN]
entities = []
for device in data.abode.get_devices(generic_type=CONST.TYPE_LOCK):
entities.append(AbodeLock(data, device))
async_add_entities(entities)
async_add_entities(
AbodeLock(data, device)
for device in data.abode.get_devices(generic_type=CONST.TYPE_LOCK)
)
class AbodeLock(AbodeDevice, LockEntity):

19
homeassistant/components/abode/sensor.py
View File

@ -42,19 +42,12 @@ async def async_setup_entry(
"""Set up Abode sensor devices."""
data: AbodeSystem = hass.data[DOMAIN]
entities = []
for device in data.abode.get_devices(generic_type=CONST.TYPE_SENSOR):
conditions = device.get_value(CONST.STATUSES_KEY)
entities.extend(
[
AbodeSensor(data, device, description)
for description in SENSOR_TYPES
if description.key in conditions
]
)
async_add_entities(entities)
async_add_entities(
AbodeSensor(data, device, description)
for description in SENSOR_TYPES
for device in data.abode.get_devices(generic_type=CONST.TYPE_SENSOR)
if description.key in device.get_value(CONST.STATUSES_KEY)
)
class AbodeSensor(AbodeDevice, SensorEntity):

16
homeassistant/components/abode/switch.py
View File

@ -25,14 +25,16 @@ async def async_setup_entry(
"""Set up Abode switch devices."""
data: AbodeSystem = hass.data[DOMAIN]
entities: list[SwitchEntity] = []
entities: list[SwitchEntity] = [
AbodeSwitch(data, device)
for device_type in DEVICE_TYPES
for device in data.abode.get_devices(generic_type=device_type)
]
for device_type in DEVICE_TYPES:
for device in data.abode.get_devices(generic_type=device_type):
entities.append(AbodeSwitch(data, device))
for automation in data.abode.get_automations():
entities.append(AbodeAutomationSwitch(data, automation))
entities.extend(
AbodeAutomationSwitch(data, automation)
for automation in data.abode.get_automations()
)
async_add_entities(entities)