Support Sonoff in DIY mode

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## [0.2.4] - 2020-01-18
+
+### Added
+
+- Support Sonoff in DIY mode
+
 ## [0.2.3] - 2020-01-18
 
 ### Added

README.md

@@ -9,6 +9,8 @@
 - работает с оригинальной прошивкой Sonoff, нет необходимости перепрошивать 
   устройства
 - работает по локальной сети, нет тормозов китайских серверов
+- работает с устройствами без DIY-режима
+- работает с устройствами в DIY-режиме
 - можно получить список устройств с серверов eWeLink, либо настроить его 
   вручную (список сохраняется локально и может больше не запрашиваться)
 - мгновенное получение нового состояния устройств по Multicast (привет 
@@ -86,6 +88,12 @@ sonoff:
         channels: [3, 4]
 ```
 
+Для устройств в режиме DIY хватит:
+
+```yaml
+sonoff:
+```
+
 ## Sonoff RF Bridge 433
 
 Хоть компонент и поддерживает обучение - рекомендуется обучать кнопки через 
@@ -94,7 +102,7 @@ sonoff:
 Компонент умеет как отправлять RF-сигналы, так и получать их, но только ранее обученные.
 
 При получении команды создаётся событие `sonoff.remote` с порядковым номером 
-кнопки и временем срабатывание (в UTC, присылает устройство).
+кнопки и временем срабатывания (в UTC, присылает устройство).
 
 `command` - порядковый номер изученной кнопки в приложении.
 

custom_components/sonoff/__init__.py

@@ -40,7 +40,7 @@ def setup(hass, hass_config):
 
     # load devices from configuration.yaml
     if not devices:
-        devices = config.get(CONF_DEVICES)
+        devices = config.get(CONF_DEVICES, {})
 
     # concat ewelink devices with yaml devices
     elif CONF_DEVICES in config:
@@ -53,15 +53,6 @@ def setup(hass, hass_config):
                 _LOGGER.debug(f"Add device config {deviceid}")
                 devices[deviceid] = devicecfg
 
-    if not devices:
-        _LOGGER.error("Empty device list")
-        return False
-
-    # add deviceid to all device config
-    for k, v in devices.items():
-        if 'deviceid' not in v:
-            v['deviceid'] = k
-
     hass.data[DOMAIN] = devices
 
     def add_device(devicecfg: dict, state: dict):
@@ -146,7 +137,6 @@ def add_service(self, zeroconf: Zeroconf, type_: str, name: str):
         host = str(ipaddress.ip_address(info.address))
         deviceid = properties['id']
 
-        # TODO: check no did
         device = self.devices.get(deviceid)
         if isinstance(device, EWeLinkDevice):
             # TODO: check update host
@@ -166,10 +156,15 @@ def add_service(self, zeroconf: Zeroconf, type_: str, name: str):
             # Fix Sonoff RF Bridge sintax bug
             if data.startswith(b'{"rf'):
                 data = data.replace(b'"="', b'":"')
-            state = json.loads(data)
-            _LOGGER.debug(f"State: {state}")
         else:
-            raise NotImplementedError()
+            data = ''.join([properties[f'data{i}'] for i in range(1, 4, 1)
+                            if f'data{i}' in properties])
+
+        state = json.loads(data)
+        _LOGGER.debug(f"State: {state}")
+
+        if 'deviceid' not in config:
+            config['deviceid'] = deviceid
 
         self.devices[deviceid] = EWeLinkDevice(host, config, state, zeroconf)
 
@@ -247,12 +242,12 @@ def update_service(self, zeroconf: Zeroconf, type_: str, name: str):
             # Fix Sonoff RF Bridge sintax bug
             if data.startswith(b'{"rf'):
                 data = data.replace(b'"="', b'":"')
-            data = json.loads(data)
-            _LOGGER.debug(f"Data: {data}")
         else:
-            raise NotImplementedError()
+            data = ''.join([properties[f'data{i}'] for i in range(1, 4, 1)
+                            if f'data{i}' in properties])
 
-        self.state = data
+        self.state = json.loads(data)
+        _LOGGER.debug(f"State: {self.state}")
 
         for handler in self._update_handlers:
             handler(self)
@@ -264,12 +259,15 @@ def send(self, command: str, data: dict):
         :param data: Данные для команды
         :return:
         """
-        payload = utils.encrypt({
+        payload = {
             'sequence': str(int(time.time())),
             'deviceid': self.deviceid,
             'selfApikey': '123',
             'data': data
-        }, self.devicekey)
+        }
+
+        if self.devicekey:
+            payload = utils.encrypt(payload, self.devicekey)
 
         _LOGGER.debug(f"Send {command} to {self.deviceid}: {payload}")
 

custom_components/sonoff/utils.py

@@ -10,6 +10,7 @@
 from typing import Optional
 
 import requests
+
 from Crypto.Cipher import AES
 from Crypto.Hash import MD5
 from Crypto.Random import get_random_bytes
@@ -100,8 +101,8 @@ def decrypt(payload: dict, devicekey: str):
         hash_.update(devicekey)
         key = hash_.digest()
 
-        encoded = ''.join([payload[f'data{i}'] for i in range(1, 4, 1) if
-                           f'data{i}' in payload])
+        encoded = ''.join([payload[f'data{i}'] for i in range(1, 4, 1)
+                           if f'data{i}' in payload])
 
         cipher = AES.new(key, AES.MODE_CBC, iv=b64decode(payload['iv']))
         ciphertext = b64decode(encoded)