Fix runtime_data in iqvia (#145181)

This commit is contained in:
epenet 2025-05-19 10:59:46 +02:00 committed by GitHub
parent bd190b9b4c
commit a34bce6202
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 9 deletions

View File

@ -6,7 +6,7 @@ from homeassistant.core import callback
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONF_ZIP_CODE, DOMAIN, TYPE_ALLERGY_FORECAST, TYPE_ALLERGY_OUTLOOK
from .const import CONF_ZIP_CODE, TYPE_ALLERGY_FORECAST, TYPE_ALLERGY_OUTLOOK
from .coordinator import IqviaConfigEntry, IqviaUpdateCoordinator
@ -44,9 +44,9 @@ class IQVIAEntity(CoordinatorEntity[IqviaUpdateCoordinator]):
if self.entity_description.key == TYPE_ALLERGY_FORECAST:
self.async_on_remove(
self.hass.data[DOMAIN][self._entry.entry_id][
TYPE_ALLERGY_OUTLOOK
].async_add_listener(self._handle_coordinator_update)
self._entry.runtime_data[TYPE_ALLERGY_OUTLOOK].async_add_listener(
self._handle_coordinator_update
)
)
self.update_from_latest_data()

View File

@ -17,7 +17,6 @@ from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import (
DOMAIN,
TYPE_ALLERGY_FORECAST,
TYPE_ALLERGY_INDEX,
TYPE_ALLERGY_OUTLOOK,
@ -145,7 +144,7 @@ async def async_setup_entry(
sensors.extend(
[
IndexSensor(
hass.data[DOMAIN][entry.entry_id][
entry.runtime_data[
API_CATEGORY_MAPPING.get(description.key, description.key)
],
entry,
@ -207,9 +206,7 @@ class ForecastSensor(IQVIAEntity, SensorEntity):
)
if self.entity_description.key == TYPE_ALLERGY_FORECAST:
outlook_coordinator = self.hass.data[DOMAIN][self._entry.entry_id][
TYPE_ALLERGY_OUTLOOK
]
outlook_coordinator = self._entry.runtime_data[TYPE_ALLERGY_OUTLOOK]
if not outlook_coordinator.last_update_success:
return