From 9d5984e4cf07e7fff3b93cfd5b0bed546e6e9e74 Mon Sep 17 00:00:00 2001
From: rospogrigio <rospogrigio@gmail.com>
Date: Wed, 7 Jun 2023 11:59:31 +0200
Subject: [PATCH] Fix for socket.send() exceptions in the logs

---
 custom_components/localtuya/manifest.json      | 2 +-
 custom_components/localtuya/pytuya/__init__.py | 7 +++----
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/custom_components/localtuya/manifest.json b/custom_components/localtuya/manifest.json
index dad10d4..28e36fa 100644
--- a/custom_components/localtuya/manifest.json
+++ b/custom_components/localtuya/manifest.json
@@ -10,5 +10,5 @@
   "iot_class": "local_push",
   "issue_tracker": "https://github.com/rospogrigio/localtuya/issues",
   "requirements": [],
-  "version": "5.2.0"
+  "version": "5.2.1"
 }
diff --git a/custom_components/localtuya/pytuya/__init__.py b/custom_components/localtuya/pytuya/__init__.py
index e2abc58..fd69e7a 100644
--- a/custom_components/localtuya/pytuya/__init__.py
+++ b/custom_components/localtuya/pytuya/__init__.py
@@ -766,10 +766,9 @@ class TuyaProtocol(asyncio.Protocol, ContextualLogger):
 
         enc_payload = self._encode_message(payload)
         self.transport.write(enc_payload)
-        try:
-            msg = await self.dispatcher.wait_for(seqno, payload.cmd)
-        except Exception as ex:
-            self.debug("Wait was aborted for seqno %d (%s)", seqno, ex)
+        msg = await self.dispatcher.wait_for(seqno, payload.cmd)
+        if msg is None:
+            self.debug("Wait was aborted for seqno %d", seqno)
             return None
 
         # TODO: Verify stuff, e.g. CRC sequence number?