Explicitly pass in the config_entry in tradfri coordinator (#137892)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 14:26:27 +01:00 committed by GitHub
parent cce03d2ee7
commit 7986e0fec1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 2 deletions

View File

@ -106,7 +106,7 @@ async def async_setup_entry(
for device in devices:
coordinator = TradfriDeviceDataUpdateCoordinator(
hass=hass, api=api, device=device
hass=hass, config_entry=entry, api=api, device=device
)
await coordinator.async_config_entry_first_refresh()

View File

@ -10,6 +10,7 @@ from pytradfri.command import Command
from pytradfri.device import Device
from pytradfri.error import RequestError
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -21,10 +22,12 @@ SCAN_INTERVAL = 60 # Interval for updating the coordinator
class TradfriDeviceDataUpdateCoordinator(DataUpdateCoordinator[Device]):
"""Coordinator to manage data for a specific Tradfri device."""
config_entry: ConfigEntry
def __init__(
self,
hass: HomeAssistant,
*,
config_entry: ConfigEntry,
api: Callable[[Command | list[Command]], Any],
device: Device,
) -> None:
@ -36,6 +39,7 @@ class TradfriDeviceDataUpdateCoordinator(DataUpdateCoordinator[Device]):
super().__init__(
hass,
LOGGER,
config_entry=config_entry,
name=f"Update coordinator for {device}",
update_interval=timedelta(seconds=SCAN_INTERVAL),
)