Skip to content

Commit 985cd9f

Browse files
committed
Handel connection reset error
1 parent d3f84eb commit 985cd9f

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

yeelib/bulbs.py

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -71,11 +71,16 @@ def send_command(self, method, params):
7171
logger.exception("Connection error")
7272
self.socket = None
7373
else:
74-
response = (yield from reader.readline()).decode()
75-
if response is not None:
76-
logger.debug("%s: ... %s",
77-
self.ip, response.strip())
78-
try:
79-
return json.loads(response)
80-
except ValueError:
81-
logger.exception("Could not read message: %s", response)
74+
try:
75+
response = (yield from reader.readline()).decode()
76+
except ConnectionError:
77+
logger.exception("Connection error")
78+
socket.socket = None
79+
else:
80+
if response is not None:
81+
logger.debug("%s: ... %s",
82+
self.ip, response.strip())
83+
try:
84+
return json.loads(response)
85+
except ValueError:
86+
logger.exception("Could not read message: %s", response)

0 commit comments

Comments
 (0)