evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use generators for async_add_entities in Accuweather ()

Browse source
This commit is contained in:
Robert Hillis 2022-08-10 18:05:32 -04:00 • committed by GitHub
parent d81298a2d6
commit 52fd63acbc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 11 deletions

14
homeassistant/components/accuweather/sensor.py
View file

@ -320,18 +320,18 @@ async def async_setup_entry(
coordinator: AccuWeatherDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id] coordinator: AccuWeatherDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
sensors: list[AccuWeatherSensor] = [] sensors = [
for description in SENSOR_TYPES: AccuWeatherSensor(coordinator, description) for description in SENSOR_TYPES
sensors.append(AccuWeatherSensor(coordinator, description)) ]
if coordinator.forecast: if coordinator.forecast:
for description in FORECAST_SENSOR_TYPES:
for day in range(MAX_FORECAST_DAYS + 1):
# Some air quality/allergy sensors are only available for certain # Some air quality/allergy sensors are only available for certain
# locations. # locations.
if description.key in coordinator.data[ATTR_FORECAST][0]: sensors.extend(
sensors.append(
AccuWeatherSensor(coordinator, description, forecast_day=day) AccuWeatherSensor(coordinator, description, forecast_day=day)
for description in FORECAST_SENSOR_TYPES
for day in range(MAX_FORECAST_DAYS + 1)
if description.key in coordinator.data[ATTR_FORECAST][0]
) )
async_add_entities(sensors) async_add_entities(sensors)