diff --git a/homeassistant/components/blebox/cover.py b/homeassistant/components/blebox/cover.py index b107dba1e7f..7ede27fb042 100644 --- a/homeassistant/components/blebox/cover.py +++ b/homeassistant/components/blebox/cover.py @@ -1,5 +1,4 @@ """BleBox cover entity.""" - from homeassistant.components.cover import ( ATTR_POSITION, STATE_CLOSED, @@ -11,12 +10,19 @@ from homeassistant.components.cover import ( SUPPORT_STOP, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import BleBoxEntity, create_blebox_entities from .const import BLEBOX_TO_HASS_COVER_STATES, BLEBOX_TO_HASS_DEVICE_CLASSES -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up a BleBox entry.""" create_blebox_entities( diff --git a/homeassistant/components/bosch_shc/cover.py b/homeassistant/components/bosch_shc/cover.py index c08984ca0c2..6048cd51631 100644 --- a/homeassistant/components/bosch_shc/cover.py +++ b/homeassistant/components/bosch_shc/cover.py @@ -10,12 +10,19 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import DATA_SESSION, DOMAIN from .entity import SHCEntity -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up the SHC cover platform.""" entities = [] diff --git a/homeassistant/components/fibaro/cover.py b/homeassistant/components/fibaro/cover.py index 6c6c6ddfe1f..f98d907945b 100644 --- a/homeassistant/components/fibaro/cover.py +++ b/homeassistant/components/fibaro/cover.py @@ -1,15 +1,25 @@ """Support for Fibaro cover - curtains, rollershutters etc.""" +from __future__ import annotations + from homeassistant.components.cover import ( ATTR_POSITION, ATTR_TILT_POSITION, DOMAIN, CoverEntity, ) +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import FIBARO_DEVICES, FibaroDevice -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 Fibaro covers.""" if discovery_info is None: return diff --git a/homeassistant/components/freedompro/cover.py b/homeassistant/components/freedompro/cover.py index 2131069eb0e..7525d37182f 100644 --- a/homeassistant/components/freedompro/cover.py +++ b/homeassistant/components/freedompro/cover.py @@ -11,10 +11,12 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_API_KEY -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import aiohttp_client from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN @@ -30,7 +32,9 @@ DEVICE_CLASS_MAP = { SUPPORTED_SENSORS = {"windowCovering", "gate", "garageDoor", "door", "window"} -async def async_setup_entry(hass, entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up Freedompro cover.""" api_key = entry.data[CONF_API_KEY] coordinator = hass.data[DOMAIN][entry.entry_id] diff --git a/homeassistant/components/hunterdouglas_powerview/cover.py b/homeassistant/components/hunterdouglas_powerview/cover.py index f8644b15a5d..d1f33387a15 100644 --- a/homeassistant/components/hunterdouglas_powerview/cover.py +++ b/homeassistant/components/hunterdouglas_powerview/cover.py @@ -21,7 +21,9 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) -from homeassistant.core import callback +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.event import async_call_later from .const import ( @@ -49,7 +51,9 @@ TRANSITION_COMPLETE_DURATION = 30 PARALLEL_UPDATES = 1 -async def async_setup_entry(hass, entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the hunter douglas shades.""" pv_data = hass.data[DOMAIN][entry.entry_id] diff --git a/homeassistant/components/insteon/cover.py b/homeassistant/components/insteon/cover.py index fd20637b174..f66b1329bc6 100644 --- a/homeassistant/components/insteon/cover.py +++ b/homeassistant/components/insteon/cover.py @@ -9,8 +9,10 @@ from homeassistant.components.cover import ( SUPPORT_SET_POSITION, CoverEntity, ) -from homeassistant.core import callback +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import SIGNAL_ADD_ENTITIES from .insteon_entity import InsteonEntity @@ -19,7 +21,11 @@ from .utils import async_add_insteon_entities SUPPORTED_FEATURES = SUPPORT_OPEN | SUPPORT_CLOSE | SUPPORT_SET_POSITION -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up the Insteon covers from a config entry.""" @callback diff --git a/homeassistant/components/lutron_caseta/cover.py b/homeassistant/components/lutron_caseta/cover.py index 111da12a6f5..68a7c16db1d 100644 --- a/homeassistant/components/lutron_caseta/cover.py +++ b/homeassistant/components/lutron_caseta/cover.py @@ -11,6 +11,9 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import LutronCasetaDevice from .const import BRIDGE_DEVICE, BRIDGE_LEAP, DOMAIN as CASETA_DOMAIN @@ -18,7 +21,11 @@ from .const import BRIDGE_DEVICE, BRIDGE_LEAP, DOMAIN as CASETA_DOMAIN _LOGGER = logging.getLogger(__name__) -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up the Lutron Caseta cover platform. Adds shades from the Caseta bridge associated with the config_entry as diff --git a/homeassistant/components/motion_blinds/cover.py b/homeassistant/components/motion_blinds/cover.py index 495c2f7078a..5ef4e687969 100644 --- a/homeassistant/components/motion_blinds/cover.py +++ b/homeassistant/components/motion_blinds/cover.py @@ -1,5 +1,4 @@ """Support for Motion Blinds using their WLAN API.""" - import logging from motionblinds import BlindType @@ -11,8 +10,11 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant from homeassistant.helpers import config_validation as cv, entity_platform from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import ( @@ -62,7 +64,11 @@ SET_ABSOLUTE_POSITION_SCHEMA = { } -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up the Motion Blind from a config entry.""" entities = [] motion_gateway = hass.data[DOMAIN][config_entry.entry_id][KEY_GATEWAY] diff --git a/homeassistant/components/myq/cover.py b/homeassistant/components/myq/cover.py index 02876524c50..f3bc27a8453 100644 --- a/homeassistant/components/myq/cover.py +++ b/homeassistant/components/myq/cover.py @@ -8,14 +8,21 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_CLOSED, STATE_CLOSING, STATE_OPEN, STATE_OPENING +from homeassistant.core import HomeAssistant from homeassistant.exceptions import HomeAssistantError +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import MyQEntity from .const import DOMAIN, MYQ_COORDINATOR, MYQ_GATEWAY, MYQ_TO_HASS -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up mysq covers.""" data = hass.data[DOMAIN][config_entry.entry_id] myq = data[MYQ_GATEWAY] diff --git a/homeassistant/components/opengarage/cover.py b/homeassistant/components/opengarage/cover.py index 22a1d24c90b..86edd074ed2 100644 --- a/homeassistant/components/opengarage/cover.py +++ b/homeassistant/components/opengarage/cover.py @@ -7,8 +7,10 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_CLOSED, STATE_CLOSING, STATE_OPEN, STATE_OPENING -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import DOMAIN from .entity import OpenGarageEntity @@ -18,7 +20,9 @@ _LOGGER = logging.getLogger(__name__) STATES_MAP = {0: STATE_CLOSED, 1: STATE_OPEN} -async def async_setup_entry(hass, entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: """Set up the OpenGarage covers.""" async_add_entities( [OpenGarageCover(hass.data[DOMAIN][entry.entry_id], entry.unique_id)] diff --git a/homeassistant/components/ozw/cover.py b/homeassistant/components/ozw/cover.py index e8f1fc99872..780e19c2ccd 100644 --- a/homeassistant/components/ozw/cover.py +++ b/homeassistant/components/ozw/cover.py @@ -9,8 +9,10 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) -from homeassistant.core import callback +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import DATA_UNSUBSCRIBE, DOMAIN from .entity import ZWaveDeviceEntity @@ -21,7 +23,11 @@ PRESS_BUTTON = True RELEASE_BUTTON = False -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up Z-Wave Cover from Config Entry.""" @callback diff --git a/homeassistant/components/rflink/cover.py b/homeassistant/components/rflink/cover.py index b98fb0fe4a9..cd109821582 100644 --- a/homeassistant/components/rflink/cover.py +++ b/homeassistant/components/rflink/cover.py @@ -1,12 +1,17 @@ """Support for Rflink Cover devices.""" +from __future__ import annotations + import logging import voluptuous as vol from homeassistant.components.cover import PLATFORM_SCHEMA, CoverEntity from homeassistant.const import CONF_DEVICES, CONF_NAME, CONF_TYPE, STATE_OPEN +from homeassistant.core import HomeAssistant import homeassistant.helpers.config_validation as cv +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.restore_state import RestoreEntity +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import ( CONF_ALIASES, @@ -106,7 +111,12 @@ def devices_from_config(domain_config): return devices -async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): +async def async_setup_platform( + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up the Rflink cover platform.""" async_add_entities(devices_from_config(config)) diff --git a/homeassistant/components/scsgate/cover.py b/homeassistant/components/scsgate/cover.py index 2b7615bb780..ddfb59c6fba 100644 --- a/homeassistant/components/scsgate/cover.py +++ b/homeassistant/components/scsgate/cover.py @@ -1,4 +1,6 @@ """Support for SCSGate covers.""" +from __future__ import annotations + import logging from scsgate.tasks import ( @@ -10,7 +12,10 @@ import voluptuous as vol from homeassistant.components.cover import PLATFORM_SCHEMA, CoverEntity from homeassistant.const import CONF_DEVICES, 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 . import CONF_SCS_ID, DOMAIN, SCSGATE_SCHEMA @@ -19,7 +24,12 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( ) -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 SCSGate cover.""" devices = config.get(CONF_DEVICES) covers = [] diff --git a/homeassistant/components/slide/cover.py b/homeassistant/components/slide/cover.py index ab8d00717ff..a0d74a3081a 100644 --- a/homeassistant/components/slide/cover.py +++ b/homeassistant/components/slide/cover.py @@ -1,4 +1,5 @@ """Support for Slide slides.""" +from __future__ import annotations import logging @@ -11,13 +12,21 @@ from homeassistant.components.cover import ( CoverEntity, ) from homeassistant.const import ATTR_ID +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from .const import API, DEFAULT_OFFSET, DOMAIN, SLIDES _LOGGER = logging.getLogger(__name__) -async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): +async def async_setup_platform( + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up cover(s) for Slide platform.""" if discovery_info is None: diff --git a/homeassistant/components/smarthab/cover.py b/homeassistant/components/smarthab/cover.py index 6a53090d3f8..3c581774b03 100644 --- a/homeassistant/components/smarthab/cover.py +++ b/homeassistant/components/smarthab/cover.py @@ -13,6 +13,9 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import DATA_HUB, DOMAIN @@ -21,7 +24,11 @@ _LOGGER = logging.getLogger(__name__) SCAN_INTERVAL = timedelta(seconds=60) -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up SmartHab covers from a config entry.""" hub = hass.data[DOMAIN][config_entry.entry_id][DATA_HUB] diff --git a/homeassistant/components/smartthings/cover.py b/homeassistant/components/smartthings/cover.py index 269c1476836..e029143f7f3 100644 --- a/homeassistant/components/smartthings/cover.py +++ b/homeassistant/components/smartthings/cover.py @@ -18,7 +18,10 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_BATTERY_LEVEL +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import SmartThingsEntity from .const import DATA_BROKERS, DOMAIN @@ -33,7 +36,11 @@ VALUE_TO_STATE = { } -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Add covers for a config entry.""" broker = hass.data[DOMAIN][DATA_BROKERS][config_entry.entry_id] async_add_entities( diff --git a/homeassistant/components/soma/cover.py b/homeassistant/components/soma/cover.py index d1f5930c02b..8ed8969b34b 100644 --- a/homeassistant/components/soma/cover.py +++ b/homeassistant/components/soma/cover.py @@ -1,17 +1,23 @@ """Support for Soma Covers.""" - import logging from requests import RequestException from homeassistant.components.cover import ATTR_POSITION, CoverEntity +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import API, DEVICES, DOMAIN, SomaEntity _LOGGER = logging.getLogger(__name__) -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up the Soma cover platform.""" devices = hass.data[DOMAIN][DEVICES] diff --git a/homeassistant/components/somfy/cover.py b/homeassistant/components/somfy/cover.py index ac89f9b609e..e3bc6ef6d94 100644 --- a/homeassistant/components/somfy/cover.py +++ b/homeassistant/components/somfy/cover.py @@ -1,5 +1,4 @@ """Support for Somfy Covers.""" - from pymfy.api.devices.blind import Blind from pymfy.api.devices.category import Category @@ -17,7 +16,10 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_OPTIMISTIC, STATE_CLOSED, STATE_OPEN +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.restore_state import RestoreEntity from .const import COORDINATOR, DOMAIN @@ -32,7 +34,11 @@ SUPPORTED_CATEGORIES = { } -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up the Somfy cover platform.""" domain_data = hass.data[DOMAIN] coordinator = domain_data[COORDINATOR] diff --git a/homeassistant/components/somfy_mylink/cover.py b/homeassistant/components/somfy_mylink/cover.py index 480eeda9fd1..7ac0aace47f 100644 --- a/homeassistant/components/somfy_mylink/cover.py +++ b/homeassistant/components/somfy_mylink/cover.py @@ -2,8 +2,11 @@ import logging from homeassistant.components.cover import CoverDeviceClass, CoverEntity +from homeassistant.config_entries import ConfigEntry from homeassistant.const import STATE_CLOSED, STATE_OPEN +from homeassistant.core import HomeAssistant from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.restore_state import RestoreEntity from .const import ( @@ -22,7 +25,11 @@ MYLINK_COVER_TYPE_TO_DEVICE_CLASS = { } -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Discover and configure Somfy covers.""" reversed_target_ids = config_entry.options.get(CONF_REVERSED_TARGET_IDS, {}) diff --git a/homeassistant/components/tellduslive/cover.py b/homeassistant/components/tellduslive/cover.py index 735219d7a8c..b64c3f887a0 100644 --- a/homeassistant/components/tellduslive/cover.py +++ b/homeassistant/components/tellduslive/cover.py @@ -1,12 +1,19 @@ """Support for Tellstick covers using Tellstick Net.""" from homeassistant.components import cover, tellduslive from homeassistant.components.cover import CoverEntity +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .entry import TelldusLiveEntity -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up tellduslive sensors dynamically.""" async def async_discover_cover(device_id): diff --git a/homeassistant/components/tellstick/cover.py b/homeassistant/components/tellstick/cover.py index bb25a601a2f..65f11bdbae6 100644 --- a/homeassistant/components/tellstick/cover.py +++ b/homeassistant/components/tellstick/cover.py @@ -1,5 +1,10 @@ """Support for Tellstick covers.""" +from __future__ import annotations + from homeassistant.components.cover import CoverEntity +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import ( ATTR_DISCOVER_CONFIG, @@ -10,7 +15,12 @@ from . import ( ) -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 Tellstick covers.""" if discovery_info is None or discovery_info[ATTR_DISCOVER_DEVICES] is None: return diff --git a/homeassistant/components/velux/cover.py b/homeassistant/components/velux/cover.py index f4ea7c90708..ca529d4c572 100644 --- a/homeassistant/components/velux/cover.py +++ b/homeassistant/components/velux/cover.py @@ -1,4 +1,6 @@ """Support for Velux covers.""" +from __future__ import annotations + from pyvlx import OpeningDevice, Position from pyvlx.opening_device import Awning, Blind, GarageDoor, Gate, RollerShutter, Window @@ -16,11 +18,19 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from . import DATA_VELUX, VeluxEntity -async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): +async def async_setup_platform( + hass: HomeAssistant, + config: ConfigType, + async_add_entities: AddEntitiesCallback, + discovery_info: DiscoveryInfoType | None = None, +) -> None: """Set up cover(s) for Velux platform.""" entities = [] for node in hass.data[DATA_VELUX].pyvlx.nodes: diff --git a/homeassistant/components/xiaomi_aqara/cover.py b/homeassistant/components/xiaomi_aqara/cover.py index db41a4d719a..422d9b21e0d 100644 --- a/homeassistant/components/xiaomi_aqara/cover.py +++ b/homeassistant/components/xiaomi_aqara/cover.py @@ -1,5 +1,8 @@ """Support for Xiaomi curtain.""" from homeassistant.components.cover import ATTR_POSITION, CoverEntity +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import XiaomiDevice from .const import DOMAIN, GATEWAYS_KEY @@ -10,7 +13,11 @@ DATA_KEY_PROTO_V1 = "status" DATA_KEY_PROTO_V2 = "curtain_status" -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Perform the setup for Xiaomi devices.""" entities = [] gateway = hass.data[DOMAIN][GATEWAYS_KEY][config_entry.entry_id] diff --git a/homeassistant/components/zwave/cover.py b/homeassistant/components/zwave/cover.py index fc88db56ae7..2a49a34554b 100644 --- a/homeassistant/components/zwave/cover.py +++ b/homeassistant/components/zwave/cover.py @@ -9,8 +9,10 @@ from homeassistant.components.cover import ( CoverDeviceClass, CoverEntity, ) -from homeassistant.core import callback +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import ( CONF_INVERT_OPENCLOSE_BUTTONS, @@ -30,7 +32,11 @@ _LOGGER = logging.getLogger(__name__) SUPPORT_GARAGE = SUPPORT_OPEN | SUPPORT_CLOSE -async def async_setup_entry(hass, config_entry, async_add_entities): +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: """Set up Z-Wave Cover from Config Entry.""" @callback