fix services

.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

@@ -1,13 +1,11 @@
 """The mega integration."""
 import asyncio
 import logging
-import typing
 from functools import partial
 
 import voluptuous as vol
-from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_PLATFORM, CONF_SCAN_INTERVAL, CONF_ID
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity_component import EntityComponent
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_SCAN_INTERVAL, CONF_ID
+from homeassistant.core import HomeAssistant, ServiceCall
 from homeassistant.helpers.service import bind_hass
 from homeassistant.components import mqtt
 from homeassistant.config_entries import ConfigEntry
@@ -51,16 +49,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', partial(_save_service, hass), schema=vol.Schema({
+            vol.Optional('mega_id'): str
+        })
+    )
+    hass.services.async_register(
+        DOMAIN, 'get_port', partial(_get_port, hass), schema=vol.Schema({
+            vol.Optional('mega_id'): str,
+            vol.Optional('port'): int,
+        })
+    )
+    hass.services.async_register(
+        DOMAIN, 'run_cmd', partial(_run_cmd, hass), 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():
+        _LOGGER.warning('YAML configuration is deprecated, please use web-interface')
         await _add_mega(hass, id, data)
-    hass.services.async_register(
-        DOMAIN, 'save', _save_service,
-    )
-    for id, hub in hass.data[DOMAIN].__items__():
+
+    for id, hub in hass.data[DOMAIN].items():
         _POLL_TASKS[id] = asyncio.create_task(hub.poll())
     return True
 
@@ -79,7 +94,6 @@ async def _add_mega(hass: HomeAssistant, id, data: dict):
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
-    print(entry.entry_id)
     id = entry.data.get('id', entry.entry_id)
     data = dict(entry.data)
     data.update(entry.options or {})
@@ -118,8 +132,42 @@ async def async_remove_entry(hass, entry) -> None:
     unsub()
 
 
-@bind_hass
-async def _save_service(hass: HomeAssistant, mega_id='def'):
+async def _save_service(hass: HomeAssistant, call: ServiceCall):
+    mega_id = call.data['mega_id']
+    if mega_id:
         hub: MegaD = hass.data[DOMAIN][mega_id]
         await hub.save()
+    else:
+        for hub in hass.data[DOMAIN].values():
+            await hub.save()
 
+
+@bind_hass
+async def _get_port(hass: HomeAssistant, call: ServiceCall):
+    port = call.data['port']
+    mega_id = call.data['mega_id']
+    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, call: ServiceCall):
+    port = call.data['port']
+    mega_id = call.data['mega_id']
+    cmd = call.data['cmd']
+    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/binary_sensor.py

@@ -1,6 +1,4 @@
 """Platform for light integration."""
-import asyncio
-import json
 import logging
 
 import voluptuous as vol
@@ -18,7 +16,6 @@ from homeassistant.const import (
     CONF_ID
 )
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.restore_state import RestoreEntity
 from .entities import BaseMegaEntity
 
 from .hub import MegaD

custom_components/mega/config_flow.py

@@ -9,7 +9,7 @@ from homeassistant.components import mqtt
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_HOST, CONF_ID, CONF_PASSWORD, CONF_SCAN_INTERVAL
 from homeassistant.core import callback
-from .const import DOMAIN, CONF_PORT_TO_SCAN, CONF_RELOAD, CONF_INVERT  # pylint:disable=unused-import
+from .const import DOMAIN, CONF_PORT_TO_SCAN, CONF_RELOAD  # pylint:disable=unused-import
 from .hub import MegaD
 from . import exceptions
 

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

@@ -178,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()
@@ -259,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

@@ -1,6 +1,4 @@
 """Platform for light integration."""
-import asyncio
-import json
 import logging
 
 import voluptuous as vol
@@ -19,7 +17,6 @@ from homeassistant.const import (
     CONF_ID
 )
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.restore_state import RestoreEntity
 from .entities import BaseMegaEntity
 
 from .hub import MegaD
@@ -119,7 +116,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

@@ -1,7 +1,5 @@
 """Platform for light integration."""
 import logging
-
-import typing
 import voluptuous as vol
 
 from homeassistant.components.sensor import (

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"

custom_components/mega/switch.py

@@ -1,5 +1,4 @@
 """Platform for light integration."""
-import json
 import logging
 
 import voluptuous as vol
@@ -13,11 +12,7 @@ from homeassistant.const import (
     CONF_PLATFORM,
     CONF_PORT,
 )
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.restore_state import RestoreEntity
 from .entities import BaseMegaEntity
-
-from .hub import MegaD
 from .const import CONF_DIMMER, CONF_SWITCH
 
 _LOGGER = logging.getLogger(__name__)

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

@@ -3,92 +3,72 @@
 Интеграция с [MegaD-2561](https://www.ab-log.ru/smart-house/ethernet/megad-2561)
 
 ## Основные особенности:
-- Настройка как из yaml так и из веб-интерфейса
-- При настройки из веба все порты автоматически добавляются как устройства (для обычных релейных выходов создается 
+- Все порты автоматически добавляются как устройства (для обычных релейных выходов создается 
   `light`, для шим - `light` с поддержкой яркости, для цифровых входов `binary_sensor`, для температурных датчиков
   `sensor`)
 - Возможность работы с несколькими megad
 - Обратная связь по mqtt
 - Команды выполняются друг за другом без конкурентного доступа к ресурсам megad
+- Поддержка температурных датчиков в режиме шины
+
 ## Устройства
 Поддерживаются устройства: 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
 
 ## Настройка из веб-интерфейса
 `Настройки` -> `Интеграции` -> `Добавить интеграцию` в поиске ищем mega
 
-## Пример настройки с помощью yaml:
-```yaml
-mega: 
-  mega1:
-    host: 192.168.0.14
-    name: hello
-    password: sec
-    mqtt_id: mega # это id в конфиге меги
-
-light:
-  - platform: mega
-    mega1:
-      switch:
-        - 1 # можно просто перечислить порты
-        - 2
-        - 3
-      dimmer:
-        - port: 7
-          name: hello # можно использовать расширенный вариант с названиями
-        - 9
-        - 10
-
-binary_sensor:
-  - platform: mega
-    mega1:
-      - port: 16
-        name: sensor1
-      - port: 18
-        name: sensor2
-
-sensor:
-  - platform: mega
-    mega1:
-      - port: 10
-        name: some temp
-        type: w1
-        key: temp
-      - port: 10
-        name: some hum
-        type: w1
-        key: hum
-
-switch:
-  - platform: mega
-    mega1:
-      - 11
-
-```
-
 ## Сервисы
-Интеграция предоставляет сервис сохранения состояния портов: `mega.save`
 ```yaml
-action:
-  service: mega.save
-  data:
-    mega_id: def
-```
+save:
+  description: Сохраняет текущее состояние портов (?cmd=s)
+  fields:
 
-## Состояния
-Так же каждое устройство megad опрашивается на предмет работоспособности, текущий статус
-хранится в mega.<id>
+    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"
+
+```
 
 ## Отладка
 Если возникают проблемы, можно включить детальный лог, для этого в конфиг добавить: