30382c3dbe
- Each retry was logging the error again - Now we set the cause of the PlatformNotReady to allow Home Assistant to log as needed
118 lines
3.4 KiB
Python
118 lines
3.4 KiB
Python
"""Support for RESTful binary sensors."""
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.binary_sensor import (
|
|
DOMAIN as BINARY_SENSOR_DOMAIN,
|
|
PLATFORM_SCHEMA,
|
|
BinarySensorEntity,
|
|
)
|
|
from homeassistant.const import (
|
|
CONF_DEVICE_CLASS,
|
|
CONF_FORCE_UPDATE,
|
|
CONF_NAME,
|
|
CONF_RESOURCE,
|
|
CONF_RESOURCE_TEMPLATE,
|
|
CONF_VALUE_TEMPLATE,
|
|
)
|
|
from homeassistant.exceptions import PlatformNotReady
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
from . import async_get_config_and_coordinator, create_rest_data_from_config
|
|
from .entity import RestEntity
|
|
from .schema import BINARY_SENSOR_SCHEMA, RESOURCE_SCHEMA
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({**RESOURCE_SCHEMA, **BINARY_SENSOR_SCHEMA})
|
|
|
|
PLATFORM_SCHEMA = vol.All(
|
|
cv.has_at_least_one_key(CONF_RESOURCE, CONF_RESOURCE_TEMPLATE), PLATFORM_SCHEMA
|
|
)
|
|
|
|
|
|
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
|
|
"""Set up the REST binary sensor."""
|
|
# Must update the sensor now (including fetching the rest resource) to
|
|
# ensure it's updating its state.
|
|
if discovery_info is not None:
|
|
conf, coordinator, rest = await async_get_config_and_coordinator(
|
|
hass, BINARY_SENSOR_DOMAIN, discovery_info
|
|
)
|
|
else:
|
|
conf = config
|
|
coordinator = None
|
|
rest = create_rest_data_from_config(hass, conf)
|
|
await rest.async_update(log_errors=False)
|
|
|
|
if rest.data is None:
|
|
if rest.last_exception:
|
|
raise PlatformNotReady from rest.last_exception
|
|
raise PlatformNotReady
|
|
|
|
name = conf.get(CONF_NAME)
|
|
device_class = conf.get(CONF_DEVICE_CLASS)
|
|
value_template = conf.get(CONF_VALUE_TEMPLATE)
|
|
force_update = conf.get(CONF_FORCE_UPDATE)
|
|
resource_template = conf.get(CONF_RESOURCE_TEMPLATE)
|
|
|
|
if value_template is not None:
|
|
value_template.hass = hass
|
|
|
|
async_add_entities(
|
|
[
|
|
RestBinarySensor(
|
|
coordinator,
|
|
rest,
|
|
name,
|
|
device_class,
|
|
value_template,
|
|
force_update,
|
|
resource_template,
|
|
)
|
|
],
|
|
)
|
|
|
|
|
|
class RestBinarySensor(RestEntity, BinarySensorEntity):
|
|
"""Representation of a REST binary sensor."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator,
|
|
rest,
|
|
name,
|
|
device_class,
|
|
value_template,
|
|
force_update,
|
|
resource_template,
|
|
):
|
|
"""Initialize a REST binary sensor."""
|
|
super().__init__(
|
|
coordinator, rest, name, device_class, resource_template, force_update
|
|
)
|
|
self._state = False
|
|
self._previous_data = None
|
|
self._value_template = value_template
|
|
self._is_on = None
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if the binary sensor is on."""
|
|
return self._is_on
|
|
|
|
def _update_from_rest_data(self):
|
|
"""Update state from the rest data."""
|
|
if self.rest.data is None:
|
|
self._is_on = False
|
|
|
|
response = self.rest.data
|
|
|
|
if self._value_template is not None:
|
|
response = self._value_template.async_render_with_possible_json_value(
|
|
self.rest.data, False
|
|
)
|
|
|
|
try:
|
|
self._is_on = bool(int(response))
|
|
except ValueError:
|
|
self._is_on = {"true": True, "on": True, "open": True, "yes": True}.get(
|
|
response.lower(), False
|
|
)
|