add 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

@@ -51,15 +51,31 @@ 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
@@ -79,7 +95,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 {})
@@ -119,7 +134,35 @@ async def async_remove_entry(hass, entry) -> None:
 
 
 @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()
 
+
+@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/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()

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/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"
+
 ```
 
 ## Состояния