64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""Support for Automation Device Specification (ADS)."""
|
|
|
|
import asyncio
|
|
from asyncio import timeout
|
|
import logging
|
|
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from . import STATE_KEY_STATE
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class AdsEntity(Entity):
|
|
"""Representation of ADS entity."""
|
|
|
|
_attr_should_poll = False
|
|
|
|
def __init__(self, ads_hub, name, ads_var):
|
|
"""Initialize ADS binary sensor."""
|
|
self._state_dict = {}
|
|
self._state_dict[STATE_KEY_STATE] = None
|
|
self._ads_hub = ads_hub
|
|
self._ads_var = ads_var
|
|
self._event = None
|
|
self._attr_unique_id = ads_var
|
|
self._attr_name = name
|
|
|
|
async def async_initialize_device(
|
|
self, ads_var, plctype, state_key=STATE_KEY_STATE, factor=None
|
|
):
|
|
"""Register device notification."""
|
|
|
|
def update(name, value):
|
|
"""Handle device notifications."""
|
|
_LOGGER.debug("Variable %s changed its value to %d", name, value)
|
|
|
|
if factor is None:
|
|
self._state_dict[state_key] = value
|
|
else:
|
|
self._state_dict[state_key] = value / factor
|
|
|
|
asyncio.run_coroutine_threadsafe(async_event_set(), self.hass.loop)
|
|
self.schedule_update_ha_state()
|
|
|
|
async def async_event_set():
|
|
"""Set event in async context."""
|
|
self._event.set()
|
|
|
|
self._event = asyncio.Event()
|
|
|
|
await self.hass.async_add_executor_job(
|
|
self._ads_hub.add_device_notification, ads_var, plctype, update
|
|
)
|
|
try:
|
|
async with timeout(10):
|
|
await self._event.wait()
|
|
except TimeoutError:
|
|
_LOGGER.debug("Variable %s: Timeout during first update", ads_var)
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return False if state has not been updated yet."""
|
|
return self._state_dict[STATE_KEY_STATE] is not None
|