Add inactivity detection to plug udp listener.

Author: jmattsson

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "powersensor-local"
-version = "2.0.0"
+version = "2.0.1"
 description = "Network-local (non-cloud) interface for Powersensor devices"
 authors = [
   { name = "Jade Mattsson", email = "[email protected]" },

src/powersensor_local/plug_listener_udp.py

@@ -34,6 +34,7 @@ def __init__(self, ip, port=49476):
         self._backoff = 0               # exponential backoff
         self._transport = None          # UDP transport/socket
         self._reconnect = None          # reconnect timer
+        self._inactive = None           # inactivity timer
         self._disconnecting = False     # disconnecting flag
         self._was_connected = False     # 'disconnected' event armed?
 
@@ -58,6 +59,10 @@ async def _close_connection(self, unsub = True):
             self._reconnect.cancel()
             self._reconnect = None
 
+        if self._inactive is not None:
+            self._inactive.cancel()
+            self._inactive = None
+
         if self._transport is not None:
             if unsub:
                 self._transport.sendto(b'subscribe(0)\n')
@@ -93,6 +98,9 @@ def _send_subscribe(self):
         if self._transport is not None:
             self._transport.sendto(b'subscribe(60)\n')
 
+    def _on_inactivity(self):
+        asyncio.create_task(self._close_connection())
+
     # DatagramProtocol support below
 
     def protocol_factory(self):
@@ -112,6 +120,11 @@ def datagram_received(self, data, addr):
         if not self._was_connected:
             self._was_connected = True
 
+        if self._inactive is not None:
+            self._inactive.cancel()
+        loop = asyncio.get_running_loop()
+        self._inactive = loop.call_later(60, self._on_inactivity)
+
         lines = data.decode('utf-8').splitlines()
         for line in lines:
             try: