Use typed ConfigEntry in twentemilieu (#131894)

This commit is contained in:
epenet 2024-12-01 16:40:30 +01:00 committed by GitHub
parent fd42c01a21
commit 8343d7f348
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 14 additions and 10 deletions

View File

@ -29,7 +29,9 @@ type TwenteMilieuDataUpdateCoordinator = DataUpdateCoordinator[
type TwenteMilieuConfigEntry = ConfigEntry[TwenteMilieuDataUpdateCoordinator] type TwenteMilieuConfigEntry = ConfigEntry[TwenteMilieuDataUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_setup_entry(
hass: HomeAssistant, entry: TwenteMilieuConfigEntry
) -> bool:
"""Set up Twente Milieu from a config entry.""" """Set up Twente Milieu from a config entry."""
session = async_get_clientsession(hass) session = async_get_clientsession(hass)
twentemilieu = TwenteMilieu( twentemilieu = TwenteMilieu(
@ -55,6 +57,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return True return True
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(
hass: HomeAssistant, entry: TwenteMilieuConfigEntry
) -> bool:
"""Unload Twente Milieu config entry.""" """Unload Twente Milieu config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

View File

@ -4,12 +4,13 @@ from __future__ import annotations
from typing import Any from typing import Any
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from . import TwenteMilieuConfigEntry
async def async_get_config_entry_diagnostics( async def async_get_config_entry_diagnostics(
hass: HomeAssistant, entry: ConfigEntry hass: HomeAssistant, entry: TwenteMilieuConfigEntry
) -> dict[str, Any]: ) -> dict[str, Any]:
"""Return diagnostics for a config entry.""" """Return diagnostics for a config entry."""
return { return {

View File

@ -2,13 +2,12 @@
from __future__ import annotations from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID from homeassistant.const import CONF_ID
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.helpers.update_coordinator import CoordinatorEntity from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import TwenteMilieuDataUpdateCoordinator from . import TwenteMilieuConfigEntry, TwenteMilieuDataUpdateCoordinator
from .const import DOMAIN from .const import DOMAIN
@ -17,7 +16,7 @@ class TwenteMilieuEntity(CoordinatorEntity[TwenteMilieuDataUpdateCoordinator], E
_attr_has_entity_name = True _attr_has_entity_name = True
def __init__(self, entry: ConfigEntry) -> None: def __init__(self, entry: TwenteMilieuConfigEntry) -> None:
"""Initialize the Twente Milieu entity.""" """Initialize the Twente Milieu entity."""
super().__init__(coordinator=entry.runtime_data) super().__init__(coordinator=entry.runtime_data)
self._attr_device_info = DeviceInfo( self._attr_device_info = DeviceInfo(

View File

@ -12,11 +12,11 @@ from homeassistant.components.sensor import (
SensorEntity, SensorEntity,
SensorEntityDescription, SensorEntityDescription,
) )
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_ID from homeassistant.const import CONF_ID
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import TwenteMilieuConfigEntry
from .const import DOMAIN from .const import DOMAIN
from .entity import TwenteMilieuEntity from .entity import TwenteMilieuEntity
@ -64,7 +64,7 @@ SENSORS: tuple[TwenteMilieuSensorDescription, ...] = (
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
entry: ConfigEntry, entry: TwenteMilieuConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Set up Twente Milieu sensor based on a config entry.""" """Set up Twente Milieu sensor based on a config entry."""
@ -80,7 +80,7 @@ class TwenteMilieuSensor(TwenteMilieuEntity, SensorEntity):
def __init__( def __init__(
self, self,
entry: ConfigEntry, entry: TwenteMilieuConfigEntry,
description: TwenteMilieuSensorDescription, description: TwenteMilieuSensorDescription,
) -> None: ) -> None:
"""Initialize the Twente Milieu entity.""" """Initialize the Twente Milieu entity."""