quizzmaster
diff --git a/‎tools/espota.py
Copy file name to clipboardExpand all lines: tools/espota.py
+8Lines changed: 8 additions & 0 deletions b/‎tools/espota.py
Copy file name to clipboardExpand all lines: tools/espota.py
+8Lines changed: 8 additions & 0 deletions
@@ -179,6 +179,7 @@ def serve(remoteAddr, localAddr, remotePort, localPort, password, filename, comm
       try:
         connection.sendall(chunk)
         res = connection.recv(10)
+        lastResponseContainedOK = 'OK' in res.decode()
       except:
         sys.stderr.write('\n')
         logging.error('Error Uploading')
@@ -187,6 +188,13 @@ def serve(remoteAddr, localAddr, remotePort, localPort, password, filename, comm
         sock.close()
         return 1
 
+    if lastResponseContainedOK:
+      logging.info('Success')
+      connection.close()
+      f.close()
+      sock.close()
+      return 0
+
     sys.stderr.write('\n')
     logging.info('Waiting for result...')
     try: