Enhance mysensors sensor units and icons (#13365)

This commit is contained in:
Martin Hjelmare 2018-03-23 18:06:07 +01:00 committed by Fabian Affolter
parent 2c7bc6eaf8
commit ba7178dc0c

View file

@ -8,6 +8,31 @@ from homeassistant.components import mysensors
from homeassistant.components.sensor import DOMAIN
from homeassistant.const import TEMP_CELSIUS, TEMP_FAHRENHEIT
SENSORS = {
'V_TEMP': [None, 'mdi:thermometer'],
'V_HUM': ['%', 'mdi:water-percent'],
'V_DIMMER': ['%', 'mdi:percent'],
'V_LIGHT_LEVEL': ['%', 'white-balance-sunny'],
'V_DIRECTION': ['°', 'mdi:compass'],
'V_WEIGHT': ['kg', 'mdi:weight-kilogram'],
'V_DISTANCE': ['m', 'mdi:ruler'],
'V_IMPEDANCE': ['ohm', None],
'V_WATT': ['W', None],
'V_KWH': ['kWh', None],
'V_FLOW': ['m', None],
'V_VOLUME': ['', None],
'V_VOLTAGE': ['V', 'mdi:flash'],
'V_CURRENT': ['A', 'mdi:flash-auto'],
'V_PERCENTAGE': ['%', 'mdi:percent'],
'V_LEVEL': {
'S_SOUND': ['dB', 'mdi:volume-high'], 'S_VIBRATION': ['Hz', None],
'S_LIGHT_LEVEL': ['lux', 'white-balance-sunny']},
'V_ORP': ['mV', None],
'V_EC': ['μS/cm', None],
'V_VAR': ['var', None],
'V_VA': ['VA', None],
}
def setup_platform(hass, config, add_devices, discovery_info=None):
"""Set up the MySensors platform for sensors."""
@ -32,45 +57,29 @@ class MySensorsSensor(mysensors.MySensorsEntity):
"""Return the state of the device."""
return self._values.get(self.value_type)
@property
def icon(self):
"""Return the icon to use in the frontend, if any."""
_, icon = self._get_sensor_type()
return icon
@property
def unit_of_measurement(self):
"""Return the unit of measurement of this entity."""
set_req = self.gateway.const.SetReq
if (float(self.gateway.protocol_version) >= 1.5 and
set_req.V_UNIT_PREFIX in self._values):
return self._values[set_req.V_UNIT_PREFIX]
unit, _ = self._get_sensor_type()
return unit
def _get_sensor_type(self):
"""Return list with unit and icon of sensor type."""
pres = self.gateway.const.Presentation
set_req = self.gateway.const.SetReq
unit_map = {
set_req.V_TEMP: (TEMP_CELSIUS
if self.gateway.metric else TEMP_FAHRENHEIT),
set_req.V_HUM: '%',
set_req.V_DIMMER: '%',
set_req.V_LIGHT_LEVEL: '%',
set_req.V_DIRECTION: '°',
set_req.V_WEIGHT: 'kg',
set_req.V_DISTANCE: 'm',
set_req.V_IMPEDANCE: 'ohm',
set_req.V_WATT: 'W',
set_req.V_KWH: 'kWh',
set_req.V_FLOW: 'm',
set_req.V_VOLUME: '',
set_req.V_VOLTAGE: 'V',
set_req.V_CURRENT: 'A',
}
if float(self.gateway.protocol_version) >= 1.5:
if set_req.V_UNIT_PREFIX in self._values:
return self._values[
set_req.V_UNIT_PREFIX]
unit_map.update({
set_req.V_PERCENTAGE: '%',
set_req.V_LEVEL: {
pres.S_SOUND: 'dB', pres.S_VIBRATION: 'Hz',
pres.S_LIGHT_LEVEL: 'lux'}})
if float(self.gateway.protocol_version) >= 2.0:
unit_map.update({
set_req.V_ORP: 'mV',
set_req.V_EC: 'μS/cm',
set_req.V_VAR: 'var',
set_req.V_VA: 'VA',
})
unit = unit_map.get(self.value_type)
if isinstance(unit, dict):
unit = unit.get(self.child_type)
return unit
SENSORS[set_req.V_TEMP.name][0] = (
TEMP_CELSIUS if self.gateway.metric else TEMP_FAHRENHEIT)
sensor_type = SENSORS.get(set_req(self.value_type).name, [None, None])
if isinstance(sensor_type, dict):
sensor_type = sensor_type.get(pres(self.child_type).name)
return sensor_type