Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Explicitly pass in the config_entry in srp_energy coordinator (#137933) 

explicitly pass in the config_entry in coordinator
pull/138061/head
Michael 2025-02-09 14:54:26 +01:00 committed by GitHub
parent 9c5928c2d0
commit 51d3e449ab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/components/srp_energy/__init__.py
View File

@ -6,7 +6,7 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID, CONF_PASSWORD, CONF_USERNAME, Platform from homeassistant.const import CONF_ID, CONF_PASSWORD, CONF_USERNAME, Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from .const import CONF_IS_TOU, DOMAIN, LOGGER from .const import DOMAIN, LOGGER
from .coordinator import SRPEnergyDataUpdateCoordinator from .coordinator import SRPEnergyDataUpdateCoordinator
PLATFORMS = [Platform.SENSOR] PLATFORMS = [Platform.SENSOR]
@ -26,9 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
api_password, api_password,
) )
coordinator = SRPEnergyDataUpdateCoordinator( coordinator = SRPEnergyDataUpdateCoordinator(hass, entry, api_instance)
hass, api_instance, entry.data[CONF_IS_TOU]
)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()

13
homeassistant/components/srp_energy/coordinator.py
View File

@ -12,7 +12,13 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
from .const import DOMAIN, LOGGER, MIN_TIME_BETWEEN_UPDATES, PHOENIX_TIME_ZONE from .const import (
CONF_IS_TOU,
DOMAIN,
LOGGER,
MIN_TIME_BETWEEN_UPDATES,
PHOENIX_TIME_ZONE,
)
TIMEOUT = 10 TIMEOUT = 10
PHOENIX_ZONE_INFO = dt_util.get_time_zone(PHOENIX_TIME_ZONE) PHOENIX_ZONE_INFO = dt_util.get_time_zone(PHOENIX_TIME_ZONE)
@ -24,14 +30,15 @@ class SRPEnergyDataUpdateCoordinator(DataUpdateCoordinator[float]):
config_entry: ConfigEntry config_entry: ConfigEntry
def __init__( def __init__(
self, hass: HomeAssistant, client: SrpEnergyClient, is_time_of_use: bool self, hass: HomeAssistant, config_entry: ConfigEntry, client: SrpEnergyClient
) -> None: ) -> None:
"""Initialize the srp_energy data coordinator.""" """Initialize the srp_energy data coordinator."""
self._client = client self._client = client
self._is_time_of_use = is_time_of_use self._is_time_of_use = config_entry.data[CONF_IS_TOU]
super().__init__( super().__init__(
hass, hass,
LOGGER, LOGGER,
config_entry=config_entry,
name=DOMAIN, name=DOMAIN,
update_interval=MIN_TIME_BETWEEN_UPDATES, update_interval=MIN_TIME_BETWEEN_UPDATES,
) )