fix 37 port

.github/workflows/hassfest.yaml

@@ -0,0 +1,14 @@
+name: Validate with hassfest
+
+on:
+  push:
+  pull_request:
+  schedule:
+    - cron: "0 0 * * *"
+
+jobs:
+  validate:
+    runs-on: "ubuntu-latest"
+    steps:
+      - uses: "actions/checkout@v2"
+      - uses: home-assistant/actions/hassfest@master

.github/workflows/validate.yaml

@@ -0,0 +1,17 @@
+name: Validate
+
+on:
+  push:
+  pull_request:
+  schedule:
+    - cron: "0 0 * * *"
+
+jobs:
+  validate:
+    runs-on: "ubuntu-latest"
+    steps:
+      - uses: "actions/checkout@v2"
+      - name: HACS validation
+        uses: "hacs/action@main"
+        with:
+          category: "integration"

custom_components/mega/__init__.py

@@ -11,7 +11,7 @@ from homeassistant.helpers.entity_component import EntityComponent
 from homeassistant.helpers.service import bind_hass
 from homeassistant.components import mqtt
 from homeassistant.config_entries import ConfigEntry
-from .const import DOMAIN, CONF_INVERT
+from .const import DOMAIN, CONF_INVERT, CONF_RELOAD
 from .hub import MegaD
 
 
@@ -51,15 +51,33 @@ async def async_setup(hass: HomeAssistant, config: dict):
     """Set up the mega component."""
     conf = config.get(DOMAIN)
     hass.data[DOMAIN] = {}
+    hass.services.async_register(
+        DOMAIN, 'save', _save_service, schema=vol.Schema({
+            vol.Optional('mega_id'): str
+        })
+    )
+    hass.services.async_register(
+        DOMAIN, 'get_port', _get_port, schema=vol.Schema({
+            vol.Optional('mega_id'): str,
+            vol.Optional('port'): int,
+        })
+    )
+    hass.services.async_register(
+        DOMAIN, 'run_cmd', _run_cmd, schema=vol.Schema({
+            vol.Required('port'): int,
+            vol.Required('cmd'): str,
+            vol.Optional('mega_id'): str,
+        })
+    )
     if conf is None:
         return True
     if CONF_HOST in conf:
         conf = {DEF_ID: conf}
     for id, data in conf.items():
         await _add_mega(hass, id, data)
-    hass.services.async_register(
-        DOMAIN, 'save', _save_service,
-    )
+
+    for id, hub in hass.data[DOMAIN].__items__():
+        _POLL_TASKS[id] = asyncio.create_task(hub.poll())
     return True
 
 
@@ -73,22 +91,24 @@ async def _add_mega(hass: HomeAssistant, id, data: dict):
         raise Exception("not authentificated")
     mid = await hub.get_mqtt_id()
     hub.mqtt_id = mid
-    _POLL_TASKS[id] = asyncio.create_task(hub.poll())
     return hub
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
-    print(entry.entry_id)
     id = entry.data.get('id', entry.entry_id)
-    hub = await _add_mega(hass, id, dict(entry.data))
+    data = dict(entry.data)
+    data.update(entry.options or {})
+    hub = await _add_mega(hass, id, data)
     _hubs[entry.entry_id] = hub
     _subs[entry.entry_id] = entry.add_update_listener(update)
+
     for platform in PLATFORMS:
         hass.async_create_task(
             hass.config_entries.async_forward_entry_setup(
                 entry, platform
             )
         )
+    _POLL_TASKS[id] = asyncio.create_task(hub.poll())
     return True
 
 
@@ -96,9 +116,9 @@ async def update(hass: HomeAssistant, entry: ConfigEntry):
     hub: MegaD = hass.data[DOMAIN][entry.data[CONF_ID]]
     hub.poll_interval = entry.options[CONF_SCAN_INTERVAL]
     hub.port_to_scan = entry.options[CONF_PORT_TO_SCAN]
-    # hub.inverted = map(lambda x: x.strip(), (
-    #     entry.options.get(CONF_INVERT, '').split(',')
-    # )
+    if entry.options[CONF_RELOAD]:
+        await async_remove_entry(hass, entry)
+        await async_setup_entry(hass, entry)
     return True
 
 
@@ -111,16 +131,38 @@ async def async_remove_entry(hass, entry) -> None:
     _hubs.pop(entry.entry_id)
     unsub = _subs.pop(entry.entry_id)
     unsub()
-    return True
 
 
 @bind_hass
-async def _save_service(hass: HomeAssistant, mega_id='def'):
-    hub: MegaD = hass.data[DOMAIN][mega_id]
-    await hub.save()
+async def _save_service(hass: HomeAssistant, mega_id=None):
+    if mega_id:
+        hub: MegaD = hass.data[DOMAIN][mega_id]
+        await hub.save()
+    else:
+        for hub in hass.data[DOMAIN].values():
+            await hub.save()
 
 
-async def _is_alive(cond: asyncio.Condition, msg):
-    async with cond:
-        cond.notify_all()
+@bind_hass
+async def _get_port(hass: HomeAssistant, port=None, mega_id=None):
+    if mega_id:
+        hub: MegaD = hass.data[DOMAIN][mega_id]
+        if port is None:
+            await hub.get_all_ports()
+        else:
+            await hub.get_port(port)
+    else:
+        for hub in hass.data[DOMAIN].values():
+            if port is None:
+                await hub.get_all_ports()
+            else:
+                await hub.get_port(port)
 
+@bind_hass
+async def _run_cmd(hass: HomeAssistant, port: int, cmd: str, mega_id=None):
+    if mega_id:
+        hub: MegaD = hass.data[DOMAIN][mega_id]
+        await hub.send_command(port=port, cmd=cmd)
+    else:
+        for hub in hass.data[DOMAIN].values():
+            await hub.send_command(port=port, cmd=cmd)

custom_components/mega/config_flow.py

@@ -104,8 +104,7 @@ class OptionsFlowHandler(config_entries.OptionsFlow):
                 vol.Optional(CONF_SCAN_INTERVAL, default=e[CONF_SCAN_INTERVAL]): int,
                 vol.Optional(CONF_PORT_TO_SCAN, default=e.get(CONF_PORT_TO_SCAN, 0)): int,
                 vol.Optional(CONF_RELOAD, default=False): bool,
-                vol.Optional(CONF_INVERT): vol.Set(),
+                # vol.Optional(CONF_INVERT, default=''): str,
             }),
         )
-        print(ret)
         return ret

custom_components/mega/entities.py

@@ -25,11 +25,12 @@ class BaseMegaEntity(RestoreEntity):
     ):
         self._state: State = None
         self.port = port
-        self._name = name
         self._mega_id = mega_id
         self._lg = None
         self._unique_id = unique_id or f"mega_{mega_id}_{port}" + \
                                        (f"_{id_suffix}" if id_suffix else "")
+        self._name = name or f"{mega_id}_{port}" + \
+                            (f"_{id_suffix}" if id_suffix else "")
 
     @property
     def lg(self) -> logging.Logger:

custom_components/mega/hub.py

@@ -1,6 +1,7 @@
 import asyncio
 import json
 import logging
+from datetime import datetime
 from functools import wraps
 
 import aiohttp
@@ -22,12 +23,12 @@ class MegaD:
             host: str,
             password: str,
             mqtt: mqtt.MQTT,
-            lg:logging.Logger,
+            lg: logging.Logger,
             id: str,
             mqtt_id: str = None,
             scan_interval=60,
             port_to_scan=0,
-            inverted:typing.List[int] = None,
+            inverted: typing.List[int] = None,
             **kwargs,
     ):
         """Initialize."""
@@ -47,6 +48,7 @@ class MegaD:
         self.sensors = []
         self.port_to_scan = port_to_scan
         self.inverted = inverted or []
+        self.last_update = datetime.now()
         if not mqtt_id:
             _id = host.split(".")[-1]
             self.mqtt_id = f"megad/{_id}"
@@ -59,10 +61,11 @@ class MegaD:
             self.entities.append(ent)
 
     async def get_sensors(self):
+        self.lg.debug(self.sensors)
         _ports = {x.port for x in self.sensors}
         for x in _ports:
             await self.get_port(x)
-            await asyncio.sleep(self.poll_interval)
+            await asyncio.sleep(0.1)
 
     async def poll(self):
         """
@@ -70,36 +73,37 @@ class MegaD:
         offline
         """
         self._loop = asyncio.get_event_loop()
-        if self.sensors:
-            await self.subscribe(self.sensors[0].port, callback=self._notify)
-        else:
-            await self.subscribe(self.port_to_scan, callback=self._notify)
-        while True:
-            async with self.is_alive:
-                if len(self.sensors) > 0:
-                    await self.get_sensors()
-                else:
-                    await self.get_port(self.port_to_scan)
 
-                try:
-                    await asyncio.wait_for(self.is_alive.wait(), timeout=5)
-                    self.hass.states.async_set(
-                        f'mega.{self.id}',
-                        'online',
-                    )
-                    self.online = True
-                except asyncio.TimeoutError:
-                    self.online = False
+        while True:
+            if len(self.sensors) > 0:
+                await self.get_sensors()
+            else:
+                await self.get_port(self.port_to_scan)
+
+            await asyncio.sleep(1)
+            if (datetime.now() - self.last_update).total_seconds() > self.poll_interval:
+                await self.get_port(self.port_to_scan)
+                await asyncio.sleep(1)
+                if (datetime.now() - self.last_update).total_seconds() > self.poll_interval:
+                    self.lg.warning('mega is offline')
                     self.hass.states.async_set(
                         f'mega.{self.id}',
                         'offline',
                     )
-                for x in self.entities:
-                    try:
-                        await x.async_update_ha_state()
-                    except RuntimeError:
-                        pass
-            await asyncio.sleep(self.poll_interval)
+                    self.online = False
+            else:
+                self.hass.states.async_set(
+                    f'mega.{self.id}',
+                    'online',
+                )
+                self.online = True
+
+            for x in self.entities:
+                try:
+                    await x.async_update_ha_state()
+                except RuntimeError:
+                    pass
+            await asyncio.sleep(self.poll_interval - 1)
 
     async def _async_notify(self):
         async with self.is_alive:
@@ -174,7 +178,7 @@ class MegaD:
 
     async def get_all_ports(self):
         for x in range(37):
-            await self.get_port(x)
+            asyncio.create_task(self.get_port(x))
 
     async def reboot(self, save=True):
         await self.save()
@@ -187,6 +191,7 @@ class MegaD:
             self.lg.debug(
                 'process incomming message: %s', msg
             )
+            self.last_update = datetime.now()
             return callback(msg)
 
         self.lg.debug(
@@ -254,7 +259,7 @@ class MegaD:
             return pty, m
 
     async def scan_ports(self,):
-        for x in range(37):
+        for x in range(38):
             ret = await self.scan_port(x)
             if ret:
                 yield [x, *ret]

custom_components/mega/light.py

@@ -119,7 +119,7 @@ class MegaLight(LightEntity, BaseMegaEntity):
         return self._state == 'ON'
 
     async def async_turn_on(self, brightness=None, **kwargs) -> None:
-        brightness = brightness or self.brightness
+        brightness = brightness or self.brightness or 255
         if self.dimmer and brightness == 0:
             cmd = 255
         elif self.dimmer:

custom_components/mega/manifest.json

@@ -2,7 +2,7 @@
   "domain": "mega",
   "name": "mega",
   "config_flow": true,
-  "documentation": "https://www.home-assistant.io/integrations/mega",
+  "documentation": "https://www.home-assistant.io/integrations/mega_hacs",
   "requirements": [
     "beautifulsoup4",
     "lxml"
@@ -15,5 +15,6 @@
   ],
   "codeowners": [
     "@andvikt"
-  ]
+  ],
+  "issue_tracker": "https://github.com/andvikt/mega_hacs/issues"
 }

custom_components/mega/sensor.py

@@ -107,6 +107,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, asyn
                         CONF_KEY: key,
                     })
                 devices.append(sensor)
+                hub.sensors.append(sensor)
 
     async_add_devices(devices)
 
@@ -135,11 +136,6 @@ class Mega1WSensor(BaseMegaEntity):
         self._device_class = device_class
         self._unit_of_measurement = unit_of_measurement
 
-    async def async_added_to_hass(self) -> None:
-
-        await super(Mega1WSensor, self).async_added_to_hass()
-        self.mega.sensors.append(self)
-
     @property
     def unit_of_measurement(self):
         return self._unit_of_measurement

custom_components/mega/services.yaml

@@ -1,13 +1,33 @@
 
 
 save:
-  # Description of the service
   description: Сохраняет текущее состояние портов (?cmd=s)
-  # Different fields that your service accepts
   fields:
-    # Key of the field
+
     mega_id:
-      # Description of the field
-      description: ID меги
-      # Example value that can be passed for this field
-      example: "def"
+      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
+      example: "mega"
+
+get_port:
+  description: Запросить текущий статус порта (или всех)
+  fields:
+    mega_id:
+      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
+      example: "mega"
+    port:
+      description: Номер порта (если не заполнять, будут запрошены все порты сразу)
+      example: 1
+
+run_cmd:
+  description: Выполнить любую произвольную команду
+  fields:
+
+    mega_id:
+      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
+      example: "mega"
+    port:
+      description: Номер порта (это не порт, которым мы управляем, а порт с которого шлем команду)
+      example: 1
+    cmd:
+      description: Любая поддерживаемая мегой команда
+      example: "1:0"

install.sh

@@ -0,0 +1,5 @@
+#!/bin/bash
+git clone https://github.com/andvikt/mega_hacs.git
+mkdir custom_components
+cp mega_hacs/custom_components/mega custom_components/mega
+rm -fR mega_hacs

readme.md

@@ -13,15 +13,18 @@
 ## Устройства
 Поддерживаются устройства: light, switch, binary_sensor, sensor. light может работать как диммер
 ## Установка
-В папке config/custom_components выполнить:
-  ```shell
-  git clone https://github.com/andvikt/mega.git
-  ```
-  Обновление:
-  ```shell
-  git pull
-  ```
+Рекомендованнй способ - через [HACS](https://hacs.xyz/docs/installation/installation).
+После установки HACS, нужно перейти в меню hacs -> integrations, далее в верхнем правом углу
+нажать три точки, где будет `Custom repositories`, открыть, нажать add и добавить `https://github.com/andvikt/mega_hacs.git`
+
+Альтернативный способ установки:
+```shell
+# из папки с конфигом
+wget -q -O - https://raw.githubusercontent.com/andvikt/mega_hacs/master/install.sh | bash -
+```
 Перезагрузить HA
+
+Для обновления повторить
 ## Зависимости
 Перед использованием необходимо настроить интеграцию mqtt в HomeAssistant
 
@@ -78,12 +81,39 @@ switch:
 ```
 
 ## Сервисы
-Интеграция предоставляет сервис сохранения состояния портов: `mega.save`
 ```yaml
-action:
-  service: mega.save
-  data:
-    mega_id: def
+save:
+  description: Сохраняет текущее состояние портов (?cmd=s)
+  fields:
+
+    mega_id:
+      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
+      example: "mega"
+
+get_port:
+  description: Запросить текущий статус порта (или всех)
+  fields:
+    mega_id:
+      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
+      example: "mega"
+    port:
+      description: Номер порта (если не заполнять, будут запрошены все порты сразу)
+      example: 1
+
+run_cmd:
+  description: Выполнить любую произвольную команду
+  fields:
+
+    mega_id:
+      description: ID меги, можно оставить пустым, тогда будут сохранены все зарегистрированные меги
+      example: "mega"
+    port:
+      description: Номер порта (это не порт, которым мы управляем, а порт с которого шлем команду)
+      example: 1
+    cmd:
+      description: Любая поддерживаемая мегой команда
+      example: "1:0"
+
 ```
 
 ## Состояния