mirror of
https://github.com/home-assistant/core.git
synced 2025-07-17 10:17:09 +00:00
Explicitly pass in the config_entry in version coordinator (#137877)
explicitly pass in the config_entry in coordinator
This commit is contained in:
parent
a2f1501943
commit
e092937c00
@ -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],
|
||||||
|
@ -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)
|
||||||
|
@ -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,
|
||||||
)
|
)
|
||||||
|
@ -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(
|
||||||
|
@ -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
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user