diff --git a/homeassistant/components/garages_amsterdam/__init__.py b/homeassistant/components/garages_amsterdam/__init__.py index 99d751cfcc8..854e41f2d89 100644 --- a/homeassistant/components/garages_amsterdam/__init__.py +++ b/homeassistant/components/garages_amsterdam/__init__.py @@ -4,24 +4,24 @@ from __future__ import annotations from odp_amsterdam import ODPAmsterdam -from homeassistant.config_entries import ConfigEntry from homeassistant.const import Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.aiohttp_client import async_get_clientsession -from .coordinator import GaragesAmsterdamDataUpdateCoordinator +from .coordinator import ( + GaragesAmsterdamConfigEntry, + GaragesAmsterdamDataUpdateCoordinator, +) PLATFORMS: list[Platform] = [Platform.BINARY_SENSOR, Platform.SENSOR] -type GaragesAmsterdamConfigEntry = ConfigEntry[GaragesAmsterdamDataUpdateCoordinator] - async def async_setup_entry( hass: HomeAssistant, entry: GaragesAmsterdamConfigEntry ) -> bool: """Set up Garages Amsterdam from a config entry.""" client = ODPAmsterdam(session=async_get_clientsession(hass)) - coordinator = GaragesAmsterdamDataUpdateCoordinator(hass, client) + coordinator = GaragesAmsterdamDataUpdateCoordinator(hass, entry, client) await coordinator.async_config_entry_first_refresh() diff --git a/homeassistant/components/garages_amsterdam/binary_sensor.py b/homeassistant/components/garages_amsterdam/binary_sensor.py index b93b43e1173..cf4b29f0af8 100644 --- a/homeassistant/components/garages_amsterdam/binary_sensor.py +++ b/homeassistant/components/garages_amsterdam/binary_sensor.py @@ -15,8 +15,10 @@ from homeassistant.components.binary_sensor import ( from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback -from . import GaragesAmsterdamConfigEntry -from .coordinator import GaragesAmsterdamDataUpdateCoordinator +from .coordinator import ( + GaragesAmsterdamConfigEntry, + GaragesAmsterdamDataUpdateCoordinator, +) from .entity import GaragesAmsterdamEntity diff --git a/homeassistant/components/garages_amsterdam/coordinator.py b/homeassistant/components/garages_amsterdam/coordinator.py index 3d06aba79e2..74f2361980d 100644 --- a/homeassistant/components/garages_amsterdam/coordinator.py +++ b/homeassistant/components/garages_amsterdam/coordinator.py @@ -4,24 +4,31 @@ from __future__ import annotations from odp_amsterdam import Garage, ODPAmsterdam, VehicleType +from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from .const import DOMAIN, LOGGER, SCAN_INTERVAL +type GaragesAmsterdamConfigEntry = ConfigEntry[GaragesAmsterdamDataUpdateCoordinator] + class GaragesAmsterdamDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Garage]]): """Class to manage fetching Garages Amsterdam data from single endpoint.""" + config_entry: GaragesAmsterdamConfigEntry + def __init__( self, hass: HomeAssistant, + config_entry: GaragesAmsterdamConfigEntry, client: ODPAmsterdam, ) -> None: """Initialize global Garages Amsterdam data updater.""" super().__init__( hass, LOGGER, + config_entry=config_entry, name=DOMAIN, update_interval=SCAN_INTERVAL, ) diff --git a/homeassistant/components/garages_amsterdam/sensor.py b/homeassistant/components/garages_amsterdam/sensor.py index b562fff841a..8c16260c58b 100644 --- a/homeassistant/components/garages_amsterdam/sensor.py +++ b/homeassistant/components/garages_amsterdam/sensor.py @@ -16,8 +16,10 @@ from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import StateType -from . import GaragesAmsterdamConfigEntry -from .coordinator import GaragesAmsterdamDataUpdateCoordinator +from .coordinator import ( + GaragesAmsterdamConfigEntry, + GaragesAmsterdamDataUpdateCoordinator, +) from .entity import GaragesAmsterdamEntity