Check the future state when setting exception

marcelblijleven · May 21, 2024 · 9289cde · 9289cde
1 parent a48299b
commit 9289cde
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/goodwe/protocol.py b/goodwe/protocol.py
@@ -146,7 +146,8 @@ def datagram_received(self, data: bytes, addr: Tuple[str, int]) -> None:
             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)
+            if self.response_future and not self.response_future.done():
+                self.response_future.set_exception(ex)
             self._close_transport()
 
     def error_received(self, exc: Exception) -> None:
@@ -290,7 +291,8 @@ def data_received(self, data: bytes) -> None:
             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)
+            if self.response_future and not self.response_future.done():
+                self.response_future.set_exception(ex)
             # self._close_transport()
 
     def error_received(self, exc: Exception) -> None: