Add auto connect to 360 eye

Author: shenxn

libpurecool/dyson_360_eye.py

@@ -17,6 +17,20 @@
 class Dyson360Eye(DysonDevice):
     """Dyson 360 Eye device."""
 
+    def auto_connect(self, timeout=5, retry=15):
+        """Try to connect to device using mDNS.
+
+        :param timeout: Timeout
+        :param retry: Max retry
+        :return: True if connected, else False
+        """
+        return self._auto_connect("_360eye_mqtt._tcp.local.", timeout, retry)
+
+    @staticmethod
+    def _device_serial_from_name(name):
+        """Get device serial from mDNS name."""
+        return (name.split(".")[0]).split("-", 1)[1]
+
     def connect(self, device_ip, device_port=DEFAULT_PORT):
         """Try to connect to device.
 
@@ -27,6 +41,10 @@ def connect(self, device_ip, device_port=DEFAULT_PORT):
         self._network_device = NetworkDevice(self._name, device_ip,
                                              device_port)
 
+        return self._mqtt_connect()
+
+    def _mqtt_connect(self):
+        """Connect to the MQTT broker."""
         self._mqtt = mqtt.Client(userdata=self, protocol=3)
         self._mqtt.username_pw_set(self._serial, self._credentials)
         self._mqtt.on_message = self.on_message

libpurecool/dyson_device.py

@@ -2,14 +2,16 @@
 
 # pylint: disable=too-many-public-methods,too-many-instance-attributes
 
-from queue import Queue
+from queue import Queue, Empty
 import logging
 import json
 import abc
 import time
+import socket
 
 from .utils import printable_fields
 from .utils import decrypt_password
+from .zeroconf import ServiceBrowser, Zeroconf
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -64,6 +66,41 @@ def __repr__(self):
 class DysonDevice:
     """Abstract Dyson device."""
 
+    class DysonDeviceListener:
+        """Message listener."""
+
+        def __init__(self, serial, add_device_function, serial_from_name):
+            """Create a new message listener.
+
+            :param serial: Device serial
+            :param add_device_function: Callback function
+            """
+            self._serial = serial
+            self.add_device_function = add_device_function
+            self.serial_from_name = serial_from_name
+
+        def remove_service(self, zeroconf, device_type, name):
+            # pylint: disable=unused-argument,no-self-use
+            """Remove listener."""
+            _LOGGER.info("Service %s removed", name)
+
+        def add_service(self, zeroconf, device_type, name):
+            """Add device.
+
+            :param zeroconf: MSDNS object
+            :param device_type: Service type
+            :param name: Device name
+            """
+            device_serial = self.serial_from_name(name)
+            if device_serial == self._serial:
+                # Find searched device
+                info = zeroconf.get_service_info(device_type, name)
+                address = socket.inet_ntoa(info.address)
+                network_device = NetworkDevice(device_serial, address,
+                                               info.port)
+                self.add_device_function(network_device)
+                zeroconf.close()
+
     @staticmethod
     def on_connect(client, userdata, flags, return_code):
         # pylint: disable=unused-argument
@@ -119,6 +156,35 @@ def connect(self, device_ip, device_port=DEFAULT_PORT):
         """
         return
 
+    def _auto_connect(self, type_, timeout=5, retry=15):
+        """Try to connect to device using mDNS."""
+        for i in range(retry):
+            zeroconf = Zeroconf()
+            listener = self.DysonDeviceListener(self._serial,
+                                                self._add_network_device,
+                                                self._device_serial_from_name)
+            ServiceBrowser(zeroconf, type_, listener)
+            try:
+                self._network_device = self._search_device_queue.get(
+                    timeout=timeout)
+            except Empty:
+                # Unable to find device
+                _LOGGER.warning("Unable to find device %s, try %s",
+                                self._serial, i)
+                zeroconf.close()
+            else:
+                break
+        if self._network_device is None:
+            _LOGGER.error("Unable to connect to device %s", self._serial)
+            return False
+        return self._mqtt_connect()
+
+    @staticmethod
+    @abc.abstractmethod
+    def _device_serial_from_name(name):
+        """Get device serial from mDNS name."""
+        return
+
     @property
     @abc.abstractmethod
     def status_topic(self):

libpurecool/dyson_pure_cool_link.py

@@ -5,7 +5,6 @@
 import json
 import logging
 import time
-import socket
 from threading import Thread
 from queue import Queue, Empty
 
@@ -20,48 +19,13 @@
     support_heating_v2
 from .dyson_pure_state import DysonPureHotCoolState, DysonPureCoolState, \
     DysonEnvironmentalSensorState
-from .zeroconf import ServiceBrowser, Zeroconf
 
 _LOGGER = logging.getLogger(__name__)
 
 
 class DysonPureCoolLink(DysonDevice):
     """Dyson device (fan)."""
 
-    class DysonDeviceListener:
-        """Message listener."""
-
-        def __init__(self, serial, add_device_function):
-            """Create a new message listener.
-
-            :param serial: Device serial
-            :param add_device_function: Callback function
-            """
-            self._serial = serial
-            self.add_device_function = add_device_function
-
-        def remove_service(self, zeroconf, device_type, name):
-            # pylint: disable=unused-argument,no-self-use
-            """Remove listener."""
-            _LOGGER.info("Service %s removed", name)
-
-        def add_service(self, zeroconf, device_type, name):
-            """Add device.
-
-            :param zeroconf: MSDNS object
-            :param device_type: Service type
-            :param name: Device name
-            """
-            device_serial = (name.split(".")[0]).split("_")[1]
-            if device_serial == self._serial:
-                # Find searched device
-                info = zeroconf.get_service_info(device_type, name)
-                address = socket.inet_ntoa(info.address)
-                network_device = NetworkDevice(device_serial, address,
-                                               info.port)
-                self.add_device_function(network_device)
-                zeroconf.close()
-
     def __init__(self, json_body):
         """Create a new Pure Cool Link device.
 
@@ -119,25 +83,12 @@ def auto_connect(self, timeout=5, retry=15):
         :param retry: Max retry
         :return: True if connected, else False
         """
-        for i in range(retry):
-            zeroconf = Zeroconf()
-            listener = self.DysonDeviceListener(self._serial,
-                                                self._add_network_device)
-            ServiceBrowser(zeroconf, "_dyson_mqtt._tcp.local.", listener)
-            try:
-                self._network_device = self._search_device_queue.get(
-                    timeout=timeout)
-            except Empty:
-                # Unable to find device
-                _LOGGER.warning("Unable to find device %s, try %s",
-                                self._serial, i)
-                zeroconf.close()
-            else:
-                break
-        if self._network_device is None:
-            _LOGGER.error("Unable to connect to device %s", self._serial)
-            return False
-        return self._mqtt_connect()
+        return self._auto_connect("_dyson_mqtt._tcp.local.", timeout, retry)
+
+    @staticmethod
+    def _device_serial_from_name(name):
+        """Get device serial from mDNS name."""
+        return (name.split(".")[0]).split("_")[1]
 
     def connect(self, device_ip, device_port=DEFAULT_PORT):
         """Connect to the device using ip address.

tests/test_libpurecoollink.py

@@ -103,6 +103,10 @@ def on_add_device(network_device):
     pass
 
 
+def device_serial_from_name(name):
+    return (name.split(".")[0]).split("_")[1]
+
+
 class TestLibPureCoolLink(unittest.TestCase):
     def setUp(self):
         pass
@@ -237,7 +241,8 @@ def test_status_topic(self):
     @mock.patch('socket.inet_ntoa', )
     def test_device_dyson_listener(self, mocked_ntoa):
         listener = DysonPureCoolLink.DysonDeviceListener('serial-1',
-                                                         on_add_device)
+                                                         on_add_device,
+                                                         device_serial_from_name)
         zeroconf = Mock()
         listener.remove_service(zeroconf, "ptype", "serial-1")
         info = Mock()