Explicitly pass in the config_entry in ukraine_alarm coordinator (#137886)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 14:25:43 +01:00 committed by GitHub
parent 0cbec3c4bb
commit b54b90a604
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 7 deletions

View File

@ -3,7 +3,6 @@
from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_REGION
from homeassistant.core import HomeAssistant
from homeassistant.helpers.aiohttp_client import async_get_clientsession
@ -13,11 +12,9 @@ from .coordinator import UkraineAlarmDataUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Ukraine Alarm as config entry."""
region_id = entry.data[CONF_REGION]
websession = async_get_clientsession(hass)
coordinator = UkraineAlarmDataUpdateCoordinator(hass, websession, region_id)
coordinator = UkraineAlarmDataUpdateCoordinator(hass, entry, websession)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator

View File

@ -10,6 +10,8 @@ import aiohttp
from aiohttp import ClientSession
from uasiren.client import Client
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_REGION
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -23,17 +25,25 @@ UPDATE_INTERVAL = timedelta(seconds=10)
class UkraineAlarmDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching Ukraine Alarm API."""
config_entry: ConfigEntry
def __init__(
self,
hass: HomeAssistant,
config_entry: ConfigEntry,
session: ClientSession,
region_id: str,
) -> None:
"""Initialize."""
self.region_id = region_id
self.region_id = config_entry.data[CONF_REGION]
self.uasiren = Client(session)
super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name=DOMAIN,
update_interval=UPDATE_INTERVAL,
)
async def _async_update_data(self) -> dict[str, Any]:
"""Update data via library."""