Pass config entry to SMS coordinator (#149955)

This commit is contained in:
Joost Lekkerkerker 2025-08-04 12:49:26 +02:00 committed by GitHub
parent 93e11aa8bc
commit 9edd242734
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions

View File

@ -83,8 +83,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if not gateway: if not gateway:
raise ConfigEntryNotReady(f"Cannot find device {device}") raise ConfigEntryNotReady(f"Cannot find device {device}")
signal_coordinator = SignalCoordinator(hass, gateway) signal_coordinator = SignalCoordinator(hass, entry, gateway)
network_coordinator = NetworkCoordinator(hass, gateway) network_coordinator = NetworkCoordinator(hass, entry, gateway)
# Fetch initial data so we have data when entities subscribe # Fetch initial data so we have data when entities subscribe
await signal_coordinator.async_config_entry_first_refresh() await signal_coordinator.async_config_entry_first_refresh()

View File

@ -16,13 +16,14 @@ _LOGGER = logging.getLogger(__name__)
class SignalCoordinator(DataUpdateCoordinator): class SignalCoordinator(DataUpdateCoordinator):
"""Signal strength coordinator.""" """Signal strength coordinator."""
def __init__(self, hass, gateway): def __init__(self, hass, entry, gateway):
"""Initialize signal strength coordinator.""" """Initialize signal strength coordinator."""
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
name="Device signal state", name="Device signal state",
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL), update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
config_entry=entry,
) )
self._gateway = gateway self._gateway = gateway
@ -38,13 +39,14 @@ class SignalCoordinator(DataUpdateCoordinator):
class NetworkCoordinator(DataUpdateCoordinator): class NetworkCoordinator(DataUpdateCoordinator):
"""Network info coordinator.""" """Network info coordinator."""
def __init__(self, hass, gateway): def __init__(self, hass, entry, gateway):
"""Initialize network info coordinator.""" """Initialize network info coordinator."""
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
name="Device network state", name="Device network state",
update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL), update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL),
config_entry=entry,
) )
self._gateway = gateway self._gateway = gateway