diff --git a/homeassistant/components/lg_thinq/__init__.py b/homeassistant/components/lg_thinq/__init__.py index 657524f0ef5..72d81af4ff0 100644 --- a/homeassistant/components/lg_thinq/__init__.py +++ b/homeassistant/components/lg_thinq/__init__.py @@ -100,7 +100,7 @@ async def async_setup_coordinators( # Setup coordinator per device. task_list = [ - hass.async_create_task(async_setup_device_coordinator(hass, bridge)) + hass.async_create_task(async_setup_device_coordinator(hass, entry, bridge)) for bridge in bridge_list ] task_result = await asyncio.gather(*task_list) diff --git a/homeassistant/components/lg_thinq/coordinator.py b/homeassistant/components/lg_thinq/coordinator.py index 9f317dc21d9..d6991d15297 100644 --- a/homeassistant/components/lg_thinq/coordinator.py +++ b/homeassistant/components/lg_thinq/coordinator.py @@ -3,7 +3,7 @@ from __future__ import annotations import logging -from typing import Any +from typing import TYPE_CHECKING, Any from thinqconnect import ThinQAPIException from thinqconnect.integration import HABridge @@ -11,6 +11,9 @@ from thinqconnect.integration import HABridge from homeassistant.core import HomeAssistant from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed +if TYPE_CHECKING: + from . import ThinqConfigEntry + from .const import DOMAIN _LOGGER = logging.getLogger(__name__) @@ -19,11 +22,16 @@ _LOGGER = logging.getLogger(__name__) class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]): """LG Device's Data Update Coordinator.""" - def __init__(self, hass: HomeAssistant, ha_bridge: HABridge) -> None: + config_entry: ThinqConfigEntry + + def __init__( + self, hass: HomeAssistant, config_entry: ThinqConfigEntry, ha_bridge: HABridge + ) -> None: """Initialize data coordinator.""" super().__init__( hass, _LOGGER, + config_entry=config_entry, name=f"{DOMAIN}_{ha_bridge.device.device_id}", ) @@ -71,10 +79,10 @@ class DeviceDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]): async def async_setup_device_coordinator( - hass: HomeAssistant, ha_bridge: HABridge + hass: HomeAssistant, config_entry: ThinqConfigEntry, ha_bridge: HABridge ) -> DeviceDataUpdateCoordinator: """Create DeviceDataUpdateCoordinator and device_api per device.""" - coordinator = DeviceDataUpdateCoordinator(hass, ha_bridge) + coordinator = DeviceDataUpdateCoordinator(hass, config_entry, ha_bridge) await coordinator.async_refresh() _LOGGER.debug(