38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
"""Base Entity for JustNimbus sensors."""
|
|
from __future__ import annotations
|
|
|
|
import justnimbus
|
|
|
|
from homeassistant.components.sensor import SensorEntity
|
|
from homeassistant.helpers import update_coordinator
|
|
from homeassistant.helpers.entity import DeviceInfo
|
|
|
|
from . import JustNimbusCoordinator
|
|
from .const import DOMAIN
|
|
|
|
|
|
class JustNimbusEntity(
|
|
update_coordinator.CoordinatorEntity[justnimbus.JustNimbusModel],
|
|
SensorEntity,
|
|
):
|
|
"""Defines a base JustNimbus entity."""
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
device_id: str,
|
|
coordinator: JustNimbusCoordinator,
|
|
) -> None:
|
|
"""Initialize the JustNimbus entity."""
|
|
super().__init__(coordinator=coordinator)
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, device_id)},
|
|
name="JustNimbus Sensor",
|
|
manufacturer="JustNimbus",
|
|
)
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return device availability."""
|
|
return super().available and getattr(self.coordinator.data, "error_code") == 0
|