Inverted rflink cover

homeassistant/components/rflink/cover.py

@@ -4,7 +4,7 @@
 import voluptuous as vol
 
 from homeassistant.components.cover import PLATFORM_SCHEMA, CoverDevice
-from homeassistant.const import CONF_NAME, STATE_OPEN
+from homeassistant.const import CONF_NAME, CONF_TYPE, STATE_OPEN
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.restore_state import RestoreEntity
 
@@ -23,6 +23,8 @@
 
 _LOGGER = logging.getLogger(__name__)
 
+TYPE_STANDARD = "standard"
+TYPE_INVERTED = "inverted"
 
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
     {
@@ -33,6 +35,7 @@
             {
                 cv.string: {
                     vol.Optional(CONF_NAME): cv.string,
+                    vol.Optional(CONF_TYPE): vol.Any(TYPE_STANDARD, TYPE_INVERTED),
                     vol.Optional(CONF_ALIASES, default=[]): vol.All(
                         cv.ensure_list, [cv.string]
                     ),
@@ -52,12 +55,51 @@
 )
 
 
+def entity_type_for_device_id(device_id):
+    """Return entity class for protocol of a given device_id.
+
+    Async friendly.
+    """
+    entity_type_mapping = {
+        # KlikAanKlikUit cover have the controls inverted
+        "newkaku": TYPE_INVERTED
+    }
+    protocol = device_id.split("_")[0]
+    return entity_type_mapping.get(protocol, TYPE_STANDARD)
+
+
+def entity_class_for_type(entity_type):
+    """Translate entity type to entity class.
+
+    Async friendly.
+    """
+    entity_device_mapping = {
+        # default cover implementation
+        TYPE_STANDARD: RflinkCover,
+        # cover with open/close commands inverted
+        # like KAKU/COCO ASUN-650
+        TYPE_INVERTED: InvertedRflinkCover,
+    }
+
+    return entity_device_mapping.get(entity_type, RflinkCover)
+
+
 def devices_from_config(domain_config):
     """Parse configuration and add Rflink cover devices."""
     devices = []
     for device_id, config in domain_config[CONF_DEVICES].items():
+        # Determine what kind of entity to create, RflinkCover
+        # or InvertedRflinkCover
+        if CONF_TYPE in config:
+            # Remove type from config to not pass it as and argument
+            # to entity instantiation
+            entity_type = config.pop(CONF_TYPE)
+        else:
+            entity_type = entity_type_for_device_id(device_id)
+
+        entity_class = entity_class_for_type(entity_type)
         device_config = dict(domain_config[CONF_DEVICE_DEFAULTS], **config)
-        device = RflinkCover(device_id, **device_config)
+        device = entity_class(device_id, **device_config)
         devices.append(device)
 
     return devices
@@ -115,3 +157,13 @@ def async_open_cover(self, **kwargs):
     def async_stop_cover(self, **kwargs):
         """Turn the device stop."""
         return self._async_handle_command("stop_cover")
+
+
+class InvertedRflinkCover(RflinkCover):
+    """Rflink cover that has inverted open/close commands."""
+
+    async def _async_send_command(self, cmd, repetitions):
+        """Will invert only the UP/DOWN commands."""
+        _LOGGER.debug("Getting command: %s for Rflink device: %s", cmd, self._device_id)
+        cmd_inv = {"UP": "DOWN", "DOWN": "UP"}
+        await super()._async_send_command(cmd_inv.get(cmd, cmd), repetitions)