RF single button names

AlexxIT · Feb 20, 2020 · b6b75aa · b6b75aa
1 parent da6bfb9
commit b6b75aa
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/sonoff/remote.py b/custom_components/sonoff/remote.py
@@ -31,8 +31,13 @@ def __init__(self, device: EWeLinkDevice):
         # init button names
         self._buttons = {}
         for remote in self.device.config.get('tags', {}).get('zyx_info', []):
-            for button in remote.get('buttonName'):
-                self._buttons.update(button)
+            buttons = remote['buttonName']
+            if len(buttons) > 1:
+                for button in buttons:
+                    self._buttons.update(button)
+            else:
+                k = next(iter(buttons[0]))
+                self._buttons.update({k: remote['name']})
 
     async def async_added_to_hass(self) -> None:
         # Присваиваем имя устройства только на этом этапе, чтоб в `entity_id`