From 5464e245a2c0bda1f4bcaac9bb9de9745dc42e67 Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Sun, 9 Feb 2025 15:35:34 +0100
Subject: [PATCH] Explicitly pass in the config_entry in ruuvi_gateway
 coordinator (#137964)

explicitly pass in the config_entry in coordinator
---
 .../components/ruuvi_gateway/__init__.py      | 12 ++-------
 .../components/ruuvi_gateway/coordinator.py   | 25 ++++++++++++-------
 2 files changed, 18 insertions(+), 19 deletions(-)

diff --git a/homeassistant/components/ruuvi_gateway/__init__.py b/homeassistant/components/ruuvi_gateway/__init__.py
index 77b3e9b57de..da93a89a9f3 100644
--- a/homeassistant/components/ruuvi_gateway/__init__.py
+++ b/homeassistant/components/ruuvi_gateway/__init__.py
@@ -5,11 +5,10 @@ from __future__ import annotations
 import logging
 
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_HOST, CONF_TOKEN
 from homeassistant.core import HomeAssistant
 
 from .bluetooth import async_connect_scanner
-from .const import DOMAIN, SCAN_INTERVAL
+from .const import DOMAIN
 from .coordinator import RuuviGatewayUpdateCoordinator
 from .models import RuuviGatewayRuntimeData
 
@@ -18,14 +17,7 @@ _LOGGER = logging.getLogger(DOMAIN)
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up Ruuvi Gateway from a config entry."""
-    coordinator = RuuviGatewayUpdateCoordinator(
-        hass,
-        logger=_LOGGER,
-        name=entry.title,
-        update_interval=SCAN_INTERVAL,
-        host=entry.data[CONF_HOST],
-        token=entry.data[CONF_TOKEN],
-    )
+    coordinator = RuuviGatewayUpdateCoordinator(hass, entry, _LOGGER)
     scanner, unload_scanner = async_connect_scanner(hass, entry, coordinator)
     hass.data.setdefault(DOMAIN, {})[entry.entry_id] = RuuviGatewayRuntimeData(
         update_coordinator=coordinator,
diff --git a/homeassistant/components/ruuvi_gateway/coordinator.py b/homeassistant/components/ruuvi_gateway/coordinator.py
index ba72dfe4cbc..0c42cd0cb38 100644
--- a/homeassistant/components/ruuvi_gateway/coordinator.py
+++ b/homeassistant/components/ruuvi_gateway/coordinator.py
@@ -2,34 +2,41 @@
 
 from __future__ import annotations
 
-from datetime import timedelta
 import logging
 
 from aioruuvigateway.api import get_gateway_history_data
 from aioruuvigateway.models import TagData
 
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_HOST, CONF_TOKEN
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.httpx_client import get_async_client
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
+from .const import SCAN_INTERVAL
+
 
 class RuuviGatewayUpdateCoordinator(DataUpdateCoordinator[list[TagData]]):
     """Polls the gateway for data and returns a list of TagData objects that have changed since the last poll."""
 
+    config_entry: ConfigEntry
+
     def __init__(
         self,
         hass: HomeAssistant,
+        config_entry: ConfigEntry,
         logger: logging.Logger,
-        *,
-        name: str,
-        update_interval: timedelta | None = None,
-        host: str,
-        token: str,
     ) -> None:
         """Initialize the coordinator using the given configuration (host, token)."""
-        super().__init__(hass, logger, name=name, update_interval=update_interval)
-        self.host = host
-        self.token = token
+        super().__init__(
+            hass,
+            logger,
+            config_entry=config_entry,
+            name=config_entry.title,
+            update_interval=SCAN_INTERVAL,
+        )
+        self.host = config_entry.data[CONF_HOST]
+        self.token = config_entry.data[CONF_TOKEN]
         self.last_tag_datas: dict[str, TagData] = {}
 
     async def _async_update_data(self) -> list[TagData]: