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

Explicitly pass in the config_entry in sense coordinator (#137958)

Browse Source 
explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 15:33:43 +01:00 committed by GitHub
parent 7fec225e79
commit 71d47aef2e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 31 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/sense/__init__.py
View File

@ -89,8 +89,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: SenseConfigEntry) -> boo
except SENSE_WEBSOCKET_EXCEPTIONS as err:
raise ConfigEntryNotReady(str(err) or "Error during realtime update") from err
trends_coordinator = SenseTrendCoordinator(hass, gateway)
realtime_coordinator = SenseRealtimeCoordinator(hass, gateway)
trends_coordinator = SenseTrendCoordinator(hass, entry, gateway)
realtime_coordinator = SenseRealtimeCoordinator(hass, entry, gateway)
# This can take longer than 60s and we already know
# sense is online since get_discovered_device_data was

34
homeassistant/components/sense/coordinator.py
View File

@ -1,7 +1,10 @@
"""Sense Coordinators."""
from __future__ import annotations
from datetime import timedelta
import logging
from typing import TYPE_CHECKING
from sense_energy import (
ASyncSenseable,
@ -13,6 +16,9 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
if TYPE_CHECKING:
from . import SenseConfigEntry
from .const import (
ACTIVE_UPDATE_RATE,
SENSE_CONNECT_EXCEPTIONS,
@ -27,13 +33,21 @@ _LOGGER = logging.getLogger(__name__)
class SenseCoordinator(DataUpdateCoordinator[None]):
"""Sense Trend Coordinator."""
config_entry: SenseConfigEntry
def __init__(
self, hass: HomeAssistant, gateway: ASyncSenseable, name: str, update: int
self,
hass: HomeAssistant,
config_entry: SenseConfigEntry,
gateway: ASyncSenseable,
name: str,
update: int,
) -> None:
"""Initialize."""
super().__init__(
hass,
logger=_LOGGER,
config_entry=config_entry,
name=f"Sense {name} {gateway.sense_monitor_id}",
update_interval=timedelta(seconds=update),
)
@ -44,9 +58,14 @@ class SenseCoordinator(DataUpdateCoordinator[None]):
class SenseTrendCoordinator(SenseCoordinator):
"""Sense Trend Coordinator."""
def __init__(self, hass: HomeAssistant, gateway: ASyncSenseable) -> None:
def __init__(
self,
hass: HomeAssistant,
config_entry: SenseConfigEntry,
gateway: ASyncSenseable,
) -> None:
"""Initialize."""
super().__init__(hass, gateway, "Trends", TREND_UPDATE_RATE)
super().__init__(hass, config_entry, gateway, "Trends", TREND_UPDATE_RATE)
async def _async_update_data(self) -> None:
"""Update the trend data."""
@ -62,9 +81,14 @@ class SenseTrendCoordinator(SenseCoordinator):
class SenseRealtimeCoordinator(SenseCoordinator):
"""Sense Realtime Coordinator."""
def __init__(self, hass: HomeAssistant, gateway: ASyncSenseable) -> None:
def __init__(
self,
hass: HomeAssistant,
config_entry: SenseConfigEntry,
gateway: ASyncSenseable,
) -> None:
"""Initialize."""
super().__init__(hass, gateway, "Realtime", ACTIVE_UPDATE_RATE)
super().__init__(hass, config_entry, gateway, "Realtime", ACTIVE_UPDATE_RATE)
async def _async_update_data(self) -> None:
"""Retrieve latest state."""