fix config scan for ds2413

Улучшение поддержки I2C устройств

Спасибо большое!

.experiment.py

@@ -1,32 +1,61 @@
 import asyncio
+from urllib.parse import urlparse, parse_qsl
+from asyncio import Event, FIRST_COMPLETED
+import signal
+import typing
+from logging import getLogger, DEBUG
 
 
-async def handle_echo(reader: asyncio.StreamReader, writer: asyncio.StreamWriter):
-    data = await reader.read(100)
-    message = data.decode()
-    addr = writer.get_extra_info('peername')
-
-    print(f"Received {message!r} from {addr!r}")
-
-    print(f"Send: {message!r}")
-    ans = '''HTTP/1.1 200 OK\nContent-Length: 6\n\nhello\n'''.encode()
-    writer.write(ans)
-    await writer.drain()
-
-    print("Close the connection")
-    writer.transport.close()
-    writer.close()
-    await writer.wait_closed()
+stop = Event()
+loop = asyncio.get_event_loop()
+lg = getLogger(__name__)
+lg.setLevel(DEBUG)
 
 
-async def main():
+def make_handler(get_ans: typing.Callable[[dict], str]):
+
+    async def handler(
+            reader: asyncio.StreamReader,
+            writer: asyncio.StreamWriter,
+    ):
+        data = await reader.read(200)
+        print(data)
+        message = data.decode()
+        addr = writer.get_extra_info('peername')
+        lg.debug('process msg "%s" from %s', message, addr)
+        try:
+            (_, p, *_) = message.split(' ')
+            p = dict(parse_qsl(urlparse(p).query))
+            lg.debug('query %s', p)
+            ans = get_ans(p)
+            ans = f'''HTTP/1.1 200 OK\nContent-Length: {len(ans)}\n\n{ans}'''.encode()  # \nContent-Length: 6
+            ans = b'HTTP/1.1 200 OK\r\n\r\n7:2'
+            print(ans)
+        except Exception as exc:
+            print(exc)
+            lg.exception('process msg "%s" from %s', message, addr)
+            ans = '''HTTP/1.1 500\n\n'''.encode()
+        writer.write(ans)
+        await writer.drain()
+        # writer.transport.close()
+        writer.close()
+        await writer.wait_closed()
+    return handler
+
+
+async def serve():
     server = await asyncio.start_server(
-        handle_echo, '127.0.0.1', 8888)
-
+        make_handler(lambda x: '7:2'),
+        host='0.0.0.0',
+        port=1111,
+    )
     addr = server.sockets[0].getsockname()
     print(f'Serving on {addr}')
-
     async with server:
-        await server.serve_forever()
+        await asyncio.wait((server.serve_forever(), stop.wait()), return_when=FIRST_COMPLETED)
 
-asyncio.run(main())
+if __name__ == '__main__':
+    loop.add_signal_handler(
+        signal.SIGINT, stop.set
+    )
+    loop.run_until_complete(serve())

.github/ISSUE_TEMPLATE/bug-report.md

@@ -8,19 +8,20 @@ assignees: ''
 ---
 
 **Описание**
-A clear and concise description of what the bug is.
+Описание проблемы
 
 **Версии систем**
 Enviroment: raspberry/linux/windows/macos/docker
 HA version: 
 mega_hacs version:
 megad firmware version:
+используется mqtt: true/false
 
 **Ожидаемое поведение**
-A clear and concise description of what you expected to happen.
+Описание правильного поведения
 
 **Screenshots**
 If applicable, add screenshots to help explain your problem.
 
 **LOG**
-Прочитайте в документации как включить подробный лог интеграции и приложите его здесь
+Просьба прикладывать детальный лог, подробная инструкция как включать отладку по [ссылке](https://github.com/andvikt/mega_hacs/wiki/Отладка)

custom_components/mega/__init__.py

@@ -17,37 +17,47 @@ from homeassistant.components import mqtt
 from homeassistant.config_entries import ConfigEntry
 from .const import DOMAIN, CONF_INVERT, CONF_RELOAD, PLATFORMS, CONF_PORTS, CONF_CUSTOM, CONF_SKIP, CONF_PORT_TO_SCAN, \
     CONF_MQTT_INPUTS, CONF_HTTP, CONF_RESPONSE_TEMPLATE, CONF_ACTION, CONF_GET_VALUE, CONF_ALLOW_HOSTS, \
-    CONF_CONV_TEMPLATE
+    CONF_CONV_TEMPLATE, CONF_ALL, CONF_FORCE_D
 from .hub import MegaD
 from .config_flow import ConfigFlow
 from .http import MegaView
 
 _LOGGER = logging.getLogger(__name__)
 
+CUSTOMIZE_PORT = vol.Schema({
+    vol.Optional(CONF_SKIP, description='исключить порт из сканирования', default=False): bool,
+    vol.Optional(CONF_INVERT, default=False): bool,
+    vol.Optional(CONF_NAME): vol.Any(str, {
+        vol.Required(str): str
+    }),
+    vol.Optional(CONF_DOMAIN): vol.Any('light', 'switch'),
+    vol.Optional(CONF_UNIT_OF_MEASUREMENT, description='единицы измерений, либо строка либо мепинг'):
+        vol.Any(str, {
+            vol.Required(str): str
+        }),
+    vol.Optional(
+        CONF_RESPONSE_TEMPLATE,
+        description='шаблон ответа когда на этот порт приходит'
+                    'сообщение из меги '): cv.template,
+    vol.Optional(CONF_ACTION): cv.script_action, # пока не реализовано
+    vol.Optional(CONF_GET_VALUE, default=True): bool,
+    vol.Optional(CONF_CONV_TEMPLATE): cv.template
+})
+CUSTOMIZE_DS2413 = vol.Schema({
+    vol.Optional(str.lower, description='адрес и индекс устройства'): CUSTOMIZE_PORT
+})
+
 CONFIG_SCHEMA = vol.Schema(
     {
         DOMAIN: {
             vol.Optional(CONF_ALLOW_HOSTS): [str],
+            # vol.Optional(CONF_FORCE_D, description='Принудительно слать d после срабатывания входа', default=False): bool,
             vol.Required(str, description='id меги из веб-интерфейса'): {
-                vol.Optional(int, description='номер порта'): {
-                    vol.Optional(CONF_SKIP, description='исключить порт из сканирования', default=False): bool,
-                    vol.Optional(CONF_INVERT, default=False): bool,
-                    vol.Optional(CONF_NAME): vol.Any(str, {
-                        vol.Required(str): str
-                    }),
-                    vol.Optional(CONF_DOMAIN): vol.Any('light', 'switch'),
-                    vol.Optional(CONF_UNIT_OF_MEASUREMENT, description='единицы измерений, либо строка либо мепинг'):
-                        vol.Any(str, {
-                            vol.Required(str): str
-                        }),
-                    vol.Optional(
-                        CONF_RESPONSE_TEMPLATE,
-                        description='шаблон ответа когда на этот порт приходит'
-                                    'сообщение из меги '): cv.template,
-                    vol.Optional(CONF_ACTION): cv.script_action,
-                    vol.Optional(CONF_GET_VALUE, default=True): bool,
-                    vol.Optional(CONF_CONV_TEMPLATE): cv.template
-                }
+                vol.Optional(CONF_FORCE_D, description='Принудительно слать d после срабатывания входа', default=False): bool,
+                vol.Optional(int, description='номер порта'): vol.Any(
+                    CUSTOMIZE_PORT,
+                    CUSTOMIZE_DS2413,
+                )
             }
         }
     },
@@ -65,6 +75,7 @@ async def async_setup(hass: HomeAssistant, config: dict):
     """YAML-конфигурация содержит только кастомизации портов"""
     hass.data[DOMAIN] = {CONF_CUSTOM: config.get(DOMAIN, {})}
     hass.data[DOMAIN][CONF_HTTP] = view = MegaView(cfg=config.get(DOMAIN, {}))
+    hass.data[DOMAIN][CONF_ALL] = {}
     view.allowed_hosts |= set(config.get(DOMAIN, {}).get(CONF_ALLOW_HOSTS, []))
     hass.http.register_view(view)
     hass.services.async_register(
@@ -75,12 +86,12 @@ async def async_setup(hass: HomeAssistant, config: dict):
     hass.services.async_register(
         DOMAIN, 'get_port', partial(_get_port, hass), schema=vol.Schema({
             vol.Optional('mega_id'): str,
-            vol.Optional('port'): int,
+            vol.Optional('port'): vol.Any(int, [int]),
         })
     )
     hass.services.async_register(
         DOMAIN, 'run_cmd', partial(_run_cmd, hass), schema=vol.Schema({
-            vol.Required('port'): int,
+            vol.Optional('port'): int,
             vol.Required('cmd'): str,
             vol.Optional('mega_id'): str,
         })
@@ -115,6 +126,7 @@ async def _add_mega(hass: HomeAssistant, entry: ConfigEntry):
     hub = await get_hub(hass, entry)
     hass.data[DOMAIN][id] = hass.data[DOMAIN]['__def'] = hub
     hass.data[DOMAIN][entry.data.get(CONF_HOST)] = hub
+    hass.data[DOMAIN][CONF_ALL][id] = hub
     if not await hub.authenticate():
         raise Exception("not authentificated")
     mid = await hub.get_mqtt_id()
@@ -164,6 +176,7 @@ async def async_remove_entry(hass, entry) -> None:
     _LOGGER.debug(f'remove {id}')
     _hubs.pop(id, None)
     hass.data[DOMAIN].pop(id, None)
+    hass.data[DOMAIN][CONF_ALL].pop(id, None)
     task: asyncio.Task = _POLL_TASKS.pop(id, None)
     if task is not None:
         task.cancel()
@@ -208,27 +221,32 @@ async def _get_port(hass: HomeAssistant, call: ServiceCall):
     if mega_id:
         hub: MegaD = hass.data[DOMAIN][mega_id]
         if port is None:
-            await hub.get_all_ports()
-        else:
+            await hub.get_all_ports(check_skip=True)
+        elif isinstance(port, int):
             await hub.get_port(port)
+        elif isinstance(port, list):
+            for x in port:
+                await hub.get_port(x)
     else:
-        for hub in hass.data[DOMAIN].values():
+        for hub in hass.data[DOMAIN][CONF_ALL].values():
             if not isinstance(hub, MegaD):
                 continue
             if port is None:
-                await hub.get_all_ports()
-            else:
+                await hub.get_all_ports(check_skip=True)
+            elif isinstance(port, int):
                 await hub.get_port(port)
+            elif isinstance(port, list):
+                for x in port:
+                    await hub.get_port(x)
 
 
 @bind_hass
 async def _run_cmd(hass: HomeAssistant, call: ServiceCall):
-    port = call.data.get('port')
     mega_id = call.data.get('mega_id')
     cmd = call.data.get('cmd')
     if mega_id:
         hub: MegaD = hass.data[DOMAIN][mega_id]
-        await hub.send_command(port=port, cmd=cmd)
+        await hub.request(cmd=cmd)
     else:
         for hub in hass.data[DOMAIN].values():
-            await hub.send_command(port=port, cmd=cmd)
+            await hub.request(cmd=cmd)

custom_components/mega/binary_sensor.py

@@ -16,7 +16,8 @@ from homeassistant.const import (
     CONF_ENTITY_ID,
 )
 from homeassistant.core import HomeAssistant
-from .const import EVENT_BINARY_SENSOR, DOMAIN, CONF_CUSTOM, CONF_SKIP
+from homeassistant.helpers.template import Template
+from .const import EVENT_BINARY_SENSOR, DOMAIN, CONF_CUSTOM, CONF_SKIP, CONF_INVERT, CONF_RESPONSE_TEMPLATE
 from .entities import  MegaPushEntity
 from .hub import MegaD
 
@@ -71,6 +72,10 @@ class MegaBinarySensor(BinarySensorEntity, MegaPushEntity):
     def state_attributes(self):
         return self._attrs
 
+    @property
+    def invert(self):
+        return self.customize.get(CONF_INVERT, False)
+
     @property
     def is_on(self) -> bool:
         val = self.mega.values.get(self.port, {}).get("value") \
@@ -79,10 +84,26 @@ class MegaBinarySensor(BinarySensorEntity, MegaPushEntity):
             return self._state == 'ON'
         elif val is not None:
             if val in ['ON', 'OFF']:
-                return val == 'ON'
+                return val == 'ON' if not self.invert else val == 'OFF'
             else:
-                return val != 1
+                return val != 1 if not self.invert else val == 1
 
     def _update(self, payload: dict):
         self.mega.values[self.port] = payload
+        if not self.mega.mqtt_inputs:
+            return
+
+        template: Template = self.customize.get(CONF_RESPONSE_TEMPLATE, None)
+        if template is not None:
+            template.hass = self.hass
+            ret = template.async_render(payload)
+            self.mega.lg.debug(f'response: %s', ret)
+            self.hass.async_create_task(
+                self.mega.request(pt=self.port, cmd=ret)
+            )
+        elif self.mega.force_d:
+            self.mega.lg.debug(f'response d')
+            self.hass.async_create_task(
+                self.mega.request(pt=self.port, cmd='d')
+            )
 

custom_components/mega/config_flow.py

@@ -10,7 +10,7 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_HOST, CONF_ID, CONF_PASSWORD, CONF_SCAN_INTERVAL
 from homeassistant.core import callback, HomeAssistant
 from .const import DOMAIN, CONF_PORT_TO_SCAN, CONF_RELOAD, PLATFORMS, CONF_MQTT_INPUTS, \
-    CONF_NPORTS, CONF_UPDATE_ALL  # pylint:disable=unused-import
+    CONF_NPORTS, CONF_UPDATE_ALL, CONF_POLL_OUTS  # pylint:disable=unused-import
 from .hub import MegaD
 from . import exceptions
 
@@ -18,10 +18,11 @@ _LOGGER = logging.getLogger(__name__)
 
 STEP_USER_DATA_SCHEMA = vol.Schema(
     {
-        vol.Required(CONF_ID, default='def'): str,
+        vol.Required(CONF_ID, default='mega'): str,
         vol.Required(CONF_HOST, default="192.168.0.14"): str,
         vol.Required(CONF_PASSWORD, default="sec"): str,
         vol.Optional(CONF_SCAN_INTERVAL, default=0): int,
+        vol.Optional(CONF_POLL_OUTS, default=False): bool,
         vol.Optional(CONF_PORT_TO_SCAN, default=0): int,
         vol.Optional(CONF_MQTT_INPUTS, default=True): bool,
         vol.Optional(CONF_NPORTS, default=37): int,
@@ -56,7 +57,7 @@ async def validate_input(hass: core.HomeAssistant, data):
 class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
     """Handle a config flow for mega."""
 
-    VERSION = 4
+    VERSION = 6
     CONNECTION_CLASS = config_entries.CONN_CLASS_ASSUMED
 
     async def async_step_user(self, user_input=None):
@@ -131,6 +132,7 @@ class OptionsFlowHandler(config_entries.OptionsFlow):
             step_id="init",
             data_schema=vol.Schema({
                 vol.Optional(CONF_SCAN_INTERVAL, default=e.get(CONF_SCAN_INTERVAL, 0)): int,
+                vol.Optional(CONF_POLL_OUTS, default=e.get(CONF_POLL_OUTS, False)): bool,
                 vol.Optional(CONF_PORT_TO_SCAN, default=e.get(CONF_PORT_TO_SCAN, 0)): int,
                 vol.Optional(CONF_MQTT_INPUTS, default=e.get(CONF_MQTT_INPUTS, True)): bool,
                 vol.Optional(CONF_NPORTS, default=e.get(CONF_NPORTS, 37)): int,

custom_components/mega/const.py

@@ -16,6 +16,7 @@ CONF_INVERT = 'invert'
 CONF_PORTS = 'ports'
 CONF_CUSTOM = '__custom'
 CONF_HTTP = '__http'
+CONF_ALL = '__all'
 CONF_SKIP = 'skip'
 CONF_MQTT_INPUTS = 'mqtt_inputs'
 CONF_NPORTS = 'nports'
@@ -25,6 +26,8 @@ CONF_UPDATE_ALL = 'update_all'
 CONF_GET_VALUE = 'get_value'
 CONF_ALLOW_HOSTS = 'allow_hosts'
 CONF_CONV_TEMPLATE = 'conv_template'
+CONF_POLL_OUTS = 'poll_outs'
+CONF_FORCE_D = 'force_d'
 PLATFORMS = [
     "light",
     "switch",
@@ -40,5 +43,6 @@ LONG = 'long'
 RELEASE = 'release'
 LONG_RELEASE = 'long_release'
 PRESS = 'press'
+LUX = 'lux'
 SINGLE_CLICK = 'single'
 DOUBLE_CLICK = 'double'

custom_components/mega/entities.py

@@ -43,6 +43,9 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
             id_suffix=None,
             name=None,
             unique_id=None,
+            http_cmd='get',
+            addr: str=None,
+            index=None,
     ):
         super().__init__(mega.updater)
         self._state: State = None
@@ -57,6 +60,9 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
         self._name = name or f"{mega.id}_{port}" + \
                             (f"_{id_suffix}" if id_suffix else "")
         self._customize: dict = None
+        self.http_cmd = http_cmd
+        self.index = index
+        self.addr = addr
 
     @property
     def customize(self):
@@ -66,7 +72,11 @@ class BaseMegaEntity(CoordinatorEntity, RestoreEntity):
             c = self.hass.data.get(DOMAIN, {}).get(CONF_CUSTOM) or {}
             c = c.get(self._mega_id) or {}
             c = c.get(self.port) or {}
+            if self.addr is not None and self.index is not None and isinstance(c, dict):
+                idx = self.addr.lower() + f'_a' if self.index == 0 else '_b'
+                c = c.get(idx, {})
             self._customize = c
+
         return self._customize
 
     @property
@@ -208,12 +218,18 @@ class MegaOutPort(MegaPushEntity):
         self._is_on = None
         self.dimmer = dimmer
 
+    # @property
+    # def assumed_state(self) -> bool:
+    #     return True if self.index is not None or self.mega.mqtt is None else False
+
     @property
     def invert(self):
         return self.customize.get(CONF_INVERT, False)
 
     @property
     def brightness(self):
+        if not self.dimmer:
+            return
         val = self.mega.values.get(self.port, {}).get("value")
         if val is None and self._state is not None:
             return self._state.attributes.get("brightness")
@@ -232,11 +248,33 @@ class MegaOutPort(MegaPushEntity):
             return self._state == 'ON'
         elif val is not None:
             val = val.get("value")
+            if self.index and self.addr:
+                _val = val.get(self.addr)
+                if not isinstance(val, str):
+                    self.mega.lg.warning(f'{self} has wrong state: {val}')
+                    return
+                _val = _val.split('/')
+                if len(_val) >= 2:
+                    val = val[self.index]
+                else:
+                    self.mega.lg.warning(f'{self} has wrong state: {val}')
+                    return
+            elif self.index and self.addr is None:
+                self.mega.lg.warning(f'{self} does not has addr')
+                return
+
             if not self.invert:
                 return val == 'ON' or str(val) == '1' or (safe_int(val) is not None and safe_int(val) > 0)
             else:
                 return val == 'OFF' or str(val) == '0' or (safe_int(val) is not None and safe_int(val) == 0)
 
+    @property
+    def cmd_port(self):
+        if self.index is not None:
+            return f'{self.port}A' if self.index == 0 else f'{self.port}B'
+        else:
+            return self.port
+
     async def async_turn_on(self, brightness=None, **kwargs) -> None:
         brightness = brightness or self.brightness or 255
 
@@ -246,17 +284,39 @@ class MegaOutPort(MegaPushEntity):
             cmd = brightness
         else:
             cmd = 1 if not self.invert else 0
-        await self.mega.send_command(self.port, f"{self.port}:{cmd}")
-        self.mega.values[self.port] = {'value': cmd}
+        cmd = {"cmd": f"{self.cmd_port}:{cmd}"}
+        if self.addr:
+            cmd['addr'] = self.addr
+        await self.mega.request(**cmd)
+        if self.index is not None:
+            # обновление текущего стейта для ds2413
+            await self.mega.get_port(
+                port=self.port,
+                force_http=True,
+                conv=False,
+                http_cmd='list',
+            )
+        else:
+            self.mega.values[self.port] = {'value': cmd}
         await self.get_state()
 
-
     async def async_turn_off(self, **kwargs) -> None:
 
         cmd = "0" if not self.invert else "1"
-
-        await self.mega.send_command(self.port, f"{self.port}:{cmd}")
-        self.mega.values[self.port] = {'value': cmd}
+        cmd = {"cmd": f"{self.cmd_port}:{cmd}"}
+        if self.addr:
+            cmd['addr'] = self.addr
+        await self.mega.request(**cmd)
+        if self.index is not None:
+            # обновление текущего стейта для ds2413
+            await self.mega.get_port(
+                port=self.port,
+                force_http=True,
+                conv=False,
+                http_cmd='list',
+            )
+        else:
+            self.mega.values[self.port] = {'value': cmd}
         await self.get_state()
 
 

custom_components/mega/exceptions.py

@@ -15,3 +15,7 @@ class DuplicateId(exceptions.HomeAssistantError):
 
 class InvalidAuth(exceptions.HomeAssistantError):
     """Error to indicate there is invalid auth."""
+
+
+class NoPort(Exception):
+    pass

custom_components/mega/http.py

@@ -8,16 +8,15 @@ from aiohttp.web_request import Request
 from aiohttp.web_response import Response
 
 from homeassistant.helpers.template import Template
-from .const import EVENT_BINARY_SENSOR, DOMAIN, CONF_RESPONSE_TEMPLATE
 from homeassistant.components.http import HomeAssistantView
 from homeassistant.core import HomeAssistant
+from .const import EVENT_BINARY_SENSOR, DOMAIN, CONF_RESPONSE_TEMPLATE
 from .tools import make_ints
 from . import hub as h
 _LOGGER = logging.getLogger(__name__).getChild('http')
 
 
 class MegaView(HomeAssistantView):
-    """Handle Yandex Smart Home unauthorized requests."""
 
     url = '/mega'
     name = 'mega'
@@ -31,8 +30,8 @@ class MegaView(HomeAssistantView):
             mid: {
                 pt: cfg[mid][pt][CONF_RESPONSE_TEMPLATE]
                 for pt in cfg[mid]
-                if CONF_RESPONSE_TEMPLATE in cfg[mid][pt]
-            } for mid in cfg
+                if isinstance(pt, int) and CONF_RESPONSE_TEMPLATE in cfg[mid][pt]
+            } for mid in cfg if isinstance(cfg[mid], dict)
         }
         _LOGGER.debug('templates: %s', self.templates)
 
@@ -62,6 +61,10 @@ class MegaView(HomeAssistantView):
         make_ints(data)
         port = data.get('pt')
         data = data.copy()
+        update_all = True
+        if 'v' in data:
+            update_all = False
+            data['value'] = data.pop('v')
         data['mega_id'] = hub.id
         ret = 'd'
         if port is not None:
@@ -69,7 +72,7 @@ class MegaView(HomeAssistantView):
             for cb in self.callbacks[hub.id][port]:
                 cb(data)
             template: Template = self.templates.get(hub.id, {}).get(port)
-            if hub.update_all:
+            if hub.update_all and update_all:
                 asyncio.create_task(self.later_update(hub))
             if template is not None:
                 template.hass = hass

custom_components/mega/hub.py

@@ -10,32 +10,49 @@ import json
 
 from bs4 import BeautifulSoup
 from homeassistant.components import mqtt
-from homeassistant.const import DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_HUMIDITY
+from homeassistant.const import (
+    DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_PRESSURE,
+    DEVICE_CLASS_ILLUMINANCE, TEMP_CELSIUS, PERCENTAGE, LIGHT_LUX
+)
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity import Entity
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
-from .const import TEMP, HUM, PATT_SPLIT, DOMAIN, CONF_HTTP, EVENT_BINARY_SENSOR
-from .entities import set_events_off
-from .exceptions import CannotConnect
+from .const import (
+    TEMP, HUM, PRESS,
+    LUX, PATT_SPLIT, DOMAIN,
+    CONF_HTTP, EVENT_BINARY_SENSOR, CONF_CUSTOM, CONF_FORCE_D
+)
+from .entities import set_events_off, BaseMegaEntity
+from .exceptions import CannotConnect, NoPort
 from .tools import make_ints
 
 TEMP_PATT = re.compile(r'temp:([01234567890\.]+)')
 HUM_PATT = re.compile(r'hum:([01234567890\.]+)')
+PRESS_PATT = re.compile(r'press:([01234567890\.]+)')
+LUX_PATT = re.compile(r'lux:([01234567890\.]+)')
 PATTERNS = {
     TEMP: TEMP_PATT,
     HUM: HUM_PATT,
+    PRESS: PRESS_PATT,
+    LUX: LUX_PATT
 }
 UNITS = {
-    TEMP: '°C',
-    HUM: '%'
+    TEMP: TEMP_CELSIUS, 
+    HUM: PERCENTAGE,
+    PRESS: 'mmHg',
+    LUX: LIGHT_LUX
 }
 CLASSES = {
     TEMP: DEVICE_CLASS_TEMPERATURE,
-    HUM: DEVICE_CLASS_HUMIDITY
+    HUM: DEVICE_CLASS_HUMIDITY,
+    PRESS: DEVICE_CLASS_PRESSURE,
+    LUX: DEVICE_CLASS_ILLUMINANCE
+}
+I2C_DEVICE_TYPES = {
+    "2":  LUX,  # BH1750
+    "3":  LUX,  # TSL2591
+    "7":  LUX,  # MAX44009
+    "70": LUX,  # OPT3001
 }
-
-class NoPort(Exception):
-    pass
 
 
 class MegaD:
@@ -57,6 +74,7 @@ class MegaD:
             nports=38,
             inverted: typing.List[int] = None,
             update_all=True,
+            poll_outs=False,
             **kwargs,
     ):
         """Initialize."""
@@ -66,6 +84,7 @@ class MegaD:
                 self.http.allowed_hosts |= {host}
         else:
             self.http = None
+        self.poll_outs = poll_outs
         self.update_all = update_all if update_all is not None else True
         self.nports = nports
         self.mqtt_inputs = mqtt_inputs
@@ -81,7 +100,7 @@ class MegaD:
         self._notif_lck = asyncio.Lock()
         self.cnd = asyncio.Condition()
         self.online = True
-        self.entities: typing.List[Entity] = []
+        self.entities: typing.List[BaseMegaEntity] = []
         self.poll_interval = scan_interval
         self.subs = None
         self.lg: logging.Logger = lg.getChild(self.id)
@@ -91,6 +110,7 @@ class MegaD:
         self.last_update = datetime.now()
         self._callbacks: typing.DefaultDict[int, typing.List[typing.Callable[[dict], typing.Coroutine]]] = defaultdict(list)
         self._loop = loop
+        self._customize = None
         self.values = {}
         self.last_port = None
         self.updater = DataUpdateCoordinator(
@@ -138,6 +158,18 @@ class MegaD:
             await self.get_port(x.port, force_http=True, http_cmd=x.http_cmd)
             ports.append(x.port)
 
+    @property
+    def customize(self):
+        if self._customize is None:
+            c = self.hass.data.get(DOMAIN, {}).get(CONF_CUSTOM) or {}
+            c = c.get(self.id) or {}
+            self._customize = c
+        return self._customize
+
+    @property
+    def force_d(self):
+        return self.customize.get(CONF_FORCE_D, False)
+
     @property
     def is_online(self):
         return (datetime.now() - self.last_update).total_seconds() < (self.poll_interval + 10)
@@ -161,15 +193,29 @@ class MegaD:
 
     async def poll(self):
         """
-        Send get port 0 every poll_interval. When answer is received, mega.<id> becomes online else mega.<id> becomes
-        offline
+        Polling ports
         """
         self.lg.debug('poll')
+        ds2413_polled = []
+        for x in self.entities:
+            # обновление ds2413 устройств
+            if x.http_cmd == 'ds2413':
+                self.lg.debug(f'poll ds2413 for {x.entity_id}')
+                if x.port in ds2413_polled:
+                    continue
+                await self.get_port(
+                    port=x.port,
+                    force_http=True,
+                    http_cmd='list',
+                    conv=False
+                )
+                ds2413_polled.append(x.port)
         if self.mqtt is None:
             await self.get_all_ports()
             await self.get_sensors(only_list=True)
-            return
-        if len(self.sensors) > 0:
+        elif self.poll_outs:
+            await self.get_all_ports(check_skip=True)
+        elif len(self.sensors) > 0:
             await self.get_sensors()
         else:
             await self.get_port(self.port_to_scan)
@@ -212,7 +258,7 @@ class MegaD:
         if 'busy' in ret:
             return None
         if ':' in ret:
-            ret = PATT_SPLIT.split(ret)
+            ret = ret.split(';')
             ret = {'value': dict([
                 x.split(':') for x in ret if x.count(':') == 1
             ])}
@@ -224,14 +270,14 @@ class MegaD:
             ret = {'value': ret}
         return ret
 
-    async def get_port(self, port, force_http=False, http_cmd='get'):
+    async def get_port(self, port, force_http=False, http_cmd='get', conv=True):
         """
         Запрос состояния порта. Состояние всегда возвращается в виде объекта, всегда сохраняется в центральное
         хранилище values
         """
         self.lg.debug(f'get port %s', port)
         if self.mqtt is None or force_http:
-            if http_cmd == 'list':
+            if http_cmd == 'list' and conv:
                 await self.request(pt=port, cmd='conv')
                 await asyncio.sleep(1)
             ret = self.parse_response(await self.request(pt=port, cmd=http_cmd))
@@ -259,15 +305,24 @@ class MegaD:
                 except asyncio.TimeoutError:
                     self.lg.error(f'timeout when getting port {port}')
 
-    async def get_all_ports(self, only_out=False):
+    @property
+    def ports(self):
+        return {e.port for e in self.entities}
+
+    async def get_all_ports(self, only_out=False, check_skip=False):
         if not self.mqtt_inputs:
             ret = await self.request(cmd='all')
             for port, x in enumerate(ret.split(';')):
+                if check_skip and not port in self.ports:
+                    continue
                 ret = self.parse_response(x)
                 self.values[port] = ret
-        else:
+        elif not check_skip:
             for x in range(self.nports + 1):
                 await self.get_port(x)
+        else:
+            for x in self.ports:
+                await self.get_port(x)
 
     async def reboot(self, save=True):
         await self.save()
@@ -376,9 +431,12 @@ class MegaD:
                     m = m.find(selected=True)['value']
                 self._scanned[port] = (pty, m)
                 return pty, m
-            elif pty == '2':
-                self._scanned[port] = (pty, '0')
-                return pty, '0'
+            elif pty in ('2', '4'):  # эта часть не очень проработана, тут есть i2c который может работать неправильно
+                m = tree.find('select', attrs={'name': 'd'})
+                if m:
+                    m = m.find(selected=True)['value']
+                self._scanned[port] = (pty, m or '0')
+                return pty, m or '0'
 
     async def scan_ports(self, nports=37):
         for x in range(0, nports+1):
@@ -394,7 +452,19 @@ class MegaD:
                 ret['binary_sensor'][port].append({})
             elif pty == "1" and (m in ['0', '1', '3'] or m is None):
                 ret['light'][port].append({'dimmer': m == '1'})
-            elif pty in ('3', '2'):
+            elif pty == "1" and m == "2":
+                # ds2413
+                _data = await self.get_port(port=port, force_http=True, http_cmd='list', conv=False)
+                data = _data.get('value', {})
+                if not isinstance(data, dict):
+                    self.lg.warning(f'can not add ds2413 on port {port}, it has wrong data: {_data}')
+                    continue
+                for addr, state in data.items():
+                    ret['light'][port].extend([
+                        {"index": 0, "addr": addr, "id_suffix": f'{addr}_a', "http_cmd": 'ds2413'},
+                        {"index": 1, "addr": addr, "id_suffix": f'{addr}_b', "http_cmd": 'ds2413'},
+                    ])
+            elif pty in ('3', '2', '4'):
                 try:
                     http_cmd = 'get'
                     values = await self.get_port(port, force_http=True)
@@ -413,7 +483,10 @@ class MegaD:
                 if isinstance(values, str) and TEMP_PATT.search(values):
                     values = {TEMP: values}
                 elif not isinstance(values, dict):
-                    values = {None: values}
+                    if pty == '4' and m in I2C_DEVICE_TYPES:
+                        values = {I2C_DEVICE_TYPES[m]: values}
+                    else:
+                        values = {None: values}
                 for key in values:
                     self.lg.debug(f'add sensor {key}')
                     ret['sensor'][port].append(dict(

custom_components/mega/manifest.json

@@ -15,5 +15,5 @@
     "@andvikt"
   ],
   "issue_tracker": "https://github.com/andvikt/mega_hacs/issues",
-  "version": "v0.3.11"
+  "version": "v0.4.1b3"
 }

custom_components/mega/sensor.py

@@ -102,7 +102,6 @@ class Mega1WSensor(MegaPushEntity):
             unit_of_measurement,
             device_class,
             key=None,
-            http_cmd='get',
             *args,
             **kwargs
     ):
@@ -118,7 +117,6 @@ class Mega1WSensor(MegaPushEntity):
         self._device_class = device_class
         self._unit_of_measurement = unit_of_measurement
         self.mega.sensors.append(self)
-        self.http_cmd = http_cmd
 
     @property
     def unit_of_measurement(self):

custom_components/mega/services.yaml

@@ -15,7 +15,7 @@ get_port:
       description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
       example: "mega"
     port:
-      description: Номер порта (если не заполнять, будут запрошены все порты сразу)
+      description: Номер порта или список портов (если не заполнять, будут запрошены все порты сразу)
       example: 1
 
 run_cmd:
@@ -25,9 +25,10 @@ run_cmd:
     mega_id:
       description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
       example: "mega"
-    port:
-      description: Номер порта (это не порт, которым мы управляем, а порт с которого шлем команду)
-      example: 1
     cmd:
       description: Любая поддерживаемая мегой команда
       example: "1:0"
+    port:
+      description: (Deprecated, больше не нужен) Номер порта (это не порт, которым мы управляем, а порт с которого шлем команду)
+      example: 1
+

custom_components/mega/strings.json

@@ -14,7 +14,8 @@
           "invert": "[%key:common::config_flow::data::invert%]",
           "mqtt_inputs": "[%key:common::config_flow::data::mqtt_inputs%]",
           "nports": "[%key:common::config_flow::data::nports%]",
-          "update_all": "[%key:common::config_flow::data::update_all%]"
+          "update_all": "[%key:common::config_flow::data::update_all%]",
+          "poll_outs": "[%key:common::config_flow::data::poll_outs%]"
         }
       }
     },
@@ -38,7 +39,8 @@
           "invert": "[%key:common::config_flow::data::invert%]",
           "mqtt_inputs": "[%key:common::config_flow::data::mqtt_inputs%]",
           "nports": "[%key:common::config_flow::data::nports%]",
-          "update_all": "[%key:common::config_flow::data::update_all%]"
+          "update_all": "[%key:common::config_flow::data::update_all%]",
+          "poll_outs": "[%key:common::config_flow::data::poll_outs%]"
         }
       }
     }

custom_components/mega/translations/en.json

@@ -22,7 +22,8 @@
                     "port_to_scan": "Port to poll aliveness (needed only if no sensors used)",
                     "nports": "Number of ports",
                     "update_all": "Update all outs when input",
-                    "mqtt_inputs": "Use MQTT"
+                    "mqtt_inputs": "Use MQTT",
+                    "poll_outs": "Poll outs"
                 }
             }
         }
@@ -35,7 +36,8 @@
                   "port_to_scan": "Port to poll aliveness (needed only if no sensors used)",
                   "reload": "Reload objects",
                   "mqtt_inputs": "Use MQTT",
-                  "update_all": "Update all outs when input"
+                  "update_all": "Update all outs when input",
+                  "poll_outs": "Poll outs"
                 }
             }
         }

custom_components/mega/translations/ru.json

@@ -21,7 +21,8 @@
                     "port_to_scan": "Порт, который сканируется когда нет датчиков",
                     "mqtt_inputs": "Использовать MQTT",
                     "nports": "Кол-во портов",
-                    "update_all": "Обновить все выходы когда срабатывает вход"
+                    "update_all": "Обновить все выходы когда срабатывает вход",
+                    "poll_outs": "Обновлять выходы (регулярно)"
                 }
             }
         }
@@ -36,7 +37,8 @@
                   "invert": "Список портов (через ,) с инвертированной логикой",
                   "mqtt_inputs": "Использовать MQTT",
                     "nports": "Кол-во портов",
-                    "update_all": "Обновить все выходы когда срабатывает вход"
+                    "update_all": "Обновить все выходы когда срабатывает вход",
+                    "poll_outs": "Обновлять выходы (регулярно)"
                 }
             }
         }

custom_components/mega/translations/uk.json

@@ -21,7 +21,8 @@
                     "port_to_scan": "Порт для сканування при відсутності датчиків",
                     "mqtt_inputs": "Використовувати MQTT",
                     "nports": "Кількість портів",
-                    "update_all": "Оновити всі виходи коли спрацьовує вхід"
+                    "update_all": "Оновити всі виходи коли спрацьовує вхід",
+                    "poll_outs": "Оновити виходи"
                 }
             }
         }
@@ -36,7 +37,8 @@
                   "invert": "Список портів з інвертованою логікою (через ,)",
                   "mqtt_inputs": "Використовувати MQTT",
                   "nports": "Кількість портів",
-                  "update_all": "Оновити всі виходи коли спрацьовує вхід"
+                  "update_all": "Оновити всі виходи коли спрацьовує вхід",
+                    "poll_outs": "Оновити виходи"
                 }
             }
         }

readme.md

@@ -4,8 +4,14 @@
 [![hacs_badge](https://img.shields.io/badge/HACS-Custom-orange.svg)](https://github.com/custom-components/hacs)
 [![Donate](https://img.shields.io/badge/donate-Yandex-red.svg)](https://yoomoney.ru/to/410013955329136)
 
-Интеграция с [MegaD-2561](https://www.ab-log.ru/smart-house/ethernet/megad-2561)
+Интеграция с [MegaD-2561, MegaD-328](https://www.ab-log.ru/smart-house/ethernet/megad-2561)
 
+Если вам понравилась интеграция, не забудьте поставить звезду на гитхабе - вам не сложно, а мне приятно ) А если
+интеграция очень понравилась - еще приятнее, если вы воспользуетесь кнопкой доната )
+
+Обновление прошивки MegaD можно делать прямо из HA с помощью [аддона](https://github.com/andvikt/mega_addon.git)
+
+Подробная документация по [ссылке](https://github.com/andvikt/mega_hacs/wiki)
 ## Основные особенности:
 - Настройка в веб-интерфейсе + yaml
 - Все порты автоматически добавляются как устройства (для обычных релейных выходов создается 
@@ -17,12 +23,20 @@
 - Команды выполняются друг за другом без конкурентного доступа к ресурсам megad, это дает гарантии надежного исполнения
   большого кол-ва команд (например в сценах). Каждая следующая команда отправляется только после получения ответа о
   выполнении предыдущей.
+- поддержка ds2413 (начиная с версии 0.4.1)
 
 ## Установка
 Рекомендованный способ с поддержкой обновлений - [HACS](https://hacs.xyz/docs/installation/installation):
 
 HACS - Integrations - Explore, в поиске ищем MegaD. 
 
+Чтобы включить возможность использования бета-версий, зайдите в HACS, найдите интеграцию MegaD, нажмите три точки, 
+там кнопка "переустановить" или reinstall, дальше нужно нажать галку "показывать бета-версии"
+
+Обновления выполняются так же в меню HACS. 
+Информация об обновлениях приходит с некоторым интервалом, чтобы вручную проверить наличие обновлений
+нажмите три точки возле интеграции в меню HACS и нажмите `обновить информацию`
+
 Альтернативный способ установки:
 ```shell
 # из папки с конфигом
@@ -35,232 +49,7 @@ wget -q -O - https://raw.githubusercontent.com/andvikt/mega_hacs/master/install.
 
 Все имеющиеся у вас порты будут настроены автоматически. Вы можете менять названия, иконки и entity_id так же из интерфейса.
 
-#### Кастомизация устройств с помощью yaml:
-```yaml
-# configuration.yaml
-
-mega:
-  hello: # ID меги, как в UI 
-    7: # номер порта
-      domain: switch # тип устройства (switch или light, по умолчанию для цифровых выходов используется light)
-      invert: true # инвертировать или нет (по умолчанию false)
-      name: Насос # имя устройства
-    8:
-      # исключить из сканирования
-      skip: true
-    33:
-      # для датчиков можно кастомизировать только имя и unit_of_measurement
-      # для температуры и влажность unit определяется автоматически, для остальных юнита нет 
-      name:
-        hum: "влажность"
-        temp: "температура"
-      unit_of_measurement:
-        hum: "%" # если датчиков несколько, то можно указывать юниты по их ключам
-        temp: "°C"
-      # можно так же указать шаблон для конвертации значения, может быть полезно для ацп-входа
-      # текущее значение порта передается в шаблон в переменной "value"
-      conv_template: "{{(value|float)/100}}"
-    14:
-      name: какой-то датчик
-      unit_of_measurement: "°C" # если датчик один, то просто строчкой
-```
-
 ## Зависимости
 Для совместимости c mqtt необходимо настроить интеграцию [mqtt](https://www.home-assistant.io/integrations/mqtt/) 
-в HomeAssistant, а так же обновить ваш контроллер до последней версии, тк были важные обновления в части mqtt
-
-## HTTP in
-Начиная с версии `0.3.1` интеграция стала поддерживать обратную связь без mqtt, используя http-сервер. Для этого в настройках
-интеграции необходимо снять галку с `использовать mqtt`
-
-В самой меге необходимо прописать настройки:
-```yaml
-srv: "192.168.1.4:8123" # ip:port вашего HA
-script: "mega" # это api интеграции, к которому будет обращаться контроллер
-```
-
-Входы будут доступны как binary_sensor, а так же в виде событий `mega.sensor`.
-События можно обрабатывать так:
-```yaml
-- alias: some double click
-  trigger:
-    - platform: event
-      event_type: mega.sensor
-      event_data:
-        pt: 1
-        click: 2
-  action:
-    - service: light.toggle
-      entity_id: light.some_light
-```
-Для binary_sensor имеет смысл использовать режим P&R, для остальных режимов - лучше пользоваться событиями.
-
-Примеры использования binary_sensor:
-```yaml
-- alias: обработка долгих/коротких нажатий
-  trigger:
-    - platform: state
-      entity_id: binary_sensor.some_sensor
-      to: on
-      for: 1 # задержка на секунду
-
-  action:
-    - choose:
-          # если кнопка все еще нажата - значит это долгое нажатие
-        - conditions: "{{ is_state('binary_sensor.some_sensor', 'on')}}"
-          sequence:
-          - service: light.turn_on
-            entity_id: light.some_light
-          # если кнопка уже не нажата - значит это короткое нажатие
-        - conditions: "{{ is_state('binary_sensor.some_sensor', 'off')}}"
-          sequence:
-          - service: light.turn_off
-            entity_id: light.some_light
-```
-
-## Ответ на входящие события от контроллера
-Контроллер ожидает ответ от сервера, который может быть сценарием (по умолчанию интеграция отвечает `d`, что означает 
-запустить то что прописано в поле act в настройках порта).
-
-Поддерживаеются шаблоны HA. Это может быть использовано, например, для запоминания яркости (тк сам контроллер этого не 
-умеет). В шаблоне можно использовать параметры, которые передает контроллер (m, click, pt, mdid, mega_id)
-
-Примеры:
-```yaml
-mega:
-  mega1: # id меги, который вы сами придумываете в конфиге в UI
-    4: # номер порта, с которого ожидаются события
-      response_template: 5:2 # простейший пример без шаблона. Каждый раз когда будет приходить сообщение на этот порт, 
-                             # будем менять состояние на противоположное
-    5:
-      # пример с использованием шаблона, порт 1 будет выключен если он сейчас включен и включен с последней сохраненной 
-      # яркостью если он сейчас выключен     
-      response_template: >-
-        {% if is_state('light.some_port_1', 'on') %}
-        1:0
-        {% else %}
-        1:{{state_attr('light.some_port_1', 'brightness')}}
-        {% endif %}
-    6:
-      # в шаблон так же передаются все параметры, которые передает контроллер (pt, cnt, m, click)
-      # эти параметры можно использовать в условиях или непосредственно в шаблоне в виде {{pt}}
-      response_template: >-
-        {% if m==2 %}1:0{% else %}d{% endif %}
-    
-```
-## Отладка ответов
-Для отладки ответов сервера можно самим имитировать запросы контроллера, если у вас есть доступ к консоли
-HA:
-```shell
-curl -v -X GET 'http://localhost:8123/mega?pt=5&m=1'
-```
-Если доступа нет, нужно в файл конфигурации добавить ip компьюетра, с которого вы хотите делать запросы, например:
-```yaml
-mega:
-  allow_hosts:
-    - 192.168.1.1
-```
-И тогда можно с локальной машины делать запросы на ваш сервер HA:
-```shell
-curl -v -X GET 'http://192.168.88.1.4:8123/mega?pt=5&m=1'
-```
-В ответ будет приходить либо `d`, либо скрипт, который вы настроили
-
-
-## События
-`binary_sensor` срабатывает когда цифровой выход принимает значение 'ON'. `binary_sensor` имеет смысл использовать
-только с режимом входа P&R
-
-При каждом срабатывании `binary_sensor` так же сообщает о событии типа `mega.sensor`.
-События можно использовать в автоматизациях, например так:
-```yaml
-# Пример события с полями как есть прямо из меги
-- alias: some double click
-  trigger:
-    - platform: event
-      event_type: mega.sensor
-      event_data:
-        pt: 1
-        cnt: 2
-  action:
-    - service: light.toggle
-      entity_id: light.some_light
-```
-События могут содержать следующие поля: 
-- mega_id: id как в конфиге HA
-- pt: номер порта
-- cnt: счетчик срабатываний
-- mdid: if как в конфиге контроллера
-- click: клик (подробнее в документации меги)
-- value: текущее значение (только для mqtt)
-- port: номер порта
-
-Начиная с версии 0.3.7 появилось так же событие типа mega.binary:
-```yaml
-# Пример события с полями как есть прямо из меги
-- alias: some long click
-  trigger:
-    - platform: event
-      event_type: mega.binary
-      event_data:
-        entity_id: binary_sensor.some_id
-        type: long
-  action:
-    - service: light.toggle
-      entity_id: light.some_light
-```
-Возможные варианты поля `type`:
-- `long`: долгое нажатие
-- `release`: размыкание (с гарантией что не было долгого нажатия)
-- `long_release`: размыкание после долгого нажатия
-- `press`: замыкание
-- `single`: одинарный клик (в режиме кликов)
-- `double`: двойной клик
-
-Чтобы понять, какие события происходят, лучше всего воспользоваться панелью разработчика и подписаться
-на вкладке события на событие `mega.sensor`, понажимать кнопки.
-
-## Сервисы
-Все сервисы доступны в меню разработчика с описанием и примерами использования
-```yaml
-mega.save:
-  description: Сохраняет текущее состояние портов (?cmd=s)
-  fields:
-    mega_id:
-      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
-      example: "mega"
-
-mega.get_port:
-  description: Запросить текущий статус порта (или всех)
-  fields:
-    mega_id:
-      description: ID меги, можно оставить пустым, тогда будут порты всех зарегистрированных мег
-      example: "mega"
-    port:
-      description: Номер порта (если не заполнять, будут запрошены все порты сразу)
-      example: 1
-
-mega.run_cmd:
-  description: Выполнить любую произвольную команду
-  fields:
-    mega_id:
-      description: ID меги
-      example: "mega"
-    port:
-      description: Номер порта (это не порт, которым мы управляем, а порт с которого шлем команду)
-      example: 1
-    cmd:
-      description: Любая поддерживаемая мегой команда
-      example: "1:0"
-```
-
-## Отладка
-Интеграция находится в активной разработке, при возникновении проблем [заводите issue](https://github.com/andvikt/mega_hacs/issues/new/choose)
-
-Просьба прикладывать детальный лог, который можно включить в конфиге так:
-```yaml
-logger:
-  default: info
-  logs:
-    custom_components.mega: debug
-```
+в HomeAssistant, а так же обновить ваш контроллер до последней версии, обновление прошивки MegaD можно делать прямо 
+из HA с помощью [аддона](https://github.com/andvikt/mega_addon.git)