trigger device update callbacks when connection state changes

pyvlx/connection.py

@@ -76,12 +76,13 @@ def connection_lost(self, exc: object) -> None:
 class Connection:
     """Class for handling TCP connection."""
 
-    def __init__(self, loop: asyncio.AbstractEventLoop, config: Config):
+    def __init__(self, loop: asyncio.AbstractEventLoop, config: Config, connection_closed_cb: CallbackType = None):
         """Init TCP connection."""
         self.loop = loop
         self.config = config
         self.transport: Optional[asyncio.Transport] = None
         self.frame_received_cbs: List[CallbackType] = []
+        self.connection_closed_cb: CallbackType = None
         self.connected = False
         self.connection_counter = 0
 
@@ -95,6 +96,8 @@ def disconnect(self) -> None:
             self.transport.close()
             self.transport = None
         self.connected = False
+        if self.connection_closed_cb is not None:
+            self.loop.create_task(self.connection_closed_cb())
 
     async def connect(self) -> None:
         """Connect to gateway via SSL."""

pyvlx/pyvlx.py

@@ -36,7 +36,7 @@ def __init__(
         """Initialize PyVLX class."""
         self.loop = loop or asyncio.get_event_loop()
         self.config = Config(self, path, host, password)
-        self.connection = Connection(loop=self.loop, config=self.config)
+        self.connection = Connection(loop=self.loop, config=self.config, connection_closed_cb=self.connection_closed_cb)
         self.heartbeat = Heartbeat(
             pyvlx=self,
             interval=heartbeat_interval,
@@ -72,6 +72,9 @@ async def connect(self) -> None:
         await self.klf200.house_status_monitor_enable(pyvlx=self)
         self.heartbeat.start()
 
+        for node in self.nodes:
+            await self.loop.create_task(node.after_update())
+
     async def reboot_gateway(self) -> None:
         """For Compatibility: Reboot the KLF 200."""
         if not self.is_connected():
@@ -116,3 +119,8 @@ async def get_limitation(self, node_id: int) -> None:
         """Return limitation."""
         limit = get_limitation.GetLimitation(self, node_id)
         await limit.do_api_call()
+
+    async def connection_closed_cb(self) -> None:
+        """Callback when connection to KLF 200 is closed."""
+        for node in self.nodes:
+            await self.loop.create_task(node.after_update())