mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Merge pull request #767 from fabaff/binary-rest-sensor
Binary REST sensor
This commit is contained in:
commit
dbad01f73d
@ -33,6 +33,7 @@ omit =
|
|||||||
homeassistant/components/*/rfxtrx.py
|
homeassistant/components/*/rfxtrx.py
|
||||||
|
|
||||||
homeassistant/components/binary_sensor/arest.py
|
homeassistant/components/binary_sensor/arest.py
|
||||||
|
homeassistant/components/binary_sensor/rest.py
|
||||||
homeassistant/components/browser.py
|
homeassistant/components/browser.py
|
||||||
homeassistant/components/camera/*
|
homeassistant/components/camera/*
|
||||||
homeassistant/components/device_tracker/actiontec.py
|
homeassistant/components/device_tracker/actiontec.py
|
||||||
|
147
homeassistant/components/binary_sensor/rest.py
Normal file
147
homeassistant/components/binary_sensor/rest.py
Normal file
@ -0,0 +1,147 @@
|
|||||||
|
"""
|
||||||
|
homeassistant.components.binary_sensor.rest
|
||||||
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||||
|
The rest binary sensor will consume responses sent by an exposed REST API.
|
||||||
|
|
||||||
|
For more details about this platform, please refer to the documentation at
|
||||||
|
https://home-assistant.io/components/binary_sensor.rest/
|
||||||
|
"""
|
||||||
|
from datetime import timedelta
|
||||||
|
import logging
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from homeassistant.const import CONF_VALUE_TEMPLATE
|
||||||
|
from homeassistant.util import template, Throttle
|
||||||
|
from homeassistant.components.binary_sensor import BinarySensorDevice
|
||||||
|
|
||||||
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
DEFAULT_NAME = 'REST Binary Sensor'
|
||||||
|
DEFAULT_METHOD = 'GET'
|
||||||
|
|
||||||
|
# Return cached results if last scan was less then this time ago
|
||||||
|
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=60)
|
||||||
|
|
||||||
|
|
||||||
|
# pylint: disable=unused-variable
|
||||||
|
def setup_platform(hass, config, add_devices, discovery_info=None):
|
||||||
|
""" Get the REST binary sensor. """
|
||||||
|
|
||||||
|
use_get = False
|
||||||
|
use_post = False
|
||||||
|
|
||||||
|
resource = config.get('resource', None)
|
||||||
|
method = config.get('method', DEFAULT_METHOD)
|
||||||
|
payload = config.get('payload', None)
|
||||||
|
verify_ssl = config.get('verify_ssl', True)
|
||||||
|
|
||||||
|
if method == 'GET':
|
||||||
|
use_get = True
|
||||||
|
elif method == 'POST':
|
||||||
|
use_post = True
|
||||||
|
|
||||||
|
try:
|
||||||
|
if use_get:
|
||||||
|
response = requests.get(resource, timeout=10, verify=verify_ssl)
|
||||||
|
elif use_post:
|
||||||
|
response = requests.post(resource, data=payload, timeout=10,
|
||||||
|
verify=verify_ssl)
|
||||||
|
if not response.ok:
|
||||||
|
_LOGGER.error('Response status is "%s"', response.status_code)
|
||||||
|
return False
|
||||||
|
except requests.exceptions.MissingSchema:
|
||||||
|
_LOGGER.error('Missing resource or schema in configuration. '
|
||||||
|
'Add http:// to your URL.')
|
||||||
|
return False
|
||||||
|
except requests.exceptions.ConnectionError:
|
||||||
|
_LOGGER.error('No route to resource/endpoint. '
|
||||||
|
'Please check the URL in the configuration file.')
|
||||||
|
return False
|
||||||
|
|
||||||
|
if use_get:
|
||||||
|
rest = RestDataGet(resource, verify_ssl)
|
||||||
|
elif use_post:
|
||||||
|
rest = RestDataPost(resource, payload, verify_ssl)
|
||||||
|
|
||||||
|
add_devices([RestBinarySensor(hass,
|
||||||
|
rest,
|
||||||
|
config.get('name', DEFAULT_NAME),
|
||||||
|
config.get(CONF_VALUE_TEMPLATE))])
|
||||||
|
|
||||||
|
|
||||||
|
# pylint: disable=too-many-arguments
|
||||||
|
class RestBinarySensor(BinarySensorDevice):
|
||||||
|
""" Implements a REST binary sensor. """
|
||||||
|
|
||||||
|
def __init__(self, hass, rest, name, value_template):
|
||||||
|
self._hass = hass
|
||||||
|
self.rest = rest
|
||||||
|
self._name = name
|
||||||
|
self._state = False
|
||||||
|
self._value_template = value_template
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
""" The name of the binary sensor. """
|
||||||
|
return self._name
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_on(self):
|
||||||
|
""" True if the binary sensor is on. """
|
||||||
|
print("### UPDATE ", self.rest.data)
|
||||||
|
|
||||||
|
if self.rest.data is False:
|
||||||
|
return False
|
||||||
|
else:
|
||||||
|
if self._value_template is not None:
|
||||||
|
self.rest.data = template.render_with_possible_json_value(
|
||||||
|
self._hass, self._value_template, self.rest.data, False)
|
||||||
|
return bool(int(self.rest.data))
|
||||||
|
|
||||||
|
def update(self):
|
||||||
|
""" Gets the latest data from REST API and updates the state. """
|
||||||
|
self.rest.update()
|
||||||
|
|
||||||
|
|
||||||
|
# pylint: disable=too-few-public-methods
|
||||||
|
class RestDataGet(object):
|
||||||
|
""" Class for handling the data retrieval with GET method. """
|
||||||
|
|
||||||
|
def __init__(self, resource, verify_ssl):
|
||||||
|
self._resource = resource
|
||||||
|
self._verify_ssl = verify_ssl
|
||||||
|
self.data = False
|
||||||
|
|
||||||
|
@Throttle(MIN_TIME_BETWEEN_UPDATES)
|
||||||
|
def update(self):
|
||||||
|
""" Gets the latest data from REST service with GET method. """
|
||||||
|
try:
|
||||||
|
response = requests.get(self._resource, timeout=10,
|
||||||
|
verify=self._verify_ssl)
|
||||||
|
self.data = response.text
|
||||||
|
except requests.exceptions.ConnectionError:
|
||||||
|
_LOGGER.error("No route to resource/endpoint: %s", self._resource)
|
||||||
|
self.data = False
|
||||||
|
|
||||||
|
|
||||||
|
# pylint: disable=too-few-public-methods
|
||||||
|
class RestDataPost(object):
|
||||||
|
""" Class for handling the data retrieval with POST method. """
|
||||||
|
|
||||||
|
def __init__(self, resource, payload, verify_ssl):
|
||||||
|
self._resource = resource
|
||||||
|
self._payload = payload
|
||||||
|
self._verify_ssl = verify_ssl
|
||||||
|
self.data = False
|
||||||
|
|
||||||
|
@Throttle(MIN_TIME_BETWEEN_UPDATES)
|
||||||
|
def update(self):
|
||||||
|
""" Gets the latest data from REST service with POST method. """
|
||||||
|
try:
|
||||||
|
response = requests.post(self._resource, data=self._payload,
|
||||||
|
timeout=10, verify=self._verify_ssl)
|
||||||
|
self.data = response.text
|
||||||
|
except requests.exceptions.ConnectionError:
|
||||||
|
_LOGGER.error("No route to resource/endpoint: %s", self._resource)
|
||||||
|
self.data = False
|
Loading…
x
Reference in New Issue
Block a user