mirror of
https://github.com/home-assistant/core.git
synced 2025-07-27 15:17:35 +00:00
Fix runtime_data in iqvia (#145181)
This commit is contained in:
parent
bd190b9b4c
commit
a34bce6202
@ -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()
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user