75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
|
"""Binary sensor platform for nodered."""
|
||
|
|
from numbers import Number
|
||
|
|
|
||
|
|
from homeassistant.const import (
|
||
|
|
CONF_STATE,
|
||
|
|
STATE_HOME,
|
||
|
|
STATE_OFF,
|
||
|
|
STATE_ON,
|
||
|
|
STATE_OPEN,
|
||
|
|
STATE_UNLOCKED,
|
||
|
|
)
|
||
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
||
|
|
|
||
|
|
from . import NodeRedEntity
|
||
|
|
from .const import CONF_ATTRIBUTES, CONF_BINARY_SENSOR, NODERED_DISCOVERY_NEW
|
||
|
|
|
||
|
|
|
||
|
|
async def async_setup_entry(hass, config_entry, async_add_devices):
|
||
|
|
"""Set up sensor platform."""
|
||
|
|
|
||
|
|
async def async_discover(config, connection):
|
||
|
|
await _async_setup_entity(hass, config, async_add_devices)
|
||
|
|
|
||
|
|
async_dispatcher_connect(
|
||
|
|
hass, NODERED_DISCOVERY_NEW.format(CONF_BINARY_SENSOR), async_discover,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
async def _async_setup_entity(hass, config, async_add_devices):
|
||
|
|
"""Set up the Node-RED binary-sensor."""
|
||
|
|
async_add_devices([NodeRedBinarySensor(hass, config)])
|
||
|
|
|
||
|
|
|
||
|
|
class NodeRedBinarySensor(NodeRedEntity):
|
||
|
|
"""Node-RED binary-sensor class."""
|
||
|
|
|
||
|
|
on_states = (
|
||
|
|
"1",
|
||
|
|
"true",
|
||
|
|
"yes",
|
||
|
|
"enable",
|
||
|
|
STATE_ON,
|
||
|
|
STATE_OPEN,
|
||
|
|
STATE_HOME,
|
||
|
|
STATE_UNLOCKED,
|
||
|
|
)
|
||
|
|
|
||
|
|
def __init__(self, hass, config):
|
||
|
|
"""Initialize the binary sensor."""
|
||
|
|
super().__init__(hass, config)
|
||
|
|
self._component = CONF_BINARY_SENSOR
|
||
|
|
self._state = config.get(CONF_STATE)
|
||
|
|
self.attr = config.get(CONF_ATTRIBUTES, {})
|
||
|
|
|
||
|
|
@property
|
||
|
|
def is_on(self):
|
||
|
|
"""Return true if the binary sensor is on."""
|
||
|
|
value = self._state
|
||
|
|
|
||
|
|
if isinstance(value, bool):
|
||
|
|
return value
|
||
|
|
if isinstance(value, str):
|
||
|
|
value = value.lower().strip()
|
||
|
|
if value in NodeRedBinarySensor.on_states:
|
||
|
|
return True
|
||
|
|
elif isinstance(value, Number):
|
||
|
|
return value != 0
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
@property
|
||
|
|
def state(self):
|
||
|
|
"""Return the state of the binary sensor."""
|
||
|
|
return STATE_ON if self.is_on else STATE_OFF
|