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.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity 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 from .coordinator import IqviaConfigEntry, IqviaUpdateCoordinator
@ -44,9 +44,9 @@ class IQVIAEntity(CoordinatorEntity[IqviaUpdateCoordinator]):
if self.entity_description.key == TYPE_ALLERGY_FORECAST: if self.entity_description.key == TYPE_ALLERGY_FORECAST:
self.async_on_remove( self.async_on_remove(
self.hass.data[DOMAIN][self._entry.entry_id][ self._entry.runtime_data[TYPE_ALLERGY_OUTLOOK].async_add_listener(
TYPE_ALLERGY_OUTLOOK self._handle_coordinator_update
].async_add_listener(self._handle_coordinator_update) )
) )
self.update_from_latest_data() 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 homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from .const import ( from .const import (
DOMAIN,
TYPE_ALLERGY_FORECAST, TYPE_ALLERGY_FORECAST,
TYPE_ALLERGY_INDEX, TYPE_ALLERGY_INDEX,
TYPE_ALLERGY_OUTLOOK, TYPE_ALLERGY_OUTLOOK,
@ -145,7 +144,7 @@ async def async_setup_entry(
sensors.extend( sensors.extend(
[ [
IndexSensor( IndexSensor(
hass.data[DOMAIN][entry.entry_id][ entry.runtime_data[
API_CATEGORY_MAPPING.get(description.key, description.key) API_CATEGORY_MAPPING.get(description.key, description.key)
], ],
entry, entry,
@ -207,9 +206,7 @@ class ForecastSensor(IQVIAEntity, SensorEntity):
) )
if self.entity_description.key == TYPE_ALLERGY_FORECAST: if self.entity_description.key == TYPE_ALLERGY_FORECAST:
outlook_coordinator = self.hass.data[DOMAIN][self._entry.entry_id][ outlook_coordinator = self._entry.runtime_data[TYPE_ALLERGY_OUTLOOK]
TYPE_ALLERGY_OUTLOOK
]
if not outlook_coordinator.last_update_success: if not outlook_coordinator.last_update_success:
return return