diff --git a/homeassistant/components/tolo/__init__.py b/homeassistant/components/tolo/__init__.py index d2a43ef525b..bbd17cc8b13 100644 --- a/homeassistant/components/tolo/__init__.py +++ b/homeassistant/components/tolo/__init__.py @@ -2,12 +2,10 @@ from __future__ import annotations -from homeassistant.config_entries import ConfigEntry from homeassistant.const import Platform from homeassistant.core import HomeAssistant -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator PLATFORMS = [ Platform.BINARY_SENSOR, @@ -22,21 +20,17 @@ PLATFORMS = [ ] -async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: +async def async_setup_entry(hass: HomeAssistant, entry: ToloConfigEntry) -> bool: """Set up tolo from a config entry.""" coordinator = ToloSaunaUpdateCoordinator(hass, entry) await coordinator.async_config_entry_first_refresh() - hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator + entry.runtime_data = coordinator await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS) return True -async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: +async def async_unload_entry(hass: HomeAssistant, entry: ToloConfigEntry) -> bool: """Unload a config entry.""" - unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS) - if unload_ok: - hass.data[DOMAIN].pop(entry.entry_id) - - return unload_ok + return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) diff --git a/homeassistant/components/tolo/binary_sensor.py b/homeassistant/components/tolo/binary_sensor.py index cb3ba46b604..0b94c60094f 100644 --- a/homeassistant/components/tolo/binary_sensor.py +++ b/homeassistant/components/tolo/binary_sensor.py @@ -4,23 +4,21 @@ from homeassistant.components.binary_sensor import ( BinarySensorDeviceClass, BinarySensorEntity, ) -from homeassistant.config_entries import ConfigEntry from homeassistant.const import EntityCategory from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up binary sensors for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities( [ ToloFlowInBinarySensor(coordinator, entry), @@ -37,7 +35,7 @@ class ToloFlowInBinarySensor(ToloSaunaCoordinatorEntity, BinarySensorEntity): _attr_device_class = BinarySensorDeviceClass.OPENING def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize TOLO Water In Valve entity.""" super().__init__(coordinator, entry) @@ -58,7 +56,7 @@ class ToloFlowOutBinarySensor(ToloSaunaCoordinatorEntity, BinarySensorEntity): _attr_device_class = BinarySensorDeviceClass.OPENING def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize TOLO Water Out Valve entity.""" super().__init__(coordinator, entry) diff --git a/homeassistant/components/tolo/button.py b/homeassistant/components/tolo/button.py index 9e4c8c84be9..472abdcb673 100644 --- a/homeassistant/components/tolo/button.py +++ b/homeassistant/components/tolo/button.py @@ -3,23 +3,21 @@ from tololib import LampMode from homeassistant.components.button import ButtonEntity -from homeassistant.config_entries import ConfigEntry from homeassistant.const import EntityCategory from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up buttons for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities( [ ToloLampNextColorButton(coordinator, entry), @@ -34,7 +32,7 @@ class ToloLampNextColorButton(ToloSaunaCoordinatorEntity, ButtonEntity): _attr_translation_key = "next_color" def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize lamp next color button entity.""" super().__init__(coordinator, entry) diff --git a/homeassistant/components/tolo/climate.py b/homeassistant/components/tolo/climate.py index 0df8635fca9..ed7ab0c3b76 100644 --- a/homeassistant/components/tolo/climate.py +++ b/homeassistant/components/tolo/climate.py @@ -20,23 +20,21 @@ from homeassistant.components.climate import ( HVACAction, HVACMode, ) -from homeassistant.config_entries import ConfigEntry from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTemperature from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up climate controls for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities([SaunaClimate(coordinator, entry)]) @@ -62,7 +60,7 @@ class SaunaClimate(ToloSaunaCoordinatorEntity, ClimateEntity): _attr_temperature_unit = UnitOfTemperature.CELSIUS def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize TOLO Sauna Climate entity.""" super().__init__(coordinator, entry) diff --git a/homeassistant/components/tolo/coordinator.py b/homeassistant/components/tolo/coordinator.py index 729073b16c4..372c67a4260 100644 --- a/homeassistant/components/tolo/coordinator.py +++ b/homeassistant/components/tolo/coordinator.py @@ -17,6 +17,8 @@ from .const import DEFAULT_RETRY_COUNT, DEFAULT_RETRY_TIMEOUT _LOGGER = logging.getLogger(__name__) +type ToloConfigEntry = ConfigEntry[ToloSaunaUpdateCoordinator] + class ToloSaunaData(NamedTuple): """Compound class for reflecting full state (status and info) of a TOLO Sauna.""" @@ -28,9 +30,9 @@ class ToloSaunaData(NamedTuple): class ToloSaunaUpdateCoordinator(DataUpdateCoordinator[ToloSaunaData]): """DataUpdateCoordinator for TOLO Sauna.""" - config_entry: ConfigEntry + config_entry: ToloConfigEntry - def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: + def __init__(self, hass: HomeAssistant, entry: ToloConfigEntry) -> None: """Initialize ToloSaunaUpdateCoordinator.""" self.client = ToloClient( address=entry.data[CONF_HOST], diff --git a/homeassistant/components/tolo/entity.py b/homeassistant/components/tolo/entity.py index 261cfc7cb0c..c6aef0fb824 100644 --- a/homeassistant/components/tolo/entity.py +++ b/homeassistant/components/tolo/entity.py @@ -2,12 +2,11 @@ from __future__ import annotations -from homeassistant.config_entries import ConfigEntry from homeassistant.helpers.device_registry import DeviceInfo from homeassistant.helpers.update_coordinator import CoordinatorEntity from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator class ToloSaunaCoordinatorEntity(CoordinatorEntity[ToloSaunaUpdateCoordinator]): @@ -16,7 +15,7 @@ class ToloSaunaCoordinatorEntity(CoordinatorEntity[ToloSaunaUpdateCoordinator]): _attr_has_entity_name = True def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize ToloSaunaCoordinatorEntity.""" super().__init__(coordinator) diff --git a/homeassistant/components/tolo/fan.py b/homeassistant/components/tolo/fan.py index 7bddf775143..41ca94055ba 100644 --- a/homeassistant/components/tolo/fan.py +++ b/homeassistant/components/tolo/fan.py @@ -5,22 +5,20 @@ from __future__ import annotations from typing import Any from homeassistant.components.fan import FanEntity, FanEntityFeature -from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up fan controls for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities([ToloFan(coordinator, entry)]) @@ -31,7 +29,7 @@ class ToloFan(ToloSaunaCoordinatorEntity, FanEntity): _attr_supported_features = FanEntityFeature.TURN_OFF | FanEntityFeature.TURN_ON def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize TOLO fan entity.""" super().__init__(coordinator, entry) diff --git a/homeassistant/components/tolo/light.py b/homeassistant/components/tolo/light.py index 9ccd4a8e407..25e1e913544 100644 --- a/homeassistant/components/tolo/light.py +++ b/homeassistant/components/tolo/light.py @@ -5,22 +5,20 @@ from __future__ import annotations from typing import Any from homeassistant.components.light import ColorMode, LightEntity -from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up light controls for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities([ToloLight(coordinator, entry)]) @@ -32,7 +30,7 @@ class ToloLight(ToloSaunaCoordinatorEntity, LightEntity): _attr_supported_color_modes = {ColorMode.ONOFF} def __init__( - self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry + self, coordinator: ToloSaunaUpdateCoordinator, entry: ToloConfigEntry ) -> None: """Initialize TOLO Sauna Light entity.""" super().__init__(coordinator, entry) diff --git a/homeassistant/components/tolo/number.py b/homeassistant/components/tolo/number.py index 902fb749d23..db06b82d002 100644 --- a/homeassistant/components/tolo/number.py +++ b/homeassistant/components/tolo/number.py @@ -15,13 +15,11 @@ from tololib import ( ) from homeassistant.components.number import NumberEntity, NumberEntityDescription -from homeassistant.config_entries import ConfigEntry from homeassistant.const import EntityCategory, UnitOfTime from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity @@ -67,11 +65,11 @@ NUMBERS = ( async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up number controls for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities( ToloNumberEntity(coordinator, entry, description) for description in NUMBERS ) @@ -85,7 +83,7 @@ class ToloNumberEntity(ToloSaunaCoordinatorEntity, NumberEntity): def __init__( self, coordinator: ToloSaunaUpdateCoordinator, - entry: ConfigEntry, + entry: ToloConfigEntry, entity_description: ToloNumberEntityDescription, ) -> None: """Initialize TOLO Number entity.""" diff --git a/homeassistant/components/tolo/select.py b/homeassistant/components/tolo/select.py index b08f37e40ae..f487fba9664 100644 --- a/homeassistant/components/tolo/select.py +++ b/homeassistant/components/tolo/select.py @@ -8,13 +8,12 @@ from dataclasses import dataclass from tololib import ToloClient, ToloSettings from homeassistant.components.select import SelectEntity, SelectEntityDescription -from homeassistant.config_entries import ConfigEntry from homeassistant.const import EntityCategory from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN, AromaTherapySlot, LampMode -from .coordinator import ToloSaunaUpdateCoordinator +from .const import AromaTherapySlot, LampMode +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity @@ -53,11 +52,11 @@ SELECTS = ( async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up select entities for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities( ToloSelectEntity(coordinator, entry, description) for description in SELECTS ) @@ -73,7 +72,7 @@ class ToloSelectEntity(ToloSaunaCoordinatorEntity, SelectEntity): def __init__( self, coordinator: ToloSaunaUpdateCoordinator, - entry: ConfigEntry, + entry: ToloConfigEntry, entity_description: ToloSelectEntityDescription, ) -> None: """Initialize TOLO select entity.""" diff --git a/homeassistant/components/tolo/sensor.py b/homeassistant/components/tolo/sensor.py index e97211c8e40..ba203dec806 100644 --- a/homeassistant/components/tolo/sensor.py +++ b/homeassistant/components/tolo/sensor.py @@ -13,7 +13,6 @@ from homeassistant.components.sensor import ( SensorEntityDescription, SensorStateClass, ) -from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( PERCENTAGE, EntityCategory, @@ -23,8 +22,7 @@ from homeassistant.const import ( from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity @@ -88,11 +86,11 @@ SENSORS = ( async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up (non-binary, general) sensors for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities( ToloSensorEntity(coordinator, entry, description) for description in SENSORS ) @@ -106,7 +104,7 @@ class ToloSensorEntity(ToloSaunaCoordinatorEntity, SensorEntity): def __init__( self, coordinator: ToloSaunaUpdateCoordinator, - entry: ConfigEntry, + entry: ToloConfigEntry, entity_description: ToloSensorEntityDescription, ) -> None: """Initialize TOLO Number entity.""" diff --git a/homeassistant/components/tolo/switch.py b/homeassistant/components/tolo/switch.py index ce863053e26..686f78b04e9 100644 --- a/homeassistant/components/tolo/switch.py +++ b/homeassistant/components/tolo/switch.py @@ -9,12 +9,10 @@ from typing import Any from tololib import ToloClient, ToloStatus from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription -from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback -from .const import DOMAIN -from .coordinator import ToloSaunaUpdateCoordinator +from .coordinator import ToloConfigEntry, ToloSaunaUpdateCoordinator from .entity import ToloSaunaCoordinatorEntity @@ -44,11 +42,11 @@ SWITCHES = ( async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: ToloConfigEntry, async_add_entities: AddConfigEntryEntitiesCallback, ) -> None: """Set up switch controls for TOLO Sauna.""" - coordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data async_add_entities( ToloSwitchEntity(coordinator, entry, description) for description in SWITCHES ) @@ -62,7 +60,7 @@ class ToloSwitchEntity(ToloSaunaCoordinatorEntity, SwitchEntity): def __init__( self, coordinator: ToloSaunaUpdateCoordinator, - entry: ConfigEntry, + entry: ToloConfigEntry, entity_description: ToloSwitchEntityDescription, ) -> None: """Initialize TOLO switch entity."""