update async_setup_entry to retrieve optionsflow entry

pantherale0 · Jan 6, 2024 · bb94e24 · bb94e24
1 parent f064d1f
commit bb94e24
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/custom_components/fuel_prices/__init__.py b/custom_components/fuel_prices/__init__.py
@@ -37,15 +37,16 @@ def _build_configured_areas(hass_areas: dict) -> list[dict]:
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Create ConfigEntry."""
-    hass.data.setdefault(DOMAIN, {})
     _LOGGER.debug("Got request to setup entry.")
+    sources = entry.options.get(CONF_SOURCES, entry.data.get(CONF_SOURCES, None))
+    areas = entry.options.get(CONF_AREAS, entry.data.get(CONF_AREAS, None))
     try:
         fuel_prices: FuelPrices = FuelPrices.create(
-            enabled_sources=entry.data.get(CONF_SOURCES, None),
-            configured_areas=_build_configured_areas(entry.data[CONF_AREAS]),
+            enabled_sources=sources,
+            configured_areas=_build_configured_areas(areas),
         )
         await fuel_prices.update()
-        hass.data[DOMAIN][entry.entry_id] = FuelPricesCoordinator(
+        hass.data.setdefault(DOMAIN, {})[entry.entry_id] = FuelPricesCoordinator(
             hass, fuel_prices, entry.entry_id
         )
     except Exception as err: