jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 08:47:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Explicitly pass in the config_entry in monarch_money coordinator (#138082)

Browse Source 
explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 16:34:30 +01:00 committed by GitHub
parent fa1a03ded1
commit 63735da5a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/monarch_money/__init__.py
View File

@ -4,13 +4,10 @@ from __future__ import annotations
from typedmonarchmoney import TypedMonarchMoney
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant
from .coordinator import MonarchMoneyDataUpdateCoordinator
type MonarchMoneyConfigEntry = ConfigEntry[MonarchMoneyDataUpdateCoordinator]
from .coordinator import MonarchMoneyConfigEntry, MonarchMoneyDataUpdateCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
@ -21,7 +18,7 @@ async def async_setup_entry(
"""Set up Monarch Money from a config entry."""
monarch_client = TypedMonarchMoney(token=entry.data.get(CONF_TOKEN))
mm_coordinator = MonarchMoneyDataUpdateCoordinator(hass, monarch_client)
mm_coordinator = MonarchMoneyDataUpdateCoordinator(hass, entry, monarch_client)
await mm_coordinator.async_config_entry_first_refresh()
entry.runtime_data = mm_coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)

7
homeassistant/components/monarch_money/coordinator.py
View File

@ -30,21 +30,26 @@ class MonarchData:
cashflow_summary: MonarchCashflowSummary
type MonarchMoneyConfigEntry = ConfigEntry[MonarchMoneyDataUpdateCoordinator]
class MonarchMoneyDataUpdateCoordinator(DataUpdateCoordinator[MonarchData]):
"""Data update coordinator for Monarch Money."""
config_entry: ConfigEntry
config_entry: MonarchMoneyConfigEntry
subscription_id: str
def __init__(
self,
hass: HomeAssistant,
config_entry: MonarchMoneyConfigEntry,
client: TypedMonarchMoney,
) -> None:
"""Initialize the coordinator."""
super().__init__(
hass=hass,
logger=LOGGER,
config_entry=config_entry,
name="monarchmoney",
update_interval=timedelta(hours=4),
)

2
homeassistant/components/monarch_money/sensor.py
View File

@ -17,7 +17,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType
from . import MonarchMoneyConfigEntry
from .coordinator import MonarchMoneyConfigEntry
from .entity import MonarchMoneyAccountEntity, MonarchMoneyCashFlowEntity