добавлена возможность восстановления состояний выходов после перезагрузки меги

UA translation update

custom_components/mega/__init__.py

@@ -128,8 +128,7 @@ async def get_hub(hass, entry):
 async def _add_mega(hass: HomeAssistant, entry: ConfigEntry):
     id = entry.data.get('id', entry.entry_id)
     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][id] = hub
     hass.data[DOMAIN][CONF_ALL][id] = hub
     if not await hub.authenticate():
         raise Exception("not authentificated")

custom_components/mega/config_flow.py

@@ -11,7 +11,7 @@ from homeassistant.const import CONF_HOST, CONF_ID, CONF_PASSWORD, CONF_SCAN_INT
 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, CONF_POLL_OUTS, CONF_FAKE_RESPONSE, CONF_FORCE_D, \
-    CONF_ALLOW_HOSTS, CONF_PROTECTED  # pylint:disable=unused-import
+    CONF_ALLOW_HOSTS, CONF_PROTECTED, CONF_RESTORE_ON_RESTART  # pylint:disable=unused-import
 from .hub import MegaD
 from . import exceptions
 
@@ -30,6 +30,7 @@ STEP_USER_DATA_SCHEMA = vol.Schema(
         vol.Optional(CONF_UPDATE_ALL, default=True): bool,
         vol.Optional(CONF_FAKE_RESPONSE, default=True): bool,
         vol.Optional(CONF_FORCE_D, default=True): bool,
+        vol.Optional(CONF_RESTORE_ON_RESTART, default=True): bool,
         vol.Optional(CONF_PROTECTED, default=True): bool,
         vol.Optional(CONF_ALLOW_HOSTS, default='::1;127.0.0.1'): str,
     },
@@ -62,7 +63,7 @@ async def validate_input(hass: core.HomeAssistant, data):
 class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
     """Handle a config flow for mega."""
 
-    VERSION = 10
+    VERSION = 11
     CONNECTION_CLASS = config_entries.CONN_CLASS_ASSUMED
 
     async def async_step_user(self, user_input=None):
@@ -145,6 +146,7 @@ class OptionsFlowHandler(config_entries.OptionsFlow):
                 vol.Optional(CONF_UPDATE_ALL, default=e.get(CONF_UPDATE_ALL, True)): bool,
                 vol.Optional(CONF_FAKE_RESPONSE, default=e.get(CONF_FAKE_RESPONSE, True)): bool,
                 vol.Optional(CONF_FORCE_D, default=e.get(CONF_FORCE_D, False)): bool,
+                vol.Optional(CONF_RESTORE_ON_RESTART, default=e.get(CONF_RESTORE_ON_RESTART, False)): bool,
                 vol.Optional(CONF_PROTECTED, default=e.get(CONF_PROTECTED, True)): bool,
                 vol.Optional(CONF_ALLOW_HOSTS, default='::1;127.0.0.1'): str,
                 # vol.Optional(CONF_INVERT, default=''): str,

custom_components/mega/const.py

@@ -31,6 +31,7 @@ CONF_CONV_TEMPLATE = 'conv_template'
 CONF_POLL_OUTS = 'poll_outs'
 CONF_FORCE_D = 'force_d'
 CONF_DEF_RESPONSE = 'def_response'
+CONF_RESTORE_ON_RESTART = 'restore_on_restart'
 PLATFORMS = [
     "light",
     "switch",

custom_components/mega/http.py

@@ -36,9 +36,10 @@ class MegaView(HomeAssistantView):
             } for mid in cfg if isinstance(cfg[mid], dict)
         }
         _LOGGER.debug('templates: %s', self.templates)
+        self.hubs = {}
 
     async def get(self, request: Request) -> Response:
-
+        _LOGGER.debug('request from %s %s', request.remote, request.headers)
         hass: HomeAssistant = request.app['hass']
         if self.protected:
             auth = False
@@ -63,10 +64,15 @@ class MegaView(HomeAssistantView):
                 _LOGGER.warning(msg)
                 return Response(status=401)
 
-        hub: 'h.MegaD' = hass.data.get(DOMAIN).get(request.remote)  # TODO: проверить какой remote
-        if hub is None and request.remote == '::1':
-            hub = hass.data.get(DOMAIN).get('__def')
-        if hub is None:
+        remote = request.headers.get('X-Real-IP', request.remote)
+        hub: 'h.MegaD' = self.hubs.get(remote)
+        if hub is None and 'mdid' in request.query:
+            hub = self.hubs.get(request.query['mdid'])
+            if hub is None:
+                _LOGGER.warning(f'can not find mdid={request.query["mdid"]} in {list(self.hubs)}')
+        if hub is None and request.remote in ['::1', '127.0.0.1']:
+            hub = self.hubs.get('__def')
+        elif hub is None:
             return Response(status=400)
         data = dict(request.query)
         hass.bus.async_fire(
@@ -75,6 +81,9 @@ class MegaView(HomeAssistantView):
         )
         _LOGGER.debug(f"Request: %s from '%s'", data, request.remote)
         make_ints(data)
+        if data.get('st') == '1' and hub.restore_on_restart:
+            asyncio.create_task(self.later_restore(hub))
+            return Response(status=200)
         port = data.get('pt')
         data = data.copy()
         update_all = True
@@ -96,13 +105,23 @@ class MegaView(HomeAssistantView):
         _LOGGER.debug('response %s', ret)
         Response(body='' if hub.fake_response else ret, content_type='text/plain')
 
-        if hub.fake_response:
+        if hub.fake_response and 'value' not in data and 'pt' in data:
             if 'd' in ret:
                 await hub.request(pt=port, cmd=ret)
             else:
                 await hub.request(cmd=ret)
         return ret
 
+    async def later_restore(self, hub):
+        """
+        Восстановление всех выходов с небольшой задержкой. Задержка нужна чтобы ответ прошел успешно
+
+        :param hub:
+        :return:
+        """
+        await asyncio.sleep(0.2)
+        await hub.restore_states()
+
     async def later_update(self, hub):
         await asyncio.sleep(1)
         _LOGGER.debug('force update')

custom_components/mega/hub.py

@@ -21,7 +21,7 @@ from .const import (
     LUX, PATT_SPLIT, DOMAIN,
     CONF_HTTP, EVENT_BINARY_SENSOR, CONF_CUSTOM, CONF_FORCE_D, CONF_DEF_RESPONSE
 )
-from .entities import set_events_off, BaseMegaEntity
+from .entities import set_events_off, BaseMegaEntity, MegaOutPort
 from .exceptions import CannotConnect, NoPort
 from .tools import make_ints
 
@@ -78,6 +78,7 @@ class MegaD:
             force_d: bool=None,
             allow_hosts: str=None,
             protected=True,
+            restore_on_restart=False,
             **kwargs,
     ):
         """Initialize."""
@@ -85,6 +86,11 @@ class MegaD:
             self.http = hass.data.get(DOMAIN, {}).get(CONF_HTTP)
             if not self.http is None:
                 self.http.allowed_hosts |= {host}
+                self.http.hubs[host] = self
+                if len(self.http.hubs) == 1:
+                    self.http.hubs['__def'] = self
+                if mqtt_id:
+                    self.http.hubs[mqtt_id] = self
         else:
             self.http = None
         self.poll_outs = poll_outs
@@ -131,7 +137,7 @@ class MegaD:
             self.mqtt_id = f"megad/{_id}"
         else:
             self.mqtt_id = mqtt_id
-
+        self.restore_on_restart = restore_on_restart
         if force_d is not None:
             self.customize[CONF_FORCE_D] = force_d
         try:
@@ -475,6 +481,7 @@ class MegaD:
 
     async def get_config(self, nports=37):
         ret = defaultdict(lambda: defaultdict(list))
+        ret['mqtt_id'] = await self.get_mqtt_id()
         async for port, pty, m in self.scan_ports(nports):
             if pty == "0":
                 ret['binary_sensor'][port].append({})
@@ -531,4 +538,12 @@ class MegaD:
                     ))
         return ret
 
+    async def restore_states(self):
+        for x in self.entities:
+            if isinstance(x, MegaOutPort):
+                if x.is_on:
+                    await x.async_turn_on(brightness=x.brightness)
+                else:
+                    await x.async_turn_off()
+
 

custom_components/mega/strings.json

@@ -19,6 +19,7 @@
           "force_d": "[%key:common::config_flow::data::force_d%]",
           "protected": "[%key:common::config_flow::data::protected%]",
           "allow_hosts": "[%key:common::config_flow::data::allow_hosts%]",
+          "restore_on_restart": "[%key:common::config_flow::data::restore_on_restart%]",
           "poll_outs": "[%key:common::config_flow::data::poll_outs%]"
         }
       }

custom_components/mega/tools.py

@@ -1,5 +1,6 @@
 _params = ['m', 'click', 'cnt', 'pt']
 
+
 def make_ints(d: dict):
     for x in _params:
         try:

custom_components/mega/translations/en.json

@@ -27,6 +27,7 @@
                     "force_d": "Force 'd' response",
                     "protected": "Protected",
                     "allow_hosts": "Allowed hosts",
+                    "restore_on_restart": "Restore outs on restart",
                     "poll_outs": "Poll outs"
                 }
             }
@@ -45,6 +46,7 @@
                   "force_d": "Force 'd' response",
                   "protected": "Protected",
                   "allow_hosts": "Allowed hosts",
+                  "restore_on_restart": "Restore outs on restart",
                   "poll_outs": "Poll outs"
                 }
             }

custom_components/mega/translations/ru.json

@@ -25,7 +25,8 @@
                     "fake_response": "Имитация http-ответа",
                     "force_d": "Ответ 'd' по умолчанию",
                     "protected": "Блокировать неразрешенные соединения",
-                     "allow_hosts": "Разрешенные ip (через ;)",
+                    "restore_on_restart": "Восстанавливать выходы при перезагрузке",
+                    "allow_hosts": "Разрешенные ip (через ;)",
                     "poll_outs": "Обновлять выходы (регулярно)"
                 }
             }
@@ -40,13 +41,14 @@
                   "reload": "Обновить объекты",
                   "invert": "Список портов (через ,) с инвертированной логикой",
                   "mqtt_inputs": "Использовать MQTT (Не рекомендуется)",
-                    "fake_response": "Имитация http-ответа",
-                    "force_d": "Ответ 'd' по умолчанию",
-                    "nports": "Кол-во портов",
-                    "update_all": "Обновить все выходы когда срабатывает вход",
+                  "fake_response": "Имитация http-ответа",
+                  "force_d": "Ответ 'd' по умолчанию",
+                  "nports": "Кол-во портов",
+                  "update_all": "Обновить все выходы когда срабатывает вход",
                   "protected": "Блокировать неразрешенные соединения",
                   "allow_hosts": "Разрешенные ip (через ;)",
-                    "poll_outs": "Обновлять выходы (регулярно)"
+                  "restore_on_restart": "Восстанавливать выходы при перезагрузке",
+                  "poll_outs": "Обновлять выходы (регулярно)"
                 }
             }
         }

custom_components/mega/translations/uk.json

@@ -22,10 +22,11 @@
                     "mqtt_inputs": "Використовувати MQTT (Deprecated)",
                     "nports": "Кількість портів",
                     "update_all": "Оновити всі виходи коли спрацьовує вхід",
-                    "fake_response": "Имитация http-ответа",
-                    "force_d": "Ответ 'd' по умолчанию",
-                  "protected": "Блокировать неразрешенные соединения",
-                  "allow_hosts": "Разрешенные ip (через ;)",
+                    "fake_response": "Імітація http-відповіді",
+                    "force_d": "Неявно відповідати 'd'",
+                    "protected": "Блокувати недозволені з'єднання",
+                    "allow_hosts": "Дозволені ip (через ;)",
+                    "restore_on_restart": "Відновлювати виходи при перезавантаженні",
                     "poll_outs": "Оновити виходи"
                 }
             }
@@ -35,17 +36,18 @@
         "step": {
             "init": {
                 "data": {
-                  "scan_interval": "Період оновлення (сек.)",
-                  "port_to_scan": "Порт для сканування при відсутності датчиків",
-                  "reload": "Оновити об'єкти",
-                  "invert": "Список портів з інвертованою логікою (через ,)",
-                  "mqtt_inputs": "Використовувати MQTT (Deprecated)",
-                  "nports": "Кількість портів",
-                    "fake_response": "Имитация http-ответа",
-                    "force_d": "Ответ 'd' по умолчанию",
-                  "update_all": "Оновити всі виходи коли спрацьовує вхід",
-                  "protected": "Блокировать неразрешенные соединения",
-                  "allow_hosts": "Разрешенные ip (через ;)",
+                    "scan_interval": "Період оновлення (сек.)",
+                    "port_to_scan": "Порт для сканування при відсутності датчиків",
+                    "reload": "Оновити об'єкти",
+                    "invert": "Список портів з інвертованою логікою (через ,)",
+                    "mqtt_inputs": "Використовувати MQTT (Deprecated)",
+                    "nports": "Кількість портів",
+                    "fake_response": "Імітація http-відповіді",
+                    "force_d": "Неявно відповідати 'd'",
+                    "update_all": "Оновити всі виходи коли спрацьовує вхід",
+                    "protected": "Блокувати недозволені з'єднання",
+                    "allow_hosts": "Дозволені ip (через ;)",
+                    "restore_on_restart": "Відновлювати виходи при перезавантаженні",
                     "poll_outs": "Оновити виходи"
                 }
             }

readme.md

@@ -58,6 +58,10 @@ wget -q -O - https://raw.githubusercontent.com/andvikt/mega_hacs/master/install.
 srv: "192.168.1.4:8123" # ip:port вашего HA
 script: "mega" # это api интеграции, к которому будет обращаться контроллер
 ```
+Так же необходимо настроить Mega-ID в настройках контроллера, для каждой меги id должен быть разным.
+
+При любых изменениях настроек контроллера (типы входов, id и тд) необходимо в настройках интеграции нажать `Обновить 
+объекты`
 
 ## Зависимости
 Для максимальной скорости реакции на команды сервера, рекомендуется выключить `имитацию http-ответа` в