From 177b5734547a1ea1cc1199d1430c6668a7d7bb95 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 5 Jan 2022 13:26:23 +0100 Subject: [PATCH] Add setup type hints to magicseaweed (#63451) Co-authored-by: epenet --- .../components/magicseaweed/sensor.py | 30 ++++++++++++------- 1 file changed, 19 insertions(+), 11 deletions(-) diff --git a/homeassistant/components/magicseaweed/sensor.py b/homeassistant/components/magicseaweed/sensor.py index 5979759f416..38559f607a4 100644 --- a/homeassistant/components/magicseaweed/sensor.py +++ b/homeassistant/components/magicseaweed/sensor.py @@ -18,7 +18,10 @@ from homeassistant.const import ( CONF_MONITORED_CONDITIONS, CONF_NAME, ) +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.util import Throttle import homeassistant.util.dt as dt_util @@ -75,7 +78,12 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=30) -def setup_platform(hass, config, add_entities, discovery_info=None): +def setup_platform( + hass: HomeAssistant, + config: ConfigType, + add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up the Magicseaweed sensor.""" name = config.get(CONF_NAME) spot_id = config[CONF_SPOT_ID] @@ -102,16 +110,16 @@ def setup_platform(hass, config, add_entities, discovery_info=None): for description in SENSOR_TYPES if description.key in monitored_conditions ] - sensors.extend( - [ - MagicSeaweedSensor(forecast_data, name, units, description, hour) - for description in SENSOR_TYPES - if description.key in monitored_conditions - and "forecast" not in description.key - for hour in hours - if hour is not None - ] - ) + if hours is not None: + sensors.extend( + [ + MagicSeaweedSensor(forecast_data, name, units, description, hour) + for description in SENSOR_TYPES + if description.key in monitored_conditions + and "forecast" not in description.key + for hour in hours + ] + ) add_entities(sensors, True)