From eafece36515b1c2206b031f79d138e03eb98121e Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 12 Jan 2022 08:49:46 +0100 Subject: [PATCH] Add setup type hints (part 1) (#63955) Co-authored-by: epenet --- homeassistant/components/acer_projector/switch.py | 2 +- homeassistant/components/androidtv/media_player.py | 4 ++-- homeassistant/components/awair/sensor.py | 2 +- .../components/eight_sleep/binary_sensor.py | 2 +- homeassistant/components/flexit/climate.py | 7 ++++--- homeassistant/components/fronius/sensor.py | 4 ++-- homeassistant/components/isy994/binary_sensor.py | 6 ++---- homeassistant/components/isy994/climate.py | 6 ++---- homeassistant/components/isy994/cover.py | 6 ++---- homeassistant/components/isy994/fan.py | 6 ++---- homeassistant/components/isy994/light.py | 6 ++---- homeassistant/components/isy994/lock.py | 6 ++---- homeassistant/components/isy994/sensor.py | 6 ++---- homeassistant/components/isy994/switch.py | 6 ++---- .../components/netgear_lte/binary_sensor.py | 12 +++++++++++- homeassistant/components/sensibo/climate.py | 2 +- homeassistant/components/systemmonitor/sensor.py | 4 ++-- .../components/trafikverket_weatherstation/sensor.py | 2 +- .../components/yamaha_musiccast/media_player.py | 4 ++-- homeassistant/components/zha/alarm_control_panel.py | 2 +- homeassistant/components/zha/binary_sensor.py | 2 +- homeassistant/components/zha/climate.py | 2 +- homeassistant/components/zha/cover.py | 2 +- homeassistant/components/zha/device_tracker.py | 2 +- homeassistant/components/zha/fan.py | 2 +- homeassistant/components/zha/light.py | 2 +- homeassistant/components/zha/number.py | 2 +- homeassistant/components/zha/switch.py | 2 +- 28 files changed, 53 insertions(+), 58 deletions(-) diff --git a/homeassistant/components/acer_projector/switch.py b/homeassistant/components/acer_projector/switch.py index f3e4b3b03e5..75e3d6081ba 100644 --- a/homeassistant/components/acer_projector/switch.py +++ b/homeassistant/components/acer_projector/switch.py @@ -53,7 +53,7 @@ def setup_platform( hass: HomeAssistant, config: ConfigType, add_entities: AddEntitiesCallback, - discovery_info: DiscoveryInfoType, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Connect with serial port and return Acer Projector.""" serial_port = config[CONF_FILENAME] diff --git a/homeassistant/components/androidtv/media_player.py b/homeassistant/components/androidtv/media_player.py index 099c4fddba2..8526ce1c524 100644 --- a/homeassistant/components/androidtv/media_player.py +++ b/homeassistant/components/androidtv/media_player.py @@ -54,7 +54,7 @@ from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, format from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback -from homeassistant.helpers.typing import ConfigType +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from .const import ( ANDROID_DEV, @@ -171,7 +171,7 @@ async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, - discovery_info=None, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Set up the Android TV / Fire TV platform.""" diff --git a/homeassistant/components/awair/sensor.py b/homeassistant/components/awair/sensor.py index f6b0c34103f..89c505bd4b9 100644 --- a/homeassistant/components/awair/sensor.py +++ b/homeassistant/components/awair/sensor.py @@ -67,7 +67,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up Awair sensor entity based on a config entry.""" coordinator = hass.data[DOMAIN][config_entry.entry_id] entities = [] diff --git a/homeassistant/components/eight_sleep/binary_sensor.py b/homeassistant/components/eight_sleep/binary_sensor.py index d2ca30ab580..a13a0fd840d 100644 --- a/homeassistant/components/eight_sleep/binary_sensor.py +++ b/homeassistant/components/eight_sleep/binary_sensor.py @@ -29,7 +29,7 @@ async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, - discovery_info: DiscoveryInfoType = None, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Set up the eight sleep binary sensor.""" if discovery_info is None: diff --git a/homeassistant/components/flexit/climate.py b/homeassistant/components/flexit/climate.py index ce3e5a68e1e..9e2b9e41255 100644 --- a/homeassistant/components/flexit/climate.py +++ b/homeassistant/components/flexit/climate.py @@ -29,6 +29,7 @@ from homeassistant.const import ( ) 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 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( @@ -47,9 +48,9 @@ SUPPORT_FLAGS = SUPPORT_TARGET_TEMPERATURE | SUPPORT_FAN_MODE async def async_setup_platform( hass: HomeAssistant, config: ConfigType, - async_add_entities, - discovery_info: DiscoveryInfoType = None, -): + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up the Flexit Platform.""" modbus_slave = config.get(CONF_SLAVE) name = config.get(CONF_NAME) diff --git a/homeassistant/components/fronius/sensor.py b/homeassistant/components/fronius/sensor.py index 67d86c1cc48..26747785837 100644 --- a/homeassistant/components/fronius/sensor.py +++ b/homeassistant/components/fronius/sensor.py @@ -31,7 +31,7 @@ from homeassistant.core import HomeAssistant, callback import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity import DeviceInfo, EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback -from homeassistant.helpers.typing import ConfigType +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN @@ -68,7 +68,7 @@ async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, - discovery_info: None = None, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Import Fronius configuration from yaml.""" _LOGGER.warning( diff --git a/homeassistant/components/isy994/binary_sensor.py b/homeassistant/components/isy994/binary_sensor.py index f0645110b86..5cf7d6b2b76 100644 --- a/homeassistant/components/isy994/binary_sensor.py +++ b/homeassistant/components/isy994/binary_sensor.py @@ -52,10 +52,8 @@ DEVICE_PARENT_REQUIRED = [ async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 binary sensor platform.""" devices = [] devices_by_address = {} diff --git a/homeassistant/components/isy994/climate.py b/homeassistant/components/isy994/climate.py index df9598b00ee..98c20bb441a 100644 --- a/homeassistant/components/isy994/climate.py +++ b/homeassistant/components/isy994/climate.py @@ -60,10 +60,8 @@ ISY_SUPPORTED_FEATURES = ( async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 thermostat platform.""" entities = [] diff --git a/homeassistant/components/isy994/cover.py b/homeassistant/components/isy994/cover.py index ca5432f4456..0fcfb30a775 100644 --- a/homeassistant/components/isy994/cover.py +++ b/homeassistant/components/isy994/cover.py @@ -27,10 +27,8 @@ from .helpers import migrate_old_unique_ids async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 cover platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] devices = [] diff --git a/homeassistant/components/isy994/fan.py b/homeassistant/components/isy994/fan.py index f9f3c6e5459..28d0675a184 100644 --- a/homeassistant/components/isy994/fan.py +++ b/homeassistant/components/isy994/fan.py @@ -23,10 +23,8 @@ SPEED_RANGE = (1, 255) # off is not included async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 fan platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] devices = [] diff --git a/homeassistant/components/isy994/light.py b/homeassistant/components/isy994/light.py index 3d208d18fa9..2fd98b6f177 100644 --- a/homeassistant/components/isy994/light.py +++ b/homeassistant/components/isy994/light.py @@ -28,10 +28,8 @@ ATTR_LAST_BRIGHTNESS = "last_brightness" async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 light platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] isy_options = entry.options diff --git a/homeassistant/components/isy994/lock.py b/homeassistant/components/isy994/lock.py index c00a12d0096..e2befc57487 100644 --- a/homeassistant/components/isy994/lock.py +++ b/homeassistant/components/isy994/lock.py @@ -15,10 +15,8 @@ VALUE_TO_STATE = {0: False, 100: True} async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 lock platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] devices = [] diff --git a/homeassistant/components/isy994/sensor.py b/homeassistant/components/isy994/sensor.py index 6439a0d198f..466a7334775 100644 --- a/homeassistant/components/isy994/sensor.py +++ b/homeassistant/components/isy994/sensor.py @@ -25,10 +25,8 @@ from .helpers import convert_isy_value_to_hass, migrate_old_unique_ids async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 sensor platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] devices = [] diff --git a/homeassistant/components/isy994/switch.py b/homeassistant/components/isy994/switch.py index 99bf6566b1b..3e72dd6f0ec 100644 --- a/homeassistant/components/isy994/switch.py +++ b/homeassistant/components/isy994/switch.py @@ -13,10 +13,8 @@ from .helpers import migrate_old_unique_ids async def async_setup_entry( - hass: HomeAssistant, - entry: ConfigEntry, - async_add_entities: AddEntitiesCallback, -) -> bool: + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the ISY994 switch platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] devices = [] diff --git a/homeassistant/components/netgear_lte/binary_sensor.py b/homeassistant/components/netgear_lte/binary_sensor.py index 7c9c326edb7..0e7cd14bea6 100644 --- a/homeassistant/components/netgear_lte/binary_sensor.py +++ b/homeassistant/components/netgear_lte/binary_sensor.py @@ -1,12 +1,22 @@ """Support for Netgear LTE binary sensors.""" +from __future__ import annotations + from homeassistant.components.binary_sensor import BinarySensorEntity +from homeassistant.core import HomeAssistant from homeassistant.exceptions import PlatformNotReady +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import CONF_BINARY_SENSOR, CONF_MONITORED_CONDITIONS, DATA_KEY, LTEEntity from .sensor_types import BINARY_SENSOR_CLASSES -async def async_setup_platform(hass, config, async_add_entities, discovery_info): +async def async_setup_platform( + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up Netgear LTE binary sensor devices.""" if discovery_info is None: return diff --git a/homeassistant/components/sensibo/climate.py b/homeassistant/components/sensibo/climate.py index e15b1a4e940..648e05dbe79 100644 --- a/homeassistant/components/sensibo/climate.py +++ b/homeassistant/components/sensibo/climate.py @@ -82,7 +82,7 @@ async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, - discovery_info: DiscoveryInfoType = None, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Set up Sensibo devices.""" _LOGGER.warning( diff --git a/homeassistant/components/systemmonitor/sensor.py b/homeassistant/components/systemmonitor/sensor.py index 53a5a806214..79e5e3f9fea 100644 --- a/homeassistant/components/systemmonitor/sensor.py +++ b/homeassistant/components/systemmonitor/sensor.py @@ -43,7 +43,7 @@ from homeassistant.helpers.dispatcher import ( from homeassistant.helpers.entity_component import DEFAULT_SCAN_INTERVAL from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.event import async_track_time_interval -from homeassistant.helpers.typing import ConfigType +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.util import slugify import homeassistant.util.dt as dt_util @@ -324,7 +324,7 @@ async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, - discovery_info: Any | None = None, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Set up the system monitor sensors.""" entities = [] diff --git a/homeassistant/components/trafikverket_weatherstation/sensor.py b/homeassistant/components/trafikverket_weatherstation/sensor.py index b4ae196ae00..7aa4b6c96e3 100644 --- a/homeassistant/components/trafikverket_weatherstation/sensor.py +++ b/homeassistant/components/trafikverket_weatherstation/sensor.py @@ -165,7 +165,7 @@ async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, - discovery_info: DiscoveryInfoType = None, + discovery_info: DiscoveryInfoType | None = None, ) -> None: """Import Trafikverket Weather configuration from YAML.""" _LOGGER.warning( diff --git a/homeassistant/components/yamaha_musiccast/media_player.py b/homeassistant/components/yamaha_musiccast/media_player.py index 9407036fcd9..bcd1a0a2c1f 100644 --- a/homeassistant/components/yamaha_musiccast/media_player.py +++ b/homeassistant/components/yamaha_musiccast/media_player.py @@ -49,7 +49,7 @@ from homeassistant.exceptions import HomeAssistantError import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity_platform import AddEntitiesCallback -from homeassistant.helpers.typing import DiscoveryInfoType +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.util import uuid from . import MusicCastDataUpdateCoordinator, MusicCastDeviceEntity @@ -87,7 +87,7 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( async def async_setup_platform( hass: HomeAssistant, - config, + config: ConfigType, async_add_devices: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None, ) -> None: diff --git a/homeassistant/components/zha/alarm_control_panel.py b/homeassistant/components/zha/alarm_control_panel.py index 4d4cba3599c..17dc47ebefa 100644 --- a/homeassistant/components/zha/alarm_control_panel.py +++ b/homeassistant/components/zha/alarm_control_panel.py @@ -61,7 +61,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation alarm control panel from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.ALARM_CONTROL_PANEL] diff --git a/homeassistant/components/zha/binary_sensor.py b/homeassistant/components/zha/binary_sensor.py index 4bc7c156f37..730748d74ae 100644 --- a/homeassistant/components/zha/binary_sensor.py +++ b/homeassistant/components/zha/binary_sensor.py @@ -43,7 +43,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation binary sensor from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.BINARY_SENSOR] diff --git a/homeassistant/components/zha/climate.py b/homeassistant/components/zha/climate.py index d15cd42e4e9..65de5fd04cc 100644 --- a/homeassistant/components/zha/climate.py +++ b/homeassistant/components/zha/climate.py @@ -125,7 +125,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation sensor from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.CLIMATE] unsub = async_dispatcher_connect( diff --git a/homeassistant/components/zha/cover.py b/homeassistant/components/zha/cover.py index 2e8c7ab45ea..9f62d4b9c02 100644 --- a/homeassistant/components/zha/cover.py +++ b/homeassistant/components/zha/cover.py @@ -49,7 +49,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation cover from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.COVER] diff --git a/homeassistant/components/zha/device_tracker.py b/homeassistant/components/zha/device_tracker.py index 78d43952f8a..c08491ab782 100644 --- a/homeassistant/components/zha/device_tracker.py +++ b/homeassistant/components/zha/device_tracker.py @@ -31,7 +31,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation device tracker from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.DEVICE_TRACKER] diff --git a/homeassistant/components/zha/fan.py b/homeassistant/components/zha/fan.py index 729a0f7e618..10b64caf974 100644 --- a/homeassistant/components/zha/fan.py +++ b/homeassistant/components/zha/fan.py @@ -56,7 +56,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation fan from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.FAN] diff --git a/homeassistant/components/zha/light.py b/homeassistant/components/zha/light.py index 2c3b6249cf9..6855db22572 100644 --- a/homeassistant/components/zha/light.py +++ b/homeassistant/components/zha/light.py @@ -110,7 +110,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation light from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.LIGHT] diff --git a/homeassistant/components/zha/number.py b/homeassistant/components/zha/number.py index e3de49756c2..3c5a374e588 100644 --- a/homeassistant/components/zha/number.py +++ b/homeassistant/components/zha/number.py @@ -239,7 +239,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation Analog Output from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.NUMBER] diff --git a/homeassistant/components/zha/switch.py b/homeassistant/components/zha/switch.py index b67ab72fa35..29fb08b9bc0 100644 --- a/homeassistant/components/zha/switch.py +++ b/homeassistant/components/zha/switch.py @@ -32,7 +32,7 @@ async def async_setup_entry( hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback, -): +) -> None: """Set up the Zigbee Home Automation switch from config entry.""" entities_to_create = hass.data[DATA_ZHA][Platform.SWITCH]