Ensure DIAL info in get_chromecasts (#47)

skorokithakis · Aug 23, 2017 · 28c5075 · 28c5075
1 parent 7ea9a5f
commit 28c5075
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/catt/controllers.py b/catt/controllers.py
@@ -19,7 +19,8 @@ def get_chromecasts():
         raise CattCastError("No devices found.")
 
     devices.sort(key=lambda cc: cc.name)
-    return devices
+    # We need to ensure that all Chromecast objects contain DIAL info.
+    return [pychromecast.Chromecast(cc.host) for cc in devices]
 
 
 def get_chromecast(device_name):
@@ -146,8 +147,7 @@ def __init__(self, device_name, state_check=True):
                 raise ValueError
             self.cast = pychromecast.Chromecast(cached_ip)
         except (pychromecast.error.ChromecastConnectionError, ValueError):
-            cast = get_chromecast(device_name)
-            self.cast = pychromecast.Chromecast(cast.host)
+            self.cast = get_chromecast(device_name)
             cache.set(self.cast.name, self.cast.host)
 
         self.cast.wait()