Improve DataUpdateCoordinator typing in integrations (1) (#84655)

homeassistant/components/bmw_connected_drive/coordinator.py

@@ -21,7 +21,7 @@ SCAN_INTERVAL = timedelta(seconds=DEFAULT_SCAN_INTERVAL_SECONDS)
 _LOGGER = logging.getLogger(__name__)
 
 
-class BMWDataUpdateCoordinator(DataUpdateCoordinator):
+class BMWDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Class to manage fetching BMW data."""
 
     account: MyBMWAccount

homeassistant/components/flux_led/coordinator.py

@@ -20,7 +20,7 @@ _LOGGER = logging.getLogger(__name__)
 REQUEST_REFRESH_DELAY: Final = 2.0
 
 
-class FluxLedUpdateCoordinator(DataUpdateCoordinator):
+class FluxLedUpdateCoordinator(DataUpdateCoordinator[None]):
     """DataUpdateCoordinator to gather data for a specific flux_led device."""
 
     def __init__(

homeassistant/components/fritz/common.py

@@ -136,7 +136,7 @@ class HostInfo(TypedDict):
     status: bool
 
 
-class FritzBoxTools(update_coordinator.DataUpdateCoordinator):
+class FritzBoxTools(update_coordinator.DataUpdateCoordinator[None]):
     """FritzBoxTools class."""
 
     def __init__(

homeassistant/components/ialarm/__init__.py

@@ -54,7 +54,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     return unload_ok
 
 
-class IAlarmDataUpdateCoordinator(DataUpdateCoordinator):
+class IAlarmDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Class to manage fetching iAlarm data."""
 
     def __init__(self, hass: HomeAssistant, ialarm: IAlarm, mac: str) -> None:

homeassistant/components/lifx/coordinator.py

@@ -64,7 +64,7 @@ class FirmwareEffect(IntEnum):
     FLAME = 3
 
 
-class LIFXUpdateCoordinator(DataUpdateCoordinator):
+class LIFXUpdateCoordinator(DataUpdateCoordinator[None]):
     """DataUpdateCoordinator to gather data for a specific lifx device."""
 
     def __init__(

homeassistant/components/nexia/coordinator.py

@@ -14,7 +14,7 @@ _LOGGER = logging.getLogger(__name__)
 DEFAULT_UPDATE_RATE = 120
 
 
-class NexiaDataUpdateCoordinator(DataUpdateCoordinator):
+class NexiaDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """DataUpdateCoordinator for nexia homes."""
 
     def __init__(

homeassistant/components/rituals_perfume_genie/__init__.py

@@ -65,7 +65,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     return unload_ok
 
 
-class RitualsDataUpdateCoordinator(DataUpdateCoordinator):
+class RitualsDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Class to manage fetching Rituals Perfume Genie device data from single endpoint."""
 
     def __init__(self, hass: HomeAssistant, device: Diffuser) -> None:

homeassistant/components/skybell/coordinator.py

@@ -11,7 +11,7 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
 from .const import LOGGER
 
 
-class SkybellDataUpdateCoordinator(DataUpdateCoordinator):
+class SkybellDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Data update coordinator for the Skybell integration."""
 
     config_entry: ConfigEntry

homeassistant/components/tautulli/coordinator.py

@@ -24,7 +24,7 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
 from .const import DOMAIN, LOGGER
 
 
-class TautulliDataUpdateCoordinator(DataUpdateCoordinator):
+class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Data update coordinator for the Tautulli integration."""
 
     config_entry: ConfigEntry

homeassistant/components/tibber/sensor.py

@@ -553,7 +553,7 @@ class TibberRtDataCoordinator(DataUpdateCoordinator):
         return self.data.get("data", {}).get("liveMeasurement")
 
 
-class TibberDataCoordinator(DataUpdateCoordinator):
+class TibberDataCoordinator(DataUpdateCoordinator[None]):
     """Handle Tibber data and insert statistics."""
 
     def __init__(self, hass: HomeAssistant, tibber_connection: tibber.Tibber) -> None:

homeassistant/components/tplink/coordinator.py

@@ -15,7 +15,7 @@ _LOGGER = logging.getLogger(__name__)
 REQUEST_REFRESH_DELAY = 0.35
 
 
-class TPLinkDataUpdateCoordinator(DataUpdateCoordinator):
+class TPLinkDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """DataUpdateCoordinator to gather data for a specific TPLink device."""
 
     def __init__(

homeassistant/components/volvooncall/__init__.py

@@ -239,7 +239,7 @@ class VolvoData:
             raise InvalidAuth from exc
 
 
-class VolvoUpdateCoordinator(DataUpdateCoordinator):
+class VolvoUpdateCoordinator(DataUpdateCoordinator[None]):
     """Volvo coordinator."""
 
     def __init__(self, hass: HomeAssistant, volvo_data: VolvoData) -> None:
@@ -254,14 +254,14 @@ class VolvoUpdateCoordinator(DataUpdateCoordinator):
 
         self.volvo_data = volvo_data
 
-    async def _async_update_data(self):
+    async def _async_update_data(self) -> None:
         """Fetch data from API endpoint."""
 
         async with async_timeout.timeout(10):
             await self.volvo_data.update()
 
 
-class VolvoEntity(CoordinatorEntity):
+class VolvoEntity(CoordinatorEntity[VolvoUpdateCoordinator]):
     """Base class for all VOC entities."""
 
     def __init__(

homeassistant/components/wemo/wemo_device.py

@@ -30,7 +30,7 @@ from .const import DOMAIN, WEMO_SUBSCRIPTION_EVENT
 _LOGGER = logging.getLogger(__name__)
 
 
-class DeviceCoordinator(DataUpdateCoordinator):
+class DeviceCoordinator(DataUpdateCoordinator[None]):
     """Home Assistant wrapper for a pyWeMo device."""
 
     def __init__(self, hass: HomeAssistant, wemo: WeMoDevice, device_id: str) -> None: