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

Explicitly pass in the config_entry in sensoterra coordinator (#137957)

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

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

@ -4,16 +4,13 @@ from __future__ import annotations
from sensoterra.customerapi import CustomerApi from sensoterra.customerapi import CustomerApi
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from .coordinator import SensoterraCoordinator from .coordinator import SensoterraConfigEntry, SensoterraCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR] PLATFORMS: list[Platform] = [Platform.SENSOR]
type SensoterraConfigEntry = ConfigEntry[SensoterraCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: SensoterraConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: SensoterraConfigEntry) -> bool:
"""Set up Sensoterra platform based on a configuration entry.""" """Set up Sensoterra platform based on a configuration entry."""
@ -24,7 +21,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: SensoterraConfigEntry) -
api.set_language(hass.config.language) api.set_language(hass.config.language)
api.set_token(entry.data[CONF_TOKEN]) api.set_token(entry.data[CONF_TOKEN])
coordinator = SensoterraCoordinator(hass, api) coordinator = SensoterraCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator entry.runtime_data = coordinator

10
homeassistant/components/sensoterra/coordinator.py
View File

@ -10,21 +10,29 @@ from sensoterra.customerapi import (
) )
from sensoterra.probe import Probe, Sensor from sensoterra.probe import Probe, Sensor
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError from homeassistant.exceptions import ConfigEntryError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import LOGGER, SCAN_INTERVAL_MINUTES from .const import LOGGER, SCAN_INTERVAL_MINUTES
type SensoterraConfigEntry = ConfigEntry[SensoterraCoordinator]
class SensoterraCoordinator(DataUpdateCoordinator[list[Probe]]): class SensoterraCoordinator(DataUpdateCoordinator[list[Probe]]):
"""Sensoterra coordinator.""" """Sensoterra coordinator."""
def __init__(self, hass: HomeAssistant, api: CustomerApi) -> None: config_entry: SensoterraConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: SensoterraConfigEntry, api: CustomerApi
) -> None:
"""Initialize Sensoterra coordinator.""" """Initialize Sensoterra coordinator."""
super().__init__( super().__init__(
hass, hass,
LOGGER, LOGGER,
config_entry=config_entry,
name="Sensoterra probe", name="Sensoterra probe",
update_interval=timedelta(minutes=SCAN_INTERVAL_MINUTES), update_interval=timedelta(minutes=SCAN_INTERVAL_MINUTES),
) )

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

@ -25,9 +25,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import SensoterraConfigEntry
from .const import CONFIGURATION_URL, DOMAIN, SENSOR_EXPIRATION_DAYS from .const import CONFIGURATION_URL, DOMAIN, SENSOR_EXPIRATION_DAYS
from .coordinator import SensoterraCoordinator from .coordinator import SensoterraConfigEntry, SensoterraCoordinator
class ProbeSensorType(StrEnum): class ProbeSensorType(StrEnum):