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

explicitly pass in the config_entry in coordinator
pull/138078/head
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

View File

@ -4,16 +4,13 @@ from __future__ import annotations
from sensoterra.customerapi import CustomerApi
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_TOKEN, Platform
from homeassistant.core import HomeAssistant
from .coordinator import SensoterraCoordinator
from .coordinator import SensoterraConfigEntry, SensoterraCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
type SensoterraConfigEntry = ConfigEntry[SensoterraCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: SensoterraConfigEntry) -> bool:
"""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_token(entry.data[CONF_TOKEN])
coordinator = SensoterraCoordinator(hass, api)
coordinator = SensoterraCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator

View File

@ -10,21 +10,29 @@ from sensoterra.customerapi import (
)
from sensoterra.probe import Probe, Sensor
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryError
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import LOGGER, SCAN_INTERVAL_MINUTES
type SensoterraConfigEntry = ConfigEntry[SensoterraCoordinator]
class SensoterraCoordinator(DataUpdateCoordinator[list[Probe]]):
"""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."""
super().__init__(
hass,
LOGGER,
config_entry=config_entry,
name="Sensoterra probe",
update_interval=timedelta(minutes=SCAN_INTERVAL_MINUTES),
)

View File

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