Explicitly pass in the config_entry in version coordinator (#137877)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 14:25:06 +01:00 committed by GitHub
parent a2f1501943
commit e092937c00
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 10 additions and 8 deletions

View File

@ -6,7 +6,6 @@ import logging
from pyhaversion import HaVersion from pyhaversion import HaVersion
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.aiohttp_client import async_get_clientsession
@ -18,12 +17,10 @@ from .const import (
CONF_SOURCE, CONF_SOURCE,
PLATFORMS, PLATFORMS,
) )
from .coordinator import VersionDataUpdateCoordinator from .coordinator import VersionConfigEntry, VersionDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
type VersionConfigEntry = ConfigEntry[VersionDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> bool:
"""Set up the version integration from a config entry.""" """Set up the version integration from a config entry."""
@ -40,6 +37,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> b
coordinator = VersionDataUpdateCoordinator( coordinator = VersionDataUpdateCoordinator(
hass=hass, hass=hass,
config_entry=entry,
api=HaVersion( api=HaVersion(
session=async_get_clientsession(hass), session=async_get_clientsession(hass),
source=entry.data[CONF_SOURCE], source=entry.data[CONF_SOURCE],

View File

@ -13,8 +13,8 @@ from homeassistant.const import CONF_NAME, EntityCategory, __version__ as HA_VER
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 VersionConfigEntry
from .const import CONF_SOURCE, DEFAULT_NAME from .const import CONF_SOURCE, DEFAULT_NAME
from .coordinator import VersionConfigEntry
from .entity import VersionEntity from .entity import VersionEntity
HA_VERSION_OBJECT = AwesomeVersion(HA_VERSION) HA_VERSION_OBJECT = AwesomeVersion(HA_VERSION)

View File

@ -14,21 +14,25 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, UPDATE_COORDINATOR_UPDATE_INTERVAL from .const import DOMAIN, LOGGER, UPDATE_COORDINATOR_UPDATE_INTERVAL
type VersionConfigEntry = ConfigEntry[VersionDataUpdateCoordinator]
class VersionDataUpdateCoordinator(DataUpdateCoordinator[None]): class VersionDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""Data update coordinator for Version entities.""" """Data update coordinator for Version entities."""
config_entry: ConfigEntry config_entry: VersionConfigEntry
def __init__( def __init__(
self, self,
hass: HomeAssistant, hass: HomeAssistant,
config_entry: VersionConfigEntry,
api: HaVersion, api: HaVersion,
) -> 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=UPDATE_COORDINATOR_UPDATE_INTERVAL, update_interval=UPDATE_COORDINATOR_UPDATE_INTERVAL,
) )

View File

@ -9,7 +9,7 @@ from attr import asdict
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er from homeassistant.helpers import device_registry as dr, entity_registry as er
from . import VersionConfigEntry from .coordinator import VersionConfigEntry
async def async_get_config_entry_diagnostics( async def async_get_config_entry_diagnostics(

View File

@ -10,8 +10,8 @@ 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 VersionConfigEntry
from .const import CONF_SOURCE, DEFAULT_NAME from .const import CONF_SOURCE, DEFAULT_NAME
from .coordinator import VersionConfigEntry
from .entity import VersionEntity from .entity import VersionEntity