enable binary_sensor for smart gardena

py-smart-gardena · Jun 6, 2020 · 11fa83b · 11fa83b
1 parent 9994493
commit 11fa83b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/custom_components/gardena_smart_system/__init__.py b/custom_components/gardena_smart_system/__init__.py
@@ -27,7 +27,7 @@
 
 _LOGGER = logging.getLogger(__name__)
 
-PLATFORMS = ("vacuum", "sensor", "switch")
+PLATFORMS = ("vacuum", "sensor", "switch", "binary_sensor")
 
 
 async def async_setup(hass: HomeAssistant, config: dict):

diff --git a/custom_components/gardena_smart_system/binary_sensor.py b/custom_components/gardena_smart_system/binary_sensor.py
@@ -3,12 +3,13 @@
     BinarySensorEntity,
 )
 
+from custom_components.gardena_smart_system import GARDENA_SYSTEM
 from .const import DOMAIN
 
 
 async def async_setup_entry(hass, config_entry, async_add_entities):
     """Perform the setup for Gardena websocket connection status."""
-    async_add_entities(SmartSystemWebsocketStatus(hass.data[DOMAIN]), True)
+    async_add_entities([SmartSystemWebsocketStatus(hass.data[DOMAIN][GARDENA_SYSTEM])], True)
 
 
 class SmartSystemWebsocketStatus(BinarySensorEntity):