Fix failing update_info

Drakulix · Oct 10, 2020 · 1e49671 · 1e49671
1 parent 476e078
commit 1e49671
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 22 deletions.
diff --git a/custom_components/googlehome/__init__.py b/custom_components/googlehome/__init__.py
@@ -91,7 +91,8 @@ async def update_info(self, host):
                 _LOGGER.debug(device_info_data)
                 self._connected = bool(device_info_data)
                 self.hass.data[DOMAIN][host]["info"] = device_info_data
-                return
+                return True
+        return False
 
     async def update_bluetooth(self, host, entry: config_entries.ConfigEntry):
         """Update bluetooth from Google Home."""

diff --git a/custom_components/googlehome/device_tracker.py b/custom_components/googlehome/device_tracker.py
@@ -26,13 +26,13 @@ async def async_setup_entry(hass, config_entry, async_see):
     async def async_cast_discovered(discover: ChromecastInfo):
         hass.data[DOMAIN].setdefault(discover.host, {})
 
-        await hass.data[CLIENT].update_info(discover.host)
-        info = hass.data[DOMAIN][discover.host].get("info", { "device_info": {} })
-        if info["device_info"]["capabilities"].get("bluetooth_supported", False):
-            scanner = GoogleHomeDeviceScanner(
-                hass, hass.data[CLIENT], config_entry, discover, async_see
-            )
-            return await scanner.async_init()
+        if await hass.data[CLIENT].update_info(discover.host):
+            info = hass.data[DOMAIN][discover.host].get("info", { "device_info": {} })
+            if info["device_info"]["capabilities"].get("bluetooth_supported", False):
+                scanner = GoogleHomeDeviceScanner(
+                    hass, hass.data[CLIENT], config_entry, discover, async_see
+                )
+                return await scanner.async_init()
 
     async_dispatcher_connect(hass, SIGNAL_CAST_DISCOVERED, async_cast_discovered)
     for chromecast in hass.data[KNOWN_CHROMECAST_INFO_KEY].values():

diff --git a/custom_components/googlehome/sensor.py b/custom_components/googlehome/sensor.py
@@ -31,20 +31,20 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     async def async_cast_discovered(discover: ChromecastInfo):
         hass.data[DOMAIN].setdefault(discover.host, {})
 
-        await hass.data[CLIENT].update_info(discover.host)
-        info = hass.data[DOMAIN][discover.host].get("info", { "device_info": {} })
-        if info["device_info"]["capabilities"].get("assistant_supported", False):
-            devices = []
-            for condition in SENSOR_TYPES:
-                device = GoogleHomeAlarm(
-                    hass.data[CLIENT],
-                    config_entry,
-                    condition,
-                    discover,
-                    info.get("name", NAME),
-                )
-                devices.append(device)
-            async_add_entities(devices, True)
+        if await hass.data[CLIENT].update_info(discover.host):
+            info = hass.data[DOMAIN][discover.host].get("info", { "device_info": {} })
+            if info["device_info"]["capabilities"].get("assistant_supported", False):
+                devices = []
+                for condition in SENSOR_TYPES:
+                    device = GoogleHomeAlarm(
+                        hass.data[CLIENT],
+                        config_entry,
+                        condition,
+                        discover,
+                        info.get("name", NAME),
+                    )
+                    devices.append(device)
+                async_add_entities(devices, True)
 
     async_dispatcher_connect(hass, SIGNAL_CAST_DISCOVERED, async_cast_discovered)
     for chromecast in hass.data[KNOWN_CHROMECAST_INFO_KEY].values():