AirSend duo

Author: jouralexandre

README.md

@@ -2,15 +2,16 @@
 
 # AirSend Home Assistant
 
-Component for sending radio commands (433-434Mhz) through the AirSend device.
+Component for sending radio commands through the AirSend (RF433) or AirSend duo (RF433 & RF868).
 
 ## Installation
 
-1. Add `airsend:` to your HA configuration (see configuration below).
-2. Into the terminal, run `wget -q -O - https://raw.githubusercontent.com/devmel/hass_airsend/master/install | bash -`
+1. Into the terminal, run `wget -q -O - https://raw.githubusercontent.com/devmel/hass_airsend/master/install | bash -`
  OR copy the `airsend` folder into your [custom_components folder](https://developers.home-assistant.io/docs/creating_integration_file_structure/#where-home-assistant-looks-for-integrations).
-3. To allow a local LAN connection please install and start [hass_airsend-addon](https://github.com/devmel/hass_airsend-addon).
-4. Restart Home Assistant
+2. To allow a local LAN connection please install and start [hass_airsend-addon](https://github.com/devmel/hass_airsend-addon).
+3. Restart Home Assistant
+4. Add `airsend:` to your HA configuration (see configuration below).
+5. Restart Home Assistant
 
 ## Configuration 
 

custom_components/airsend/__init__.py

@@ -2,17 +2,33 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.typing import ConfigType
 from homeassistant.helpers import discovery
-from homeassistant.const import (
-    CONF_DEVICES,
+from homeassistant.components.hassio import (
+    async_get_addon_info,
 )
+from homeassistant.const import CONF_INTERNAL_URL
 
 DOMAIN = "airsend"
-AS_TYPE = ['switch', 'cover', 'button']
+AS_TYPE = ["switch", "cover", "button"]
 
-async def async_setup(hass : HomeAssistant, config : ConfigType):
+
+async def async_setup(hass: HomeAssistant, config: ConfigType):
     """Set up the AirSend component."""
     if DOMAIN not in config:
         return True
+    internalurl = ""
+    try:
+        internalurl = config[DOMAIN][CONF_INTERNAL_URL]
+    except KeyError:
+        pass
+    if internalurl == "":
+        try:
+            addon_info: dict = await async_get_addon_info(hass, "local_airsend")
+            ip = addon_info["ip_address"]
+            if ip:
+                internalurl = "http://" + str(ip) + ":33863/"
+        except KeyError:
+            pass
+    config[DOMAIN][CONF_INTERNAL_URL] = internalurl
     for plateform in AS_TYPE:
-        discovery.load_platform(hass, plateform, DOMAIN, config[DOMAIN][CONF_DEVICES].copy(), config)
+        discovery.load_platform(hass, plateform, DOMAIN, config[DOMAIN].copy(), config)
     return True

custom_components/airsend/button.py

@@ -1,84 +1,44 @@
 """AirSend buttons."""
-import logging
-import json
 from typing import Any
-from requests import get, post
+
+from .device import Device
+
+from homeassistant.components.button import ButtonEntity
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.typing import ConfigType
-from homeassistant.helpers.entity import async_generate_entity_id
-from homeassistant.components.hassio import (
-    async_get_addon_discovery_info,
-    async_get_addon_info,
-)
-from homeassistant.components.button import (
-    ButtonEntity,
-)
-from . import (
-    DOMAIN,
-)
-
-_LOGGER = logging.getLogger(DOMAIN)
-
-async def async_setup_platform(hass : HomeAssistant, config : ConfigType, async_add_entities, discovery_info=None):
-    addons_url = ""
-    try:
-        addon_info: dict = await async_get_addon_info(hass, 'local_airsend')
-        ip = addon_info["ip_address"]
-        if ip:
-            addons_url = "http://"+str(ip)+":33863/"
-        # _LOGGER.warning("Addon '%s'", addon_info)
-    except:
-        pass
+
+from homeassistant.const import CONF_DEVICES, CONF_INTERNAL_URL
+
+from . import DOMAIN
+
+
+async def async_setup_platform(
+    hass: HomeAssistant, config: ConfigType, async_add_entities, discovery_info=None
+) -> None:
     if discovery_info is None:
         return
-    for name, options in discovery_info.items():
-        if options['type'] == 4096 :
-            id = ""
-            apiKey = ""
-            spurl = ""
-            channel = {}
-            note = {"method":1,"type":0,"value": "TOGGLE"}
-            try:
-                id = options['id']
-            except KeyError:
-                pass
-            try:
-                apiKey = options['apiKey']
-            except KeyError:
-                pass
-            try:
-                spurl = options['spurl']
-            except KeyError:
-                pass
-            try:
-                channel = options['channel']
-            except KeyError:
-                pass
-            try:
-                note = options['note']
-            except KeyError:
-                pass
-            entity = AirSendButton(hass, name, id, options['type'], apiKey, addons_url, spurl, channel, note)
+    for name, options in discovery_info[CONF_DEVICES].items():
+        device = Device(name, options, discovery_info[CONF_INTERNAL_URL])
+        if device.is_button:
+            entity = AirSendButton(
+                hass,
+                device,
+            )
             async_add_entities([entity])
-    return
 
 
 class AirSendButton(ButtonEntity):
     """Representation of an AirSend Button."""
 
-    def __init__(self, hass : HomeAssistant, name : str, id: str, type : int, apikey : str, addons_url : str, spurl : str, channel : dict, note : dict):
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        device: Device,
+    ) -> None:
         """Initialize a button."""
-        self._name = name
-        self._id = id
-        self._type = type
-        self._apikey = apikey
-        self._addons_url = addons_url
-        self._spurl = spurl
-        self._channel = channel
-        self._note = note
-        uname = DOMAIN+name
+        self._device = device
+        uname = DOMAIN + device.name
         self._unique_id = "_".join(x for x in uname)
-        self._state = False
 
     @property
     def unique_id(self):
@@ -97,7 +57,7 @@ def should_poll(self):
     @property
     def name(self):
         """Return the name of the device if any."""
-        return self._name
+        return self._device.name
 
     @property
     def extra_state_attributes(self):
@@ -109,37 +69,8 @@ def assumed_state(self):
         """Return true if unable to access real state of entity."""
         return True
 
-    def press(self) -> None:
+    def press(self, **kwargs: Any) -> None:
         """Handle the button press."""
-        command = "6"
-        note = self._note
-        url = "https://airsend.cloud/device/" + str(self._id) + "/command/" + command + "/"
-        payload = '{"wait": true, "channel":' + json.dumps(self._channel) + ', "thingnotes":{"notes":[' + json.dumps(
-            note) + ']}}'
-        self._action(url, payload, True)
-
-    def _action(self, cloud_url : str, payload : str, new_state : bool):
-        status_code = 404
-        if self._addons_url and self._spurl:
-            headers = {"Authorization": "Bearer "+self._spurl, "content-type": "application/json", "User-Agent": "hass_airsend"}
-            try:
-                response = post(self._addons_url+"airsend/transfer", headers=headers, data=payload, timeout=6)
-                status_code = 500
-                jdata = json.loads(response.text)
-                if jdata["type"] < 0x100:
-                    status_code = response.status_code
-            except:
-                pass
-        if status_code != 200 and self._apikey and cloud_url:
-            headers = {"Authorization": "Bearer "+self._apikey, "content-type": "application/json", "User-Agent": "hass_airsend"}
-            try:
-                response = get(cloud_url, headers=headers, timeout=10)
-                status_code = response.status_code
-            except:
-                pass
-        if status_code == 200:
-            self._state = new_state
+        note = {"method": 1, "type": 0, "value": "TOGGLE"}
+        if self._device.transfer(note):
             self.schedule_update_ha_state()
-        else:
-            _LOGGER.error("action error : "+str(status_code))
-            raise Exception("action error : "+str(status_code))