Explicitly pass in the config_entry in simplefin coordinator (#137948)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-08 20:40:51 +01:00 committed by GitHub
parent 743873b2e9
commit 22d6913bc5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 12 additions and 11 deletions

View File

@ -4,12 +4,11 @@ from __future__ import annotations
from simplefin4py import SimpleFin
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from .const import CONF_ACCESS_URL
from .coordinator import SimpleFinDataUpdateCoordinator
from .coordinator import SimpleFinConfigEntry, SimpleFinDataUpdateCoordinator
PLATFORMS: list[str] = [
Platform.BINARY_SENSOR,
@ -17,20 +16,17 @@ PLATFORMS: list[str] = [
]
type SimpleFinConfigEntry = ConfigEntry[SimpleFinDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: SimpleFinConfigEntry) -> bool:
"""Set up from a config entry."""
access_url = entry.data[CONF_ACCESS_URL]
sf_client = SimpleFin(access_url)
sf_coordinator = SimpleFinDataUpdateCoordinator(hass, sf_client)
sf_coordinator = SimpleFinDataUpdateCoordinator(hass, entry, sf_client)
await sf_coordinator.async_config_entry_first_refresh()
entry.runtime_data = sf_coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: SimpleFinConfigEntry) -> bool:
"""Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

View File

@ -14,7 +14,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import SimpleFinConfigEntry
from .coordinator import SimpleFinConfigEntry
from .entity import SimpleFinEntity

View File

@ -15,17 +15,22 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import LOGGER
type SimpleFinConfigEntry = ConfigEntry[SimpleFinDataUpdateCoordinator]
class SimpleFinDataUpdateCoordinator(DataUpdateCoordinator[FinancialData]):
"""Data update coordinator for the SimpleFIN integration."""
config_entry: ConfigEntry
config_entry: SimpleFinConfigEntry
def __init__(self, hass: HomeAssistant, client: SimpleFin) -> None:
def __init__(
self, hass: HomeAssistant, config_entry: SimpleFinConfigEntry, client: SimpleFin
) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
config_entry=config_entry,
name="simplefin",
update_interval=timedelta(hours=4),
)

View File

@ -19,7 +19,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import SimpleFinConfigEntry
from .coordinator import SimpleFinConfigEntry
from .entity import SimpleFinEntity