Use generators for async_add_entities in Ambient Station (#76586)

This commit is contained in:
Robert Hillis 2022-08-10 21:05:09 -04:00 committed by GitHub
parent 519d478d61
commit 8ecbb85852
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 14 deletions

View File

@ -333,14 +333,12 @@ async def async_setup_entry(
ambient = hass.data[DOMAIN][entry.entry_id] ambient = hass.data[DOMAIN][entry.entry_id]
async_add_entities( async_add_entities(
[
AmbientWeatherBinarySensor( AmbientWeatherBinarySensor(
ambient, mac_address, station[ATTR_NAME], description ambient, mac_address, station[ATTR_NAME], description
) )
for mac_address, station in ambient.stations.items() for mac_address, station in ambient.stations.items()
for description in BINARY_SENSOR_DESCRIPTIONS for description in BINARY_SENSOR_DESCRIPTIONS
if description.key in station[ATTR_LAST_DATA] if description.key in station[ATTR_LAST_DATA]
]
) )

View File

@ -646,12 +646,10 @@ async def async_setup_entry(
ambient = hass.data[DOMAIN][entry.entry_id] ambient = hass.data[DOMAIN][entry.entry_id]
async_add_entities( async_add_entities(
[
AmbientWeatherSensor(ambient, mac_address, station[ATTR_NAME], description) AmbientWeatherSensor(ambient, mac_address, station[ATTR_NAME], description)
for mac_address, station in ambient.stations.items() for mac_address, station in ambient.stations.items()
for description in SENSOR_DESCRIPTIONS for description in SENSOR_DESCRIPTIONS
if description.key in station[ATTR_LAST_DATA] if description.key in station[ATTR_LAST_DATA]
]
) )