mirror of https://github.com/home-assistant/core
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
"""The Ecowitt Weather Station Entity."""
|
|
from __future__ import annotations
|
|
|
|
import time
|
|
|
|
from aioecowitt import EcoWittSensor
|
|
|
|
from homeassistant.helpers.device_registry import DeviceInfo
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from .const import DOMAIN
|
|
|
|
|
|
class EcowittEntity(Entity):
|
|
"""Base class for Ecowitt Weather Station."""
|
|
|
|
_attr_has_entity_name = True
|
|
_attr_should_poll = False
|
|
|
|
def __init__(self, sensor: EcoWittSensor) -> None:
|
|
"""Construct the entity."""
|
|
self.ecowitt: EcoWittSensor = sensor
|
|
|
|
self._attr_unique_id = f"{sensor.station.key}-{sensor.key}"
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={
|
|
(DOMAIN, sensor.station.key),
|
|
},
|
|
name=sensor.station.model,
|
|
model=sensor.station.model,
|
|
sw_version=sensor.station.version,
|
|
)
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Install listener for updates later."""
|
|
|
|
def _update_state() -> None:
|
|
"""Update the state on callback."""
|
|
self.async_write_ha_state()
|
|
|
|
self.ecowitt.update_cb.append(_update_state) # type: ignore[arg-type] # upstream bug
|
|
self.async_on_remove(lambda: self.ecowitt.update_cb.remove(_update_state)) # type: ignore[arg-type] # upstream bug
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return whether the state is based on actual reading from device."""
|
|
return (self.ecowitt.last_update_m + 5 * 60) > time.monotonic()
|