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