Upstream/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2024-11-22 11:47:30 +01:00
Code Issues Projects Releases Wiki Activity

Improve OTAv2 error messages

Browse Source
This commit is contained in:
Otto Winter 2018-11-10 11:45:38 +01:00
parent 8203b8fcd3
commit 04dc848620
No known key found for this signature in database
GPG Key ID: DB66C0BE6013F97E
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
esphomeyaml/espota2.py
View File

@ -3,6 +3,7 @@ import logging
import random import random
import socket import socket
import sys import sys
import time
from esphomeyaml.core import ESPHomeYAMLError from esphomeyaml.core import ESPHomeYAMLError
@ -75,9 +76,9 @@ def receive_exactly(sock, amount, msg, expect, decode=True):
try: try:
check_error(data, expect) check_error(data, expect)
except OTAError: except OTAError as err:
sock.close() sock.close()
raise raise OTAError("Error {}: {}".format(msg, err))
while len(data) < amount: while len(data) < amount:
try: try:
@ -201,6 +202,7 @@ def perform_ota(sock, password, file_handle, filename):
receive_exactly(sock, 1, 'receive OK', RESPONSE_RECEIVE_OK) receive_exactly(sock, 1, 'receive OK', RESPONSE_RECEIVE_OK)
receive_exactly(sock, 1, 'Update end', RESPONSE_UPDATE_END_OK) receive_exactly(sock, 1, 'Update end', RESPONSE_UPDATE_END_OK)
send_check(sock, RESPONSE_OK, 'end acknowledgement') send_check(sock, RESPONSE_OK, 'end acknowledgement')
time.sleep(0.25)
_LOGGER.info("OTA successful") _LOGGER.info("OTA successful")