diff --git a/homeassistant/components/acmeda/cover.py b/homeassistant/components/acmeda/cover.py index 6c1de528abe..8c934e51771 100644 --- a/homeassistant/components/acmeda/cover.py +++ b/homeassistant/components/acmeda/cover.py @@ -1,4 +1,6 @@ """Support for Acmeda Roller Blinds.""" +from __future__ import annotations + from homeassistant.components.cover import ( ATTR_POSITION, SUPPORT_CLOSE, @@ -11,19 +13,25 @@ from homeassistant.components.cover import ( SUPPORT_STOP_TILT, 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 .base import AcmedaBase from .const import ACMEDA_HUB_UPDATE, DOMAIN from .helpers import async_add_acmeda_entities -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 Acmeda Rollers from a config entry.""" hub = hass.data[DOMAIN][config_entry.entry_id] - current = set() + current: set[int] = set() @callback def async_add_acmeda_covers(): diff --git a/homeassistant/components/acmeda/helpers.py b/homeassistant/components/acmeda/helpers.py index 1162aba5dc8..a1a262be77e 100644 --- a/homeassistant/components/acmeda/helpers.py +++ b/homeassistant/components/acmeda/helpers.py @@ -1,13 +1,21 @@ """Helper functions for Acmeda Pulse.""" -from homeassistant.core import callback +from __future__ import annotations + +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.device_registry import async_get_registry as get_dev_reg +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import DOMAIN, LOGGER @callback def async_add_acmeda_entities( - hass, entity_class, config_entry, current, async_add_entities + hass: HomeAssistant, + entity_class: type, + config_entry: ConfigEntry, + current: set[int], + async_add_entities: AddEntitiesCallback, ): """Add any new entities.""" hub = hass.data[DOMAIN][config_entry.entry_id] @@ -26,7 +34,7 @@ def async_add_acmeda_entities( async_add_entities(new_items) -async def update_devices(hass, config_entry, api): +async def update_devices(hass: HomeAssistant, config_entry: ConfigEntry, api): """Tell hass that device info has been updated.""" dev_registry = await get_dev_reg(hass) diff --git a/homeassistant/components/acmeda/sensor.py b/homeassistant/components/acmeda/sensor.py index 57e5b50bd1f..88a0886a84e 100644 --- a/homeassistant/components/acmeda/sensor.py +++ b/homeassistant/components/acmeda/sensor.py @@ -1,19 +1,27 @@ """Support for Acmeda Roller Blind Batteries.""" +from __future__ import annotations + from homeassistant.components.sensor import SensorDeviceClass, SensorEntity +from homeassistant.config_entries import ConfigEntry from homeassistant.const import PERCENTAGE -from homeassistant.core import callback +from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.dispatcher import async_dispatcher_connect +from homeassistant.helpers.entity_platform import AddEntitiesCallback from .base import AcmedaBase from .const import ACMEDA_HUB_UPDATE, DOMAIN from .helpers import async_add_acmeda_entities -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 Acmeda Rollers from a config entry.""" hub = hass.data[DOMAIN][config_entry.entry_id] - current = set() + current: set[int] = set() @callback def async_add_acmeda_sensors():