Add auto sensors creation

Author: AlexxIT

custom_components/sonoff/__init__.py

@@ -5,7 +5,7 @@
 from homeassistant.components.binary_sensor import DEVICE_CLASSES
 from homeassistant.const import CONF_USERNAME, CONF_PASSWORD, CONF_DEVICES, \
     CONF_NAME, CONF_DEVICE_CLASS, EVENT_HOMEASSISTANT_STOP, CONF_MODE, \
-    CONF_SCAN_INTERVAL, CONF_FORCE_UPDATE, CONF_EXCLUDE
+    CONF_SCAN_INTERVAL, CONF_FORCE_UPDATE, CONF_EXCLUDE, CONF_SENSORS
 from homeassistant.core import ServiceCall
 from homeassistant.helpers import config_validation as cv, discovery
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
@@ -40,6 +40,7 @@
         vol.Optional(CONF_DEFAULT_CLASS, default='switch'): cv.string,
         vol.Optional(CONF_SCAN_INTERVAL): cv.time_period,
         vol.Optional(CONF_FORCE_UPDATE): cv.ensure_list,
+        vol.Optional(CONF_SENSORS): cv.ensure_list,
         vol.Optional(CONF_DEBUG, default=False): cv.boolean,
         vol.Optional(CONF_DEVICES): {
             cv.string: vol.Schema({
@@ -110,6 +111,12 @@ async def async_setup(hass: HomeAssistantType, hass_config: dict):
     else:
         force_update = None
 
+    if CONF_SENSORS in config:
+        sensors = config[CONF_SENSORS]
+        _LOGGER.debug(f"Init auto sensors for: {sensors}")
+    else:
+        sensors = []
+
     def add_device(deviceid: str, state: dict, *args):
         device = registry.devices[deviceid]
 
@@ -166,6 +173,12 @@ def add_device(deviceid: str, state: dict, *args):
                 hass.async_create_task(discovery.async_load_platform(
                     hass, info.pop('component'), DOMAIN, info, hass_config))
 
+        for attribute in sensors:
+            if attribute in state:
+                info = {'deviceid': deviceid, 'attribute': attribute}
+                hass.async_create_task(discovery.async_load_platform(
+                    hass, 'sensor', DOMAIN, info, hass_config))
+
     async def send_command(call: ServiceCall):
         """Service for send raw command to device.
 

custom_components/sonoff/sensor.py

@@ -1,21 +1,28 @@
 from typing import Optional
 
-from homeassistant.const import DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, \
-    DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_ILLUMINANCE
+from homeassistant.const import DEVICE_CLASS_TEMPERATURE, \
+    DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_ILLUMINANCE, DEVICE_CLASS_POWER, \
+    DEVICE_CLASS_SIGNAL_STRENGTH
 from homeassistant.helpers.entity import Entity
 
 from . import DOMAIN, EWeLinkRegistry
 from .sonoff_main import EWeLinkDevice
 
-SONOFF_SC = {
-    'temperature': [DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS, None],
+SENSORS = {
+    'temperature': [DEVICE_CLASS_TEMPERATURE, '°C', None],
     # UNIT_PERCENTAGE is not on old versions
-    'humidity': [DEVICE_CLASS_HUMIDITY, "%", None],
+    'humidity': [DEVICE_CLASS_HUMIDITY, '%', None],
     'dusty': [None, None, 'mdi:cloud'],
     'light': [DEVICE_CLASS_ILLUMINANCE, None, None],
-    'noise': [None, None, 'mdi:bell-ring']
+    'noise': [None, None, 'mdi:bell-ring'],
+    'power': [DEVICE_CLASS_POWER, 'W', None],
+    'current': [DEVICE_CLASS_POWER, 'A', None],
+    'voltage': [DEVICE_CLASS_POWER, 'V', None],
+    'rssi': [DEVICE_CLASS_SIGNAL_STRENGTH, 'dBm', None]
 }
 
+SONOFF_SC = {'temperature', 'humidity', 'dusty', 'light', 'noise'}
+
 
 async def async_setup_platform(hass, config, add_entities,
                                discovery_info=None):
@@ -24,13 +31,19 @@ async def async_setup_platform(hass, config, add_entities,
 
     deviceid = discovery_info['deviceid']
     registry = hass.data[DOMAIN]
+
+    if 'attribute' in discovery_info:
+        add_entities([EWeLinkSensor(registry, deviceid,
+                                    discovery_info['attribute'])])
+        return
+
     device = registry.devices[deviceid]
     if device.get('uiid') == 18:
-        add_entities([SonoffSC(registry, deviceid, attr)
+        add_entities([EWeLinkSensor(registry, deviceid, attr)
                       for attr in SONOFF_SC])
 
 
-class SonoffSC(EWeLinkDevice, Entity):
+class EWeLinkSensor(EWeLinkDevice, Entity):
     _state = None
 
     def __init__(self, registry: EWeLinkRegistry, deviceid: str, attr: str):
@@ -76,12 +89,12 @@ def available(self) -> bool:
 
     @property
     def device_class(self):
-        return SONOFF_SC[self._attr][0]
+        return SENSORS[self._attr][0] if self._attr in SENSORS else None
 
     @property
     def unit_of_measurement(self):
-        return SONOFF_SC[self._attr][1]
+        return SENSORS[self._attr][1] if self._attr in SENSORS else None
 
     @property
     def icon(self):
-        return SONOFF_SC[self._attr][2]
+        return SENSORS[self._attr][2] if self._attr in SENSORS else None