Fix data_key of the xiaomi_aqara cover for LAN protocol v2 (home-assi…

…stant#22358)
9TurnbullHA · Mar 26, 2019 · 6fa8fdf · 6fa8fdf
1 parent 79445a7
commit 6fa8fdf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/homeassistant/components/xiaomi_aqara/cover.py b/homeassistant/components/xiaomi_aqara/cover.py
@@ -17,8 +17,12 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
         for device in gateway.devices['cover']:
             model = device['model']
             if model == 'curtain':
+                if 'proto' not in device or int(device['proto'][0:1]) == 1:
+                    data_key = 'status'
+                else:
+                    data_key = 'curtain_status'
                 devices.append(XiaomiGenericCover(device, "Curtain",
-                                                  'status', gateway))
+                                                  data_key, gateway))
     add_entities(devices)