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

Explicitly pass in the config_entry in rympro coordinator init (#137464)

Browse Source 
explicitly pass in the config_entry in coordinator init
This commit is contained in:
Michael 2025-02-05 22:00:50 +01:00 committed by GitHub
parent db6bd6aad1
commit 2aea078d9a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/rympro/__init__.py
View File

@ -38,7 +38,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
data={**data, CONF_TOKEN: token}, data={**data, CONF_TOKEN: token},
) )
coordinator = RymProDataUpdateCoordinator(hass, rympro) coordinator = RymProDataUpdateCoordinator(hass, entry, rympro)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {}) hass.data.setdefault(DOMAIN, {})

9
homeassistant/components/rympro/coordinator.py
View File

@ -7,6 +7,7 @@ import logging
from pyrympro import CannotConnectError, OperationError, RymPro, UnauthorizedError from pyrympro import CannotConnectError, OperationError, RymPro, UnauthorizedError
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -20,13 +21,18 @@ _LOGGER = logging.getLogger(__name__)
class RymProDataUpdateCoordinator(DataUpdateCoordinator[dict[int, dict]]): class RymProDataUpdateCoordinator(DataUpdateCoordinator[dict[int, dict]]):
"""Class to manage fetching RYM Pro data.""" """Class to manage fetching RYM Pro data."""
def __init__(self, hass: HomeAssistant, rympro: RymPro) -> None: config_entry: ConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: ConfigEntry, rympro: RymPro
) -> None:
"""Initialize global RymPro data updater.""" """Initialize global RymPro data updater."""
self.rympro = rympro self.rympro = rympro
interval = timedelta(seconds=SCAN_INTERVAL) interval = timedelta(seconds=SCAN_INTERVAL)
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
config_entry=config_entry,
name=DOMAIN, name=DOMAIN,
update_interval=interval, update_interval=interval,
) )
@ -40,7 +46,6 @@ class RymProDataUpdateCoordinator(DataUpdateCoordinator[dict[int, dict]]):
meter_id meter_id
) )
except UnauthorizedError as error: except UnauthorizedError as error:
assert self.config_entry
await self.hass.config_entries.async_reload(self.config_entry.entry_id) await self.hass.config_entries.async_reload(self.config_entry.entry_id)
raise UpdateFailed(error) from error raise UpdateFailed(error) from error
except (CannotConnectError, OperationError) as error: except (CannotConnectError, OperationError) as error: