Explicitly pass in the config_entry in garages_amsterdam coordinator (#137829)

explicitly pass in the config_entry in coordinator
pull/131282/head
Michael 2025-02-08 13:17:42 +01:00 committed by GitHub
parent 13f6f045f5
commit 239408aa5d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 20 additions and 9 deletions

View File

@ -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()

View File

@ -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

View File

@ -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,
)

View File

@ -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