Explicitly pass in the config_entry in netgear_lte coordinator (#138074)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 16:31:11 +01:00 committed by GitHub
parent 711d423877
commit 5dba229c67
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 10 additions and 11 deletions

View File

@ -6,7 +6,7 @@ from aiohttp.cookiejar import CookieJar
import eternalegypt import eternalegypt
from eternalegypt.eternalegypt import SMS from eternalegypt.eternalegypt import SMS
from homeassistant.config_entries import ConfigEntry, ConfigEntryState from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, Platform from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
@ -23,7 +23,7 @@ from .const import (
DATA_SESSION, DATA_SESSION,
DOMAIN, DOMAIN,
) )
from .coordinator import NetgearLTEDataUpdateCoordinator from .coordinator import NetgearLTEConfigEntry, NetgearLTEDataUpdateCoordinator
from .services import async_setup_services from .services import async_setup_services
EVENT_SMS = "netgear_lte_sms" EVENT_SMS = "netgear_lte_sms"
@ -55,7 +55,6 @@ PLATFORMS = [
Platform.NOTIFY, Platform.NOTIFY,
Platform.SENSOR, Platform.SENSOR,
] ]
type NetgearLTEConfigEntry = ConfigEntry[NetgearLTEDataUpdateCoordinator]
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN) CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
@ -94,7 +93,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: NetgearLTEConfigEntry) -
await modem.add_sms_listener(fire_sms_event) await modem.add_sms_listener(fire_sms_event)
coordinator = NetgearLTEDataUpdateCoordinator(hass, modem) coordinator = NetgearLTEDataUpdateCoordinator(hass, entry, modem)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator entry.runtime_data = coordinator

View File

@ -11,7 +11,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import NetgearLTEConfigEntry from .coordinator import NetgearLTEConfigEntry
from .entity import LTEEntity from .entity import LTEEntity
BINARY_SENSORS: tuple[BinarySensorEntityDescription, ...] = ( BINARY_SENSORS: tuple[BinarySensorEntityDescription, ...] = (

View File

@ -3,17 +3,16 @@
from __future__ import annotations from __future__ import annotations
from datetime import timedelta from datetime import timedelta
from typing import TYPE_CHECKING
from eternalegypt.eternalegypt import Error, Information, Modem from eternalegypt.eternalegypt import Error, Information, Modem
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN, LOGGER from .const import DOMAIN, LOGGER
if TYPE_CHECKING: type NetgearLTEConfigEntry = ConfigEntry[NetgearLTEDataUpdateCoordinator]
from . import NetgearLTEConfigEntry
class NetgearLTEDataUpdateCoordinator(DataUpdateCoordinator[Information]): class NetgearLTEDataUpdateCoordinator(DataUpdateCoordinator[Information]):
@ -24,12 +23,14 @@ class NetgearLTEDataUpdateCoordinator(DataUpdateCoordinator[Information]):
def __init__( def __init__(
self, self,
hass: HomeAssistant, hass: HomeAssistant,
config_entry: NetgearLTEConfigEntry,
modem: Modem, modem: Modem,
) -> None: ) -> None:
"""Initialize the coordinator.""" """Initialize the coordinator."""
super().__init__( super().__init__(
hass=hass, hass=hass,
logger=LOGGER, logger=LOGGER,
config_entry=config_entry,
name=DOMAIN, name=DOMAIN,
update_interval=timedelta(seconds=10), update_interval=timedelta(seconds=10),
) )

View File

@ -5,9 +5,8 @@ from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import NetgearLTEConfigEntry
from .const import DOMAIN, MANUFACTURER from .const import DOMAIN, MANUFACTURER
from .coordinator import NetgearLTEDataUpdateCoordinator from .coordinator import NetgearLTEConfigEntry, NetgearLTEDataUpdateCoordinator
class LTEEntity(CoordinatorEntity[NetgearLTEDataUpdateCoordinator]): class LTEEntity(CoordinatorEntity[NetgearLTEDataUpdateCoordinator]):

View File

@ -22,7 +22,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from . import NetgearLTEConfigEntry from .coordinator import NetgearLTEConfigEntry
from .entity import LTEEntity from .entity import LTEEntity