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

This commit is contained in:
Marc Mueller 2022-12-27 22:44:25 +01:00 committed by GitHub
parent 5a5f01fcd4
commit 1de41ab123
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
13 changed files with 15 additions and 15 deletions

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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