From fe2adff017f85d5de097a53892d4c0e81de18204 Mon Sep 17 00:00:00 2001 From: pavoni Date: Mon, 14 Mar 2016 10:29:12 +0000 Subject: [PATCH] Handle startup race condition. --- homeassistant/components/binary_sensor/wemo.py | 3 +++ homeassistant/components/switch/wemo.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/homeassistant/components/binary_sensor/wemo.py b/homeassistant/components/binary_sensor/wemo.py index f3cd91baa04..0e3259a3a96 100644 --- a/homeassistant/components/binary_sensor/wemo.py +++ b/homeassistant/components/binary_sensor/wemo.py @@ -45,6 +45,9 @@ class WemoBinarySensor(BinarySensorDevice): _LOGGER.info( 'Subscription update for %s', _device) + if not hasattr(self, 'hass'): + self.update() + return self.update_ha_state(True) @property diff --git a/homeassistant/components/switch/wemo.py b/homeassistant/components/switch/wemo.py index 90d0f46ba68..63b2665449e 100644 --- a/homeassistant/components/switch/wemo.py +++ b/homeassistant/components/switch/wemo.py @@ -63,6 +63,9 @@ class WemoSwitch(SwitchDevice): _LOGGER.info( 'Subscription update for %s', _device) + if not hasattr(self, 'hass'): + self.update() + return self.update_ha_state(True) @property