Add callback for abode connection status

abodepy/event_controller.py

@@ -20,8 +20,10 @@ def __init__(self, abode, url=CONST.SOCKETIO_URL):
         self._abode = abode
         self._thread = None
         self._running = False
+        self._connected = False
 
         # Setup callback dicts
+        self._connection_status_callbacks = collections.defaultdict(list)
         self._device_callbacks = collections.defaultdict(list)
         self._event_callbacks = collections.defaultdict(list)
         self._timeline_callbacks = collections.defaultdict(list)
@@ -33,6 +35,7 @@ def __init__(self, abode, url=CONST.SOCKETIO_URL):
         # Setup SocketIO Callbacks
         self._socketio.on(sio.STARTED, self._on_socket_started)
         self._socketio.on(sio.CONNECTED, self._on_socket_connected)
+        self._socketio.on(sio.DISCONNECTED, self._on_socket_disconnected)
         self._socketio.on(CONST.DEVICE_UPDATE_EVENT, self._on_device_update)
         self._socketio.on(CONST.GATEWAY_MODE_EVENT, self._on_mode_change)
         self._socketio.on(CONST.TIMELINE_EVENT, self._on_timeline_update)
@@ -46,6 +49,30 @@ def stop(self):
         """Tell the subscription thread to terminate - will block."""
         self._socketio.stop()
 
+    def add_connection_status_callback(self, devices, callback):
+        """Register callback for Abode server connection status."""
+        if not devices:
+            return False
+
+        if not isinstance(devices, (tuple, list)):
+            devices = [devices]
+
+        for device in devices:
+            device_id = device
+
+            if isinstance(device, AbodeDevice):
+                device_id = device.device_id
+
+            if not self._abode.get_device(device_id):
+                raise AbodeException((ERROR.EVENT_DEVICE_INVALID))
+
+            _LOGGER.debug(
+                "Subscribing to Abode connection updates for device_id: %s", device_id)
+
+        self._connection_status_callbacks[device_id].append((callback))
+
+        return True
+
     def add_device_callback(self, devices, callback):
         """Register a device callback."""
         if not devices:
@@ -98,6 +125,8 @@ def remove_all_device_callbacks(self, devices):
 
             self._device_callbacks[device_id].clear()
 
+            self._connection_status_callbacks[device_id].clear()
+
         return True
 
     def add_event_callback(self, event_groups, callback):
@@ -142,6 +171,11 @@ def add_timeline_callback(self, timeline_events, callback):
 
         return True
 
+    @property
+    def connected(self):
+        """Get the Abode connection status."""
+        return self._connected
+
     @property
     def socketio(self):
         """Get the SocketIO instance."""
@@ -158,8 +192,23 @@ def _on_socket_started(self):
 
     def _on_socket_connected(self):
         """Socket IO connected callback."""
+        self._connected = True
+
         self._abode.refresh()
 
+        for callback in self._connection_status_callbacks.items():
+            _execute_callback(callback[1][0])
+
+    def _on_socket_disconnected(self):
+        """Socket IO disconnected callback."""
+        self._connected = False
+
+        for callback in self._connection_status_callbacks.items():
+            # Checks if list is empty for when remove_all_device_callbacks
+            # is called before _on_socket_disconnected.
+            if callback[1]:
+                _execute_callback(callback[1][0])
+
     def _on_device_update(self, devid):
         """Device callback from Abode SocketIO server."""
         if isinstance(devid, (tuple, list)):