Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-22 19:23:34 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add timeout to query (#19)

Browse Source
This commit is contained in:
Austin 2017-01-07 14:42:31 -08:00 committed by GadgetReactor
parent fd4e363f56
commit 1e01530447
2 changed files with 15 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pyHS100/protocol.py
View File

@ -39,6 +39,8 @@ class TPLinkSmartHomeProtocol:
request = json.dumps(request)
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.settimeout(5.0)
try:
sock.connect((host, port))
_LOGGER.debug("> (%i) %s", len(request), request)
@ -51,6 +53,7 @@ class TPLinkSmartHomeProtocol:
if not chunk:
break
finally:
sock.shutdown(socket.SHUT_RDWR)
sock.close()

2
pyHS100/pyHS100.py
View File

@ -103,7 +103,7 @@ class SmartPlug:
request={target: {cmd: arg}}
)
except Exception as ex:
raise SmartPlugException(ex)
raise SmartPlugException(ex) from ex
result = response[target]
if "err_code" in result and result["err_code"] != 0: