Fix: not all the places catch exceptions from exchange()

xZetsubou · Jul 29, 2024 · 7b0950f · 7b0950f
1 parent 21af3e1
commit 7b0950f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/custom_components/localtuya/core/pytuya/__init__.py b/custom_components/localtuya/core/pytuya/__init__.py
@@ -1086,7 +1086,10 @@ async def exchange(self, command, dps=None, nodeID=None, payload=None):
 
         enc_payload = self._encode_message(payload)
 
-        await self.transport_write(enc_payload)
+        try:
+            await self.transport_write(enc_payload)
+        except Exception:  # pylint: disable=broad-except
+            return self.clean_up_session()
         msg = await self.dispatcher.wait_for(seqno, payload.cmd)
         if msg is None:
             self.debug("Wait was aborted for seqno %d", seqno)