86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
"""Entity for the SleepIQ integration."""
|
|
from abc import abstractmethod
|
|
from typing import TypeVar
|
|
|
|
from asyncsleepiq import SleepIQBed, SleepIQSleeper
|
|
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.helpers.entity import DeviceInfo, Entity
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import ENTITY_TYPES, ICON_OCCUPIED
|
|
from .coordinator import SleepIQDataUpdateCoordinator, SleepIQPauseUpdateCoordinator
|
|
|
|
_SleepIQCoordinatorT = TypeVar(
|
|
"_SleepIQCoordinatorT",
|
|
bound=SleepIQDataUpdateCoordinator | SleepIQPauseUpdateCoordinator,
|
|
)
|
|
|
|
|
|
def device_from_bed(bed: SleepIQBed) -> DeviceInfo:
|
|
"""Create a device given a bed."""
|
|
return DeviceInfo(
|
|
connections={(dr.CONNECTION_NETWORK_MAC, bed.mac_addr)},
|
|
manufacturer="SleepNumber",
|
|
name=bed.name,
|
|
model=bed.model,
|
|
)
|
|
|
|
|
|
class SleepIQEntity(Entity):
|
|
"""Implementation of a SleepIQ entity."""
|
|
|
|
def __init__(self, bed: SleepIQBed) -> None:
|
|
"""Initialize the SleepIQ entity."""
|
|
self.bed = bed
|
|
self._attr_device_info = device_from_bed(bed)
|
|
|
|
|
|
class SleepIQBedEntity(CoordinatorEntity[_SleepIQCoordinatorT]):
|
|
"""Implementation of a SleepIQ sensor."""
|
|
|
|
_attr_icon = ICON_OCCUPIED
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: _SleepIQCoordinatorT,
|
|
bed: SleepIQBed,
|
|
) -> None:
|
|
"""Initialize the SleepIQ sensor entity."""
|
|
super().__init__(coordinator)
|
|
self.bed = bed
|
|
self._attr_device_info = device_from_bed(bed)
|
|
self._async_update_attrs()
|
|
|
|
@callback
|
|
def _handle_coordinator_update(self) -> None:
|
|
"""Handle updated data from the coordinator."""
|
|
self._async_update_attrs()
|
|
super()._handle_coordinator_update()
|
|
|
|
@callback
|
|
@abstractmethod
|
|
def _async_update_attrs(self) -> None:
|
|
"""Update sensor attributes."""
|
|
|
|
|
|
class SleepIQSleeperEntity(SleepIQBedEntity[_SleepIQCoordinatorT]):
|
|
"""Implementation of a SleepIQ sensor."""
|
|
|
|
_attr_icon = ICON_OCCUPIED
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: _SleepIQCoordinatorT,
|
|
bed: SleepIQBed,
|
|
sleeper: SleepIQSleeper,
|
|
name: str,
|
|
) -> None:
|
|
"""Initialize the SleepIQ sensor entity."""
|
|
self.sleeper = sleeper
|
|
super().__init__(coordinator, bed)
|
|
|
|
self._attr_name = f"SleepNumber {bed.name} {sleeper.name} {ENTITY_TYPES[name]}"
|
|
self._attr_unique_id = f"{sleeper.sleeper_id}_{name}"
|