From b54b90a60452b58c247f0e9bbac719dbea99b2d3 Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Sun, 9 Feb 2025 14:25:43 +0100
Subject: [PATCH] Explicitly pass in the config_entry in ukraine_alarm
 coordinator (#137886)

explicitly pass in the config_entry in coordinator
---
 .../components/ukraine_alarm/__init__.py         |  5 +----
 .../components/ukraine_alarm/coordinator.py      | 16 +++++++++++++---
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/ukraine_alarm/__init__.py b/homeassistant/components/ukraine_alarm/__init__.py
index d850ed6eba8..3658b821625 100644
--- a/homeassistant/components/ukraine_alarm/__init__.py
+++ b/homeassistant/components/ukraine_alarm/__init__.py
@@ -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
diff --git a/homeassistant/components/ukraine_alarm/coordinator.py b/homeassistant/components/ukraine_alarm/coordinator.py
index fbf7c9f81c2..267358e4aa6 100644
--- a/homeassistant/components/ukraine_alarm/coordinator.py
+++ b/homeassistant/components/ukraine_alarm/coordinator.py
@@ -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."""