jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 06:37:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
homeassistant/components/ukraine_alarm/__init__.py
View File

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

16
homeassistant/components/ukraine_alarm/coordinator.py
View File

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