35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""Support for Steamist sensors."""
|
|
from __future__ import annotations
|
|
|
|
from aiosteamist import SteamistStatus
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.helpers.entity import Entity, EntityDescription
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .coordinator import SteamistDataUpdateCoordinator
|
|
|
|
|
|
class SteamistEntity(CoordinatorEntity, Entity):
|
|
"""Representation of an Steamist entity."""
|
|
|
|
coordinator: SteamistDataUpdateCoordinator
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: SteamistDataUpdateCoordinator,
|
|
entry: ConfigEntry,
|
|
description: EntityDescription,
|
|
) -> None:
|
|
"""Initialize the entity."""
|
|
super().__init__(coordinator)
|
|
self.entity_description = description
|
|
if coordinator.device_name:
|
|
self._attr_name = f"{coordinator.device_name} {description.name}"
|
|
self._attr_unique_id = f"{entry.entry_id}_{description.key}"
|
|
|
|
@property
|
|
def _status(self) -> SteamistStatus:
|
|
"""Return the steamist status."""
|
|
return self.coordinator.data
|