Fix typing of ConfigEntry in WLED (#155571)

Co-authored-by: mik-laj <[email protected]>

Author: mik-laj

homeassistant/components/wled/__init__.py

@@ -2,15 +2,18 @@
 
 from __future__ import annotations
 
-from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.typing import ConfigType
 from homeassistant.util.hass_dict import HassKey
 
 from .const import DOMAIN
-from .coordinator import WLEDDataUpdateCoordinator, WLEDReleasesDataUpdateCoordinator
+from .coordinator import (
+    WLEDConfigEntry,
+    WLEDDataUpdateCoordinator,
+    WLEDReleasesDataUpdateCoordinator,
+)
 
 PLATFORMS = (
     Platform.BUTTON,
@@ -22,8 +25,6 @@
     Platform.UPDATE,
 )
 
-type WLEDConfigEntry = ConfigEntry[WLEDDataUpdateCoordinator]
-
 WLED_KEY: HassKey[WLEDReleasesDataUpdateCoordinator] = HassKey(DOMAIN)
 CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
 

homeassistant/components/wled/button.py

@@ -7,8 +7,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
-from . import WLEDConfigEntry
-from .coordinator import WLEDDataUpdateCoordinator
+from .coordinator import WLEDConfigEntry, WLEDDataUpdateCoordinator
 from .entity import WLEDEntity
 from .helpers import wled_exception_handler
 

homeassistant/components/wled/config_flow.py

@@ -9,7 +9,6 @@
 
 from homeassistant.components import onboarding
 from homeassistant.config_entries import (
-    ConfigEntry,
     ConfigFlow,
     ConfigFlowResult,
     OptionsFlowWithReload,
@@ -20,6 +19,7 @@
 from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
 
 from .const import CONF_KEEP_MAIN_LIGHT, DEFAULT_KEEP_MAIN_LIGHT, DOMAIN
+from .coordinator import WLEDConfigEntry
 
 
 class WLEDFlowHandler(ConfigFlow, domain=DOMAIN):
@@ -32,7 +32,7 @@ class WLEDFlowHandler(ConfigFlow, domain=DOMAIN):
     @staticmethod
     @callback
     def async_get_options_flow(
-        config_entry: ConfigEntry,
+        config_entry: WLEDConfigEntry,
     ) -> WLEDOptionsFlowHandler:
         """Get the options flow for this handler."""
         return WLEDOptionsFlowHandler()

homeassistant/components/wled/coordinator.py

@@ -26,18 +26,20 @@
     SCAN_INTERVAL,
 )
 
+type WLEDConfigEntry = ConfigEntry[WLEDDataUpdateCoordinator]
+
 
 class WLEDDataUpdateCoordinator(DataUpdateCoordinator[WLEDDevice]):
     """Class to manage fetching WLED data from single endpoint."""
 
     keep_main_light: bool
-    config_entry: ConfigEntry
+    config_entry: WLEDConfigEntry
 
     def __init__(
         self,
         hass: HomeAssistant,
         *,
-        entry: ConfigEntry,
+        entry: WLEDConfigEntry,
     ) -> None:
         """Initialize global WLED data updater."""
         self.keep_main_light = entry.options.get(

homeassistant/components/wled/diagnostics.py

@@ -7,7 +7,7 @@
 from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.core import HomeAssistant
 
-from . import WLEDConfigEntry
+from .coordinator import WLEDConfigEntry
 
 
 async def async_get_config_entry_diagnostics(

homeassistant/components/wled/light.py

@@ -19,7 +19,6 @@
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
-from . import WLEDConfigEntry
 from .const import (
     ATTR_CCT,
     ATTR_COLOR_PRIMARY,
@@ -29,7 +28,7 @@
     COLOR_TEMP_K_MIN,
     LIGHT_CAPABILITIES_COLOR_MODE_MAPPING,
 )
-from .coordinator import WLEDDataUpdateCoordinator
+from .coordinator import WLEDConfigEntry, WLEDDataUpdateCoordinator
 from .entity import WLEDEntity
 from .helpers import kelvin_to_255, kelvin_to_255_reverse, wled_exception_handler
 

homeassistant/components/wled/number.py

@@ -13,9 +13,8 @@
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
-from . import WLEDConfigEntry
 from .const import ATTR_INTENSITY, ATTR_SPEED
-from .coordinator import WLEDDataUpdateCoordinator
+from .coordinator import WLEDConfigEntry, WLEDDataUpdateCoordinator
 from .entity import WLEDEntity
 from .helpers import wled_exception_handler
 

homeassistant/components/wled/select.py

@@ -11,8 +11,7 @@
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
-from . import WLEDConfigEntry
-from .coordinator import WLEDDataUpdateCoordinator
+from .coordinator import WLEDConfigEntry, WLEDDataUpdateCoordinator
 from .entity import WLEDEntity
 from .helpers import wled_exception_handler
 

homeassistant/components/wled/sensor.py

@@ -26,8 +26,7 @@
 from homeassistant.helpers.typing import StateType
 from homeassistant.util.dt import utcnow
 
-from . import WLEDConfigEntry
-from .coordinator import WLEDDataUpdateCoordinator
+from .coordinator import WLEDConfigEntry, WLEDDataUpdateCoordinator
 from .entity import WLEDEntity
 
 # Coordinator is used to centralize the data updates

homeassistant/components/wled/switch.py

@@ -10,9 +10,8 @@
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
-from . import WLEDConfigEntry
 from .const import ATTR_DURATION, ATTR_TARGET_BRIGHTNESS, ATTR_UDP_PORT
-from .coordinator import WLEDDataUpdateCoordinator
+from .coordinator import WLEDConfigEntry, WLEDDataUpdateCoordinator
 from .entity import WLEDEntity
 from .helpers import wled_exception_handler