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

Close connection properly on error

Browse Source
This commit is contained in:
Pierre Ståhl
2020-10-04 22:07:38 +02:00
committed by rospogrigio
parent 2565074c9c
commit f971646333
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
custom_components/localtuya/common.py
View File

@@ -108,7 +108,7 @@ class TuyaDevice(pytuya.TuyaListener):
for entity in config_entry[CONF_ENTITIES]: for entity in config_entry[CONF_ENTITIES]:
self._dps_to_request[entity[CONF_ID]] = None self._dps_to_request[entity[CONF_ID]] = None
def connect(self, delay=None): def connect(self):
"""Connet to device if not already connected.""" """Connet to device if not already connected."""
if self._connect_task is None: if self._connect_task is None:
self._connect_task = asyncio.ensure_future(self._make_connection()) self._connect_task = asyncio.ensure_future(self._make_connection())

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

@@ -336,11 +336,13 @@ class TuyaProtocol(asyncio.Protocol):
def close(self): def close(self):
"""Close connection and abort all outstanding listeners.""" """Close connection and abort all outstanding listeners."""
if self.transport is not None: if self.transport is not None:
self.dispatcher.abort()
self.heartbeater.cancel()
transport = self.transport transport = self.transport
self.transport = None self.transport = None
transport.close() transport.close()
if self.dispatcher is not None:
self.dispatcher.abort()
if self.heartbeater is not None:
self.heartbeater.cancel()
async def exchange(self, command, dps=None): async def exchange(self, command, dps=None):
"""Send and receive a message, returning response from device.""" """Send and receive a message, returning response from device."""