fix mqtt

custom_components/mega/entities.py

@@ -155,10 +155,12 @@ class MegaOutPort(MegaPushEntity):
 
     @property
     def is_on(self) -> bool:
-        val = self.mega.values.get(self.port, {}).get("value")
+        val = self.mega.values.get(self.port, {})
+
         if val is None and self._state is not None:
             return self._state == 'ON'
         elif val is not None:
+            val = val.get("value")
             if not self.invert:
                 return val == 'ON' or str(val) == '1' or (safe_int(val) is not None and safe_int(val) > 0)
             else:
@@ -186,10 +188,11 @@ class MegaOutPort(MegaPushEntity):
         self.mega.values[self.port] = {'value': cmd}
         await self.get_state()
 
+
 def safe_int(v):
     if v in ['ON', 'OFF']:
         return None
     try:
         return int(v)
-    except ValueError:
+    except (ValueError, TypeError):
         return None

custom_components/mega/http.py

@@ -8,10 +8,10 @@ from aiohttp.web_request import Request
 from aiohttp.web_response import Response
 
 from homeassistant.helpers.template import Template
-from .const import EVENT_BINARY_SENSOR, CONF_HTTP, DOMAIN, CONF_CUSTOM, CONF_RESPONSE_TEMPLATE
+from .const import EVENT_BINARY_SENSOR, DOMAIN, CONF_RESPONSE_TEMPLATE
 from homeassistant.components.http import HomeAssistantView
-from homeassistant.core import callback, HomeAssistant
+from homeassistant.core import HomeAssistant
-from . import hub
+from .tools import make_ints
 
 _LOGGER = logging.getLogger(__name__).getChild('http')
 
@@ -26,8 +26,7 @@ class MegaView(HomeAssistantView):
     def __init__(self, cfg: dict):
         self._try = 0
         self.allowed_hosts = {'::1'}
-        self.callbacks: typing.DefaultDict[int, typing.List[typing.Callable[[dict], typing.Coroutine]]] \
+        self.callbacks = defaultdict(lambda: defaultdict(list))
-            = defaultdict(list)
         self.templates: typing.Dict[str, typing.Dict[str, Template]] = {
             mid: {
                 pt: cfg[mid][pt][CONF_RESPONSE_TEMPLATE]
@@ -62,9 +61,10 @@ class MegaView(HomeAssistantView):
         make_ints(data)
         port = data.get('pt')
         data = data.copy()
+        data['mega_id'] = hub.id
         ret = 'd'
         if port is not None:
-            for cb in self.callbacks[port]:
+            for cb in self.callbacks[hub.id][port]:
                 cb(data)
             template: Template = self.templates.get(hub.id, {}).get(port)
             if hub.update_all:
@@ -73,8 +73,7 @@ class MegaView(HomeAssistantView):
                 template.hass = hass
                 ret = template.async_render(data)
         _LOGGER.debug('response %s', ret)
-        ret = Response(body=ret or 'd', content_type='text/plain', headers={})
+        ret = Response(body=ret or 'd', content_type='text/plain', headers={'Server': 's', 'Date': 'n'})
-        ret.headers.clear()
         return ret
 
     async def later_update(self, hub):
@@ -83,13 +82,3 @@ class MegaView(HomeAssistantView):
         await hub.updater.async_refresh()
 
 
-def make_ints(d: dict):
-    for x in d:
-        try:
-            d[x] = float(d[x])
-        except ValueError:
-            pass
-    if 'm' not in d:
-        d['m'] = 0
-    if 'click' not in d:
-        d['click'] = 0

custom_components/mega/hub.py

@@ -14,9 +14,10 @@ from homeassistant.const import DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_HUMIDITY
 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
+from .const import TEMP, HUM, PATT_SPLIT, DOMAIN, CONF_HTTP, EVENT_BINARY_SENSOR
 from .exceptions import CannotConnect, MqttNotConfigured
 from .http import MegaView
+from .tools import make_ints
 
 TEMP_PATT = re.compile(r'temp:([01234567890\.]+)')
 HUM_PATT = re.compile(r'hum:([01234567890\.]+)')
@@ -290,9 +291,18 @@ class MegaD:
         value = None
         try:
             value = json.loads(msg.payload)
+            if isinstance(value, dict):
+                make_ints(value)
             self.values[port] = value
             for cb in self._callbacks[port]:
                 cb(value)
+            if isinstance(value, dict):
+                value = value.copy()
+                value['mega_id'] = self.id
+                self.hass.bus.async_fire(
+                    EVENT_BINARY_SENSOR,
+                    value,
+                )
         except Exception as exc:
             self.lg.warning(f'could not parse json ({msg.payload}): {exc}')
             return
@@ -307,7 +317,7 @@ class MegaD:
         if self.mqtt_inputs:
             self._callbacks[port].append(callback)
         else:
-            self.http.callbacks[port].append(callback)
+            self.http.callbacks[self.id][port].append(callback)
 
     async def authenticate(self) -> bool:
         """Test if we can authenticate with the host."""

custom_components/mega/tools.py

@@ -0,0 +1,10 @@
+def make_ints(d: dict):
+    for x in d:
+        try:
+            d[x] = float(d[x])
+        except (ValueError, TypeError):
+            pass
+    if 'm' not in d:
+        d['m'] = 0
+    if 'click' not in d:
+        d['click'] = 0

readme.md

@@ -81,18 +81,42 @@ script: "mega" # это api интеграции, к которому будет
       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, value)
+умеет). В шаблоне можно использовать параметры, которые передает контроллер (m, click, pt, mdid, mega_id)
 
 Примеры:
 ```yaml
@@ -154,6 +178,15 @@ curl -v -X GET 'http://192.168.88.1.4:8123/mega?pt=5&m=1'
     - service: light.toggle
       entity_id: light.some_light
 ```
+События могут содержать следующие поля: 
+- mega_id: id как в конфиге HA
+- pt: номер порта
+- cnt: счетчик срабатываний
+- mdid: if как в конфиге контроллера
+- click: клик (подробнее в документации меги)
+- value: текущее значение (только для mqtt)
+- port: номер порта
+
 Чтобы понять, какие события происходят, лучше всего воспользоваться панелью разработчика и подписаться
 на вкладке события на событие `mega.sensor`, понажимать кнопки.