From 378cfab501c70a6f8566aef6828031da8520bd9a Mon Sep 17 00:00:00 2001 From: Oncleben31 Date: Mon, 4 Oct 2021 06:15:41 +0200 Subject: [PATCH] Meteofrance fix #56975 (#57016) --- .../components/meteo_france/sensor.py | 29 +++++++++++-------- 1 file changed, 17 insertions(+), 12 deletions(-) diff --git a/homeassistant/components/meteo_france/sensor.py b/homeassistant/components/meteo_france/sensor.py index 9f24cf02a2c..1a5b3c4a33a 100644 --- a/homeassistant/components/meteo_france/sensor.py +++ b/homeassistant/components/meteo_france/sensor.py @@ -44,18 +44,23 @@ async def async_setup_entry( MeteoFranceSensor(coordinator_forecast, description) for description in SENSOR_TYPES ] - entities.extend( - [ - MeteoFranceRainSensor(coordinator_rain, description) - for description in SENSOR_TYPES_RAIN - ] - ) - entities.extend( - [ - MeteoFranceAlertSensor(coordinator_alert, description) - for description in SENSOR_TYPES_ALERT - ] - ) + # Add rain forecast entity only if location support this feature + if coordinator_rain: + entities.extend( + [ + MeteoFranceRainSensor(coordinator_rain, description) + for description in SENSOR_TYPES_RAIN + ] + ) + # Add weather alert entity only if location support this feature + if coordinator_alert: + entities.extend( + [ + MeteoFranceAlertSensor(coordinator_alert, description) + for description in SENSOR_TYPES_ALERT + ] + ) + # Add weather probability entities only if location support this feature if coordinator_forecast.data.probability_forecast: entities.extend( [