Explicitly pass in the config_entry in garages_amsterdam coordinator (#137829)
explicitly pass in the config_entry in coordinatorpull/131282/head
parent
13f6f045f5
commit
239408aa5d
|
@ -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()
|
||||
|
||||
|
|
|
@ -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
|
||||
|
||||
|
||||
|
|
|
@ -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,
|
||||
)
|
||||
|
|
|
@ -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
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue