evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for sensor classes

Browse source
This commit is contained in:
Fabian Affolter 2016-03-25 18:34:58 +01:00
parent 70ce179224
commit f0970f4104
1 changed files with 20 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

25
homeassistant/components/binary_sensor/rest.py
View file

@ -6,7 +6,8 @@ https://home-assistant.io/components/binary_sensor.rest/
""" """
import logging import logging
from homeassistant.components.binary_sensor import BinarySensorDevice from homeassistant.components.binary_sensor import (BinarySensorDevice,
SENSOR_CLASSES)
from homeassistant.components.sensor.rest import RestData from homeassistant.components.sensor.rest import RestData
from homeassistant.const import CONF_VALUE_TEMPLATE from homeassistant.const import CONF_VALUE_TEMPLATE
from homeassistant.helpers import template from homeassistant.helpers import template
@ -19,12 +20,17 @@ DEFAULT_METHOD = 'GET'
# pylint: disable=unused-variable # pylint: disable=unused-variable
def setup_platform(hass, config, add_devices, discovery_info=None): def setup_platform(hass, config, add_devices, discovery_info=None):
"""Setup REST binary sensors.""" """Setup the REST binary sensor."""
resource = config.get('resource', None) resource = config.get('resource', None)
method = config.get('method', DEFAULT_METHOD) method = config.get('method', DEFAULT_METHOD)
payload = config.get('payload', None) payload = config.get('payload', None)
verify_ssl = config.get('verify_ssl', True) verify_ssl = config.get('verify_ssl', True)
sensor_class = config.get('sensor_class')
if sensor_class not in SENSOR_CLASSES:
_LOGGER.warning('Unknown sensor class: %s', sensor_class)
sensor_class = None
rest = RestData(method, resource, payload, verify_ssl) rest = RestData(method, resource, payload, verify_ssl)
rest.update() rest.update()
@ -33,19 +39,23 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
return False return False
add_devices([RestBinarySensor( add_devices([RestBinarySensor(
hass, rest, config.get('name', DEFAULT_NAME), hass,
rest,
config.get('name', DEFAULT_NAME),
sensor_class,
config.get(CONF_VALUE_TEMPLATE))]) config.get(CONF_VALUE_TEMPLATE))])
# pylint: disable=too-many-arguments # pylint: disable=too-many-arguments
class RestBinarySensor(BinarySensorDevice): class RestBinarySensor(BinarySensorDevice):
"""A REST binary sensor.""" """Representation of a REST binary sensor."""
def __init__(self, hass, rest, name, value_template): def __init__(self, hass, rest, name, sensor_class, value_template):
"""Initialize a REST binary sensor.""" """Initialize a REST binary sensor."""
self._hass = hass self._hass = hass
self.rest = rest self.rest = rest
self._name = name self._name = name
self._sensor_class = sensor_class
self._state = False self._state = False
self._value_template = value_template self._value_template = value_template
self.update() self.update()
@ -55,6 +65,11 @@ class RestBinarySensor(BinarySensorDevice):
"""Return the name of the binary sensor.""" """Return the name of the binary sensor."""
return self._name return self._name
@property
def sensor_class(self):
"""Return the class of this sensor."""
return self._sensor_class
@property @property
def is_on(self): def is_on(self):
"""Return true if the binary sensor is on.""" """Return true if the binary sensor is on."""