williamp/localtuya-modded
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug when unloading an entry

Browse Source
This commit is contained in:
Pierre Ståhl
2020-10-05 12:48:45 +02:00
committed by rospogrigio
parent 5cd76f87ea
commit 59dfc35759
3 changed files with 20 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
custom_components/localtuya/pytuya/__init__.py
View File

@@ -359,7 +359,11 @@ class TuyaProtocol(asyncio.Protocol):
dev_type = self.dev_type
# Wait for special sequence number if heartbeat
seqno = MessageDispatcher.HEARTBEAT_SEQNO if command == HEARTBEAT else (self.seqno - 1)
seqno = (
MessageDispatcher.HEARTBEAT_SEQNO
if command == HEARTBEAT
else (self.seqno - 1)
)
self.transport.write(payload)
msg = await self.dispatcher.wait_for(seqno)
@@ -387,8 +391,6 @@ class TuyaProtocol(asyncio.Protocol):
async def heartbeat(self):
"""Send a heartbeat message."""
# We don't expect a response to this, just send blindly
#self.transport.write(self._generate_payload(HEARTBEAT))
return await self.exchange(HEARTBEAT)
async def set_dps(self, value, dps_index):