jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-18 02:37:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3110 from home-assistant/rfxtrx_sensor_cleanup

Browse Source 
rfxtrx sensor clean up
This commit is contained in:
Daniel Høyer Iversen 2016-09-01 19:42:40 +02:00 committed by GitHub
commit 1094de7ad9
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
homeassistant/components/sensor/rfxtrx.py
View File

@ -9,6 +9,7 @@ import voluptuous as vol
import homeassistant.components.rfxtrx as rfxtrx import homeassistant.components.rfxtrx as rfxtrx
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.const import CONF_PLATFORM
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.util import slugify from homeassistant.util import slugify
from homeassistant.components.rfxtrx import ( from homeassistant.components.rfxtrx import (
@ -20,7 +21,7 @@ DEPENDENCIES = ['rfxtrx']
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
PLATFORM_SCHEMA = vol.Schema({ PLATFORM_SCHEMA = vol.Schema({
vol.Required("platform"): rfxtrx.DOMAIN, vol.Required(CONF_PLATFORM): rfxtrx.DOMAIN,
vol.Optional(CONF_DEVICES, default={}): vol.All(dict, rfxtrx.valid_sensor), vol.Optional(CONF_DEVICES, default={}): vol.All(dict, rfxtrx.valid_sensor),
vol.Optional(ATTR_AUTOMATIC_ADD, default=False): cv.boolean, vol.Optional(ATTR_AUTOMATIC_ADD, default=False): cv.boolean,
}, extra=vol.ALLOW_EXTRA) }, extra=vol.ALLOW_EXTRA)
@ -31,7 +32,7 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
# pylint: disable=too-many-locals # pylint: disable=too-many-locals
from RFXtrx import SensorEvent from RFXtrx import SensorEvent
sensors = [] sensors = []
for packet_id, entity_info in config['devices'].items(): for packet_id, entity_info in config[CONF_DEVICES].items():
event = rfxtrx.get_rfx_object(packet_id) event = rfxtrx.get_rfx_object(packet_id)
device_id = "sensor_" + slugify(event.device.id_string.lower()) device_id = "sensor_" + slugify(event.device.id_string.lower())
if device_id in rfxtrx.RFX_DEVICES: if device_id in rfxtrx.RFX_DEVICES:
@ -41,7 +42,7 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
sub_sensors = {} sub_sensors = {}
data_types = entity_info[ATTR_DATA_TYPE] data_types = entity_info[ATTR_DATA_TYPE]
if len(data_types) == 0: if len(data_types) == 0:
data_types = ["Unknown"] data_types = ['']
for data_type in DATA_TYPES: for data_type in DATA_TYPES:
if data_type in event.values: if data_type in event.values:
data_types = [data_type] data_types = [data_type]
@ -52,7 +53,6 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
sensors.append(new_sensor) sensors.append(new_sensor)
sub_sensors[_data_type] = new_sensor sub_sensors[_data_type] = new_sensor
rfxtrx.RFX_DEVICES[device_id] = sub_sensors rfxtrx.RFX_DEVICES[device_id] = sub_sensors
add_devices_callback(sensors) add_devices_callback(sensors)
def sensor_update(event): def sensor_update(event):
@ -75,7 +75,6 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
sensors[key].entity_id, sensors[key].entity_id,
} }
) )
return return
# Add entity if not exist and the automatic_add is True # Add entity if not exist and the automatic_add is True
@ -86,7 +85,7 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
_LOGGER.info("Automatic add rfxtrx.sensor: %s", _LOGGER.info("Automatic add rfxtrx.sensor: %s",
pkt_id) pkt_id)
data_type = "Unknown" data_type = ''
for _data_type in DATA_TYPES: for _data_type in DATA_TYPES:
if _data_type in event.values: if _data_type in event.values:
data_type = _data_type data_type = _data_type
@ -119,9 +118,9 @@ class RfxtrxSensor(Entity):
@property @property
def state(self): def state(self):
"""Return the state of the sensor.""" """Return the state of the sensor."""
if self.event and self.data_type in self.event.values: if not self.event:
return self.event.values[self.data_type] return None
return None return self.event.values.get(self.data_type)
@property @property
def name(self): def name(self):
@ -131,8 +130,9 @@ class RfxtrxSensor(Entity):
@property @property
def device_state_attributes(self): def device_state_attributes(self):
"""Return the state attributes.""" """Return the state attributes."""
if self.event: if not self.event:
return self.event.values return None
return self.event.values
@property @property
def unit_of_measurement(self): def unit_of_measurement(self):