Handle late arriving response which was already handled/canceled

marcelblijleven · May 12, 2024 · 35f6465 · 35f6465
1 parent 337a799
commit 35f6465
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/goodwe/protocol.py b/goodwe/protocol.py
@@ -114,6 +114,8 @@ def datagram_received(self, data: bytes, addr: Tuple[str, int]) -> None:
             else:
                 logger.debug("Received invalid response: %s", data.hex())
                 asyncio.get_running_loop().call_soon(self._retry_mechanism)
+        except asyncio.InvalidStateError:
+            logger.debug("Response already handled: %s", data.hex())
         except RequestRejectedException as ex:
             logger.debug("Received exception response: %s", data.hex())
             self.response_future.set_exception(ex)
@@ -226,6 +228,8 @@ def data_received(self, data: bytes) -> None:
                 logger.debug("Received invalid response: %s", data.hex())
                 self.response_future.set_exception(RequestRejectedException())
                 self._close_transport()
+        except asyncio.InvalidStateError:
+            logger.debug("Response already handled: %s", data.hex())
         except RequestRejectedException as ex:
             logger.debug("Received exception response: %s", data.hex())
             self.response_future.set_exception(ex)