Update the sensor classes in the demo binary_sensor

This updates the two demo sensors we have so they show contextual icons
in the UI.
pull/1308/head
Dan Smith 2016-02-18 17:59:58 -08:00
parent d93883f153
commit d57df2ddde
1 changed files with 9 additions and 3 deletions

View File

@ -9,17 +9,23 @@ from homeassistant.components.binary_sensor import BinarySensorDevice
def setup_platform(hass, config, add_devices, discovery_info=None):
""" Sets up the Demo binary sensors. """
add_devices([
DemoBinarySensor('Basement Floor Wet', False),
DemoBinarySensor('Movement Backyard', True),
DemoBinarySensor('Basement Floor Wet', False, 'moisture'),
DemoBinarySensor('Movement Backyard', True, 'motion'),
])
class DemoBinarySensor(BinarySensorDevice):
""" A Demo binary sensor. """
def __init__(self, name, state):
def __init__(self, name, state, sensor_class):
self._name = name
self._state = state
self._sensor_type = sensor_class
@property
def sensor_class(self):
""" Return our class. """
return self._sensor_type
@property
def should_poll(self):