atse
/
thermostat
forked from M-Labs/thermostat
1
0
Fork 0
Code Issues Pull Requests Activity

Proper timeout implementation

This commit is contained in:
atse 2023-08-08 16:23:07 +08:00 committed by Tse Kwok Yan
parent 7ce39b3840
commit 77089b3e98
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytec/pytec/aioclient.py
View File

@ -15,6 +15,7 @@ class Client:
self._connecting_task = None
self._command_lock = asyncio.Lock()
self._report_mode_on = False
self.timeout = None
async def connect(self, host='192.168.1.26', port=23, timeout=None):
"""Connect to the Thermostat with host and port.
@ -31,6 +32,7 @@ class Client:
self._connecting_task = asyncio.create_task(
asyncio.wait_for(asyncio.open_connection(host, port), timeout)
)
self.timeout = timeout
try:
self._reader, self._writer = await self._connecting_task
except asyncio.CancelledError:
@ -71,7 +73,7 @@ class Client:
async def _read_line(self):
# read 1 line
chunk = await self._reader.readline()
chunk = await asyncio.wait_for(self._reader.readline(), self.timeout) # Only wait for response until timeout
return chunk.decode('utf-8', errors='ignore')
async def _read_write(self, command):