Set config_entry explicitely in scrape coordinator (home-assistant#12…

…9416)
zxdavb · Oct 29, 2024 · 10fdf81 · 10fdf81
1 parent 0292860
commit 10fdf81
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/homeassistant/components/scrape/__init__.py b/homeassistant/components/scrape/__init__.py
@@ -72,7 +72,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
         scan_interval: timedelta = resource_config.get(
             CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL
         )
-        coordinator = ScrapeCoordinator(hass, rest, scan_interval)
+        coordinator = ScrapeCoordinator(hass, None, rest, scan_interval)
 
         sensors: list[ConfigType] = resource_config.get(SENSOR_DOMAIN, [])
         if sensors:
@@ -100,6 +100,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ScrapeConfigEntry) -> bo
 
     coordinator = ScrapeCoordinator(
         hass,
+        entry,
         rest,
         DEFAULT_SCAN_INTERVAL,
     )

diff --git a/homeassistant/components/scrape/coordinator.py b/homeassistant/components/scrape/coordinator.py
@@ -8,6 +8,7 @@
 from bs4 import BeautifulSoup
 
 from homeassistant.components.rest import RestData
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
@@ -18,12 +19,17 @@ class ScrapeCoordinator(DataUpdateCoordinator[BeautifulSoup]):
     """Scrape Coordinator."""
 
     def __init__(
-        self, hass: HomeAssistant, rest: RestData, update_interval: timedelta
+        self,
+        hass: HomeAssistant,
+        config_entry: ConfigEntry | None,
+        rest: RestData,
+        update_interval: timedelta,
     ) -> None:
         """Initialize Scrape coordinator."""
         super().__init__(
             hass,
             _LOGGER,
+            config_entry=config_entry,
             name="Scrape Coordinator",
             update_interval=update_interval,
         )