fix assumed state

.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())

custom_components/mega/__init__.py

@@ -24,31 +24,40 @@ 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.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,
+                )
             }
         }
     },

custom_components/mega/config_flow.py

@@ -57,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 = 5
     CONNECTION_CLASS = config_entries.CONN_CLASS_ASSUMED
 
     async def async_step_user(self, user_input=None):

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,16 +284,39 @@ class MegaOutPort(MegaPushEntity):
             cmd = brightness
         else:
             cmd = 1 if not self.invert else 0
-        await self.mega.request(cmd=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
+            self.hass.async_create_task(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.request(cmd=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
+            self.hass.async_create_task(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,15 +10,19 @@ import json
 
 from bs4 import BeautifulSoup
 from homeassistant.components import mqtt
-from homeassistant.const import (DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_PRESSURE, 
-        DEVICE_CLASS_ILLUMINANCE, TEMP_CELSIUS, PERCENTAGE, LIGHT_LUX)
+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, PRESS, LUX, PATT_SPLIT, DOMAIN, CONF_HTTP, EVENT_BINARY_SENSOR, CONF_CUSTOM, CONF_SKIP, \
-    CONF_FORCE_D
+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
+from .exceptions import CannotConnect, NoPort
 from .tools import make_ints
 
 TEMP_PATT = re.compile(r'temp:([01234567890\.]+)')
@@ -44,13 +48,11 @@ CLASSES = {
     LUX: DEVICE_CLASS_ILLUMINANCE
 }
 I2C_DEVICE_TYPES = {
-    "2":  LUX, # BH1750
-    "3":  LUX, # TSL2591
-    "7":  LUX, # MAX44009
-    "70": LUX, # OPT3001
+    "2":  LUX,  # BH1750
+    "3":  LUX,  # TSL2591
+    "7":  LUX,  # MAX44009
+    "70": LUX,  # OPT3001
 }
-class NoPort(Exception):
-    pass
 
 
 class MegaD:
@@ -191,10 +193,18 @@ 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')
+        for x in self.entities:
+            # обновление ds2413 устройств
+            if x.http_cmd == 'ds2413':
+                await self.get_port(
+                    port=x.port,
+                    force_http=True,
+                    http_cmd='list',
+                    conv=False
+                )
         if self.mqtt is None:
             await self.get_all_ports()
             await self.get_sensors(only_list=True)
@@ -255,14 +265,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))
@@ -437,6 +447,18 @@ 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 == "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].append([
+                        {"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'

custom_components/mega/manifest.json

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

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):

readme.md

@@ -4,8 +4,12 @@
 [![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)
 ## Основные особенности:
 - Настройка в веб-интерфейсе + yaml
 - Все порты автоматически добавляются как устройства (для обычных релейных выходов создается 
@@ -17,6 +21,7 @@
 - Команды выполняются друг за другом без конкурентного доступа к ресурсам megad, это дает гарантии надежного исполнения
   большого кол-ва команд (например в сценах). Каждая следующая команда отправляется только после получения ответа о
   выполнении предыдущей.
+- поддержка ds2413 (начиная с версии 0.4.1)
 
 ## Установка
 Рекомендованный способ с поддержкой обновлений - [HACS](https://hacs.xyz/docs/installation/installation):
@@ -48,6 +53,15 @@ mega:
     8:
       # исключить из сканирования
       skip: true
+    10:
+      skip: false # если нужен skip, то он работает только целиком для всего порта
+      # в случае если порт настроен как ds2413  
+      address_a:  #address - это адрес ds2413
+        # здесь внутри работают все стандартные поля конфигурирования порта
+        invert: true
+        domain: switch
+      c6c439000000_b:
+        name: какое-то имя
     33:
       # для датчиков можно кастомизировать только имя и unit_of_measurement
       # для температуры и влажность unit определяется автоматически, для остальных юнита нет 
@@ -110,7 +124,6 @@ mega:
       # эти параметры можно использовать в условиях или непосредственно в шаблоне в виде {{pt}}
       response_template: >-
         {% if m==2 %}1:0{% else %}d{% endif %}
-    
 ```
 
 Начиная с версии v0.3.17 ответ можно слать так же и в режиме MQTT. Аналогично, темплейт должен возвращать готовую команду
@@ -125,6 +138,12 @@ mega:
 **Внимание!** Нельзя использовать чекбокс напротив поля act если планируется использовать ответ сервера - у вас и 
 сработает act и команда от сервера, а вслучае ответа d сработает act два раза.
 
+Так же следует понимать, что это не "ответ" в нормальном понимании - это вызов следом за полученным mqtt-сообщением
+http команды такого вида `http://megaurl/?pt=port&cmd=rendered_template`, где `port` - это номер порта сработавшего входа,
+а `cmd` - текст команды, который получен из темплейта. Те это имитация ответа. У этого подхода есть минус - задержка в 
+исполнении будет значительно выше чем при ответе в режиме http, но тем не менее эта задержка скорее всего не будет 
+сильно заметна.
+
 ## binary_sensor и события
 
 Входы будут доступны как binary_sensor, а так же в виде событий `mega.sensor` и `mega.binary`.