From b797287359f4fd76b6125fea29012a3f15fc9e92 Mon Sep 17 00:00:00 2001 From: John Doe Date: Fri, 3 Jul 2015 14:08:10 +0300 Subject: [PATCH] Catch lost connection while uploading --- tools/espota.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tools/espota.py b/tools/espota.py index 58305c6df..52ef3e95a 100755 --- a/tools/espota.py +++ b/tools/espota.py @@ -56,7 +56,14 @@ def serve(remoteAddr, remotePort, filename): if not chunk: break sys.stderr.write('.') sys.stderr.flush() - connection.sendall(chunk) + try: + connection.sendall(chunk) + except: + print('\nError Uploading', file=sys.stderr) + connection.close() + f.close() + sock.close() + return 1 print('\nWaiting for result...\n', file=sys.stderr) try: @@ -65,11 +72,13 @@ def serve(remoteAddr, remotePort, filename): print('Result: %s' % data, file=sys.stderr) connection.close() f.close() + sock.close() return 0 except: print('Result: No Answer!', file=sys.stderr) connection.close() f.close() + sock.close() return 1 finally: