Prevent duplicate websockets

Author: shbatm

pyisy/events.py

@@ -337,6 +337,7 @@ def __init__(
         self._lasthb = None
         self._sid = None
         self._program_key = None
+        self.websocket_task = None
 
         if websession is None:
             websession = get_new_client_session(use_https, tls_ver)
@@ -358,12 +359,15 @@ def start(self):
     def stop(self):
         """Close websocket connection."""
         self.status = ES_STOP_UPDATES
-        self.websocket_task.cancel()
+        if self.websocket_task is not None and not self.websocket_task.done():
+            self.websocket_task.cancel()
 
     async def reconnect(self, delay=RECONNECT_DELAY, retries=0):
         """Reconnect to a disconnected websocket."""
         if self.status == ES_CONNECTED:
             return
+        if self.websocket_task is not None and not self.websocket_task.done():
+            self.websocket_task.cancel()
         self.status = ES_RECONNECTING
         _LOGGER.info("PyISY attempting stream reconnect in %ss.", delay)
         await asyncio.sleep(delay)
@@ -490,6 +494,8 @@ async def websocket(self, retries=0):
                 _LOGGER.error("Websocket Client Connector Error.")
                 self.status = ES_LOST_STREAM_CONNECTION
                 await self.reconnect(RECONNECT_DELAY)
+        except asyncio.CancelledError:
+            self.status = ES_DISCONNECTED
         except Exception as err:
             if self.status != ES_STOP_UPDATES:
                 _LOGGER.exception("Unexpected error %s", err)

pyisy/nodes/group.py

@@ -1,4 +1,6 @@
 """Representation of groups (scenes) from an ISY."""
+import asyncio
+
 from ..constants import ISY_VALUE_UNKNOWN, PROTO_GROUP
 from ..helpers import now
 from .nodebase import NodeBase
@@ -46,7 +48,7 @@ def __init__(
         ]
 
         # get and update the status
-        self.isy.loop.create_task(self.update())
+        asyncio.create_task(self.update())
 
     def __del__(self):
         """Cleanup event handlers before deleting."""
@@ -105,4 +107,4 @@ async def update(self, event=None, wait_time=0, hint=None, xmldoc=None):
 
     def update_callback(self, event=None):
         """Handle synchronous callbacks for subscriber events."""
-        self.isy.loop.create_task(self.update(event))
+        asyncio.create_task(self.update(event))