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

Explicitly pass in the config_entry in PEGELONLINE coordinator init (#136773)

Browse Source
This commit is contained in:
Michael 2025-01-29 01:11:54 +01:00 committed by GitHub
parent 177bb29f69
commit 032e17720c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 19 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/pegel_online/__init__.py
View File

@ -7,21 +7,18 @@ import logging
from aiopegelonline import PegelOnline from aiopegelonline import PegelOnline
from aiopegelonline.const import CONNECT_ERRORS from aiopegelonline.const import CONNECT_ERRORS
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform from homeassistant.const import Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.aiohttp_client import async_get_clientsession
from .const import CONF_STATION from .const import CONF_STATION
from .coordinator import PegelOnlineDataUpdateCoordinator from .coordinator import PegelOnlineConfigEntry, PegelOnlineDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
PLATFORMS = [Platform.SENSOR] PLATFORMS = [Platform.SENSOR]
type PegelOnlineConfigEntry = ConfigEntry[PegelOnlineDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: PegelOnlineConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: PegelOnlineConfigEntry) -> bool:
"""Set up PEGELONLINE entry.""" """Set up PEGELONLINE entry."""
@ -35,7 +32,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PegelOnlineConfigEntry)
except CONNECT_ERRORS as err: except CONNECT_ERRORS as err:
raise ConfigEntryNotReady("Failed to connect") from err raise ConfigEntryNotReady("Failed to connect") from err
coordinator = PegelOnlineDataUpdateCoordinator(hass, entry.title, api, station) coordinator = PegelOnlineDataUpdateCoordinator(hass, entry, api, station)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
@ -46,6 +43,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: PegelOnlineConfigEntry)
return True return True
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(
hass: HomeAssistant, entry: PegelOnlineConfigEntry
) -> bool:
"""Unload PEGELONLINE entry.""" """Unload PEGELONLINE entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

14
homeassistant/components/pegel_online/coordinator.py
View File

@ -4,6 +4,7 @@ import logging
from aiopegelonline import CONNECT_ERRORS, PegelOnline, Station, StationMeasurements from aiopegelonline import CONNECT_ERRORS, PegelOnline, Station, StationMeasurements
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
@ -11,12 +12,20 @@ from .const import DOMAIN, MIN_TIME_BETWEEN_UPDATES
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
type PegelOnlineConfigEntry = ConfigEntry[PegelOnlineDataUpdateCoordinator]
class PegelOnlineDataUpdateCoordinator(DataUpdateCoordinator[StationMeasurements]): class PegelOnlineDataUpdateCoordinator(DataUpdateCoordinator[StationMeasurements]):
"""DataUpdateCoordinator for the pegel_online integration.""" """DataUpdateCoordinator for the pegel_online integration."""
config_entry: PegelOnlineConfigEntry
def __init__( def __init__(
self, hass: HomeAssistant, name: str, api: PegelOnline, station: Station self,
hass: HomeAssistant,
config_entry: PegelOnlineConfigEntry,
api: PegelOnline,
station: Station,
) -> None: ) -> None:
"""Initialize the PegelOnlineDataUpdateCoordinator.""" """Initialize the PegelOnlineDataUpdateCoordinator."""
self.api = api self.api = api
@ -24,7 +33,8 @@ class PegelOnlineDataUpdateCoordinator(DataUpdateCoordinator[StationMeasurements
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
name=name, config_entry=config_entry,
name=config_entry.title,
update_interval=MIN_TIME_BETWEEN_UPDATES, update_interval=MIN_TIME_BETWEEN_UPDATES,
) )

2
homeassistant/components/pegel_online/diagnostics.py
View File

@ -6,7 +6,7 @@ from typing import Any
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from . import PegelOnlineConfigEntry from .coordinator import PegelOnlineConfigEntry
async def async_get_config_entry_diagnostics( async def async_get_config_entry_diagnostics(

3
homeassistant/components/pegel_online/sensor.py
View File

@ -16,8 +16,7 @@ from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import PegelOnlineConfigEntry from .coordinator import PegelOnlineConfigEntry, PegelOnlineDataUpdateCoordinator
from .coordinator import PegelOnlineDataUpdateCoordinator
from .entity import PegelOnlineEntity from .entity import PegelOnlineEntity