From eef0da41fde2b19a68832a386f47a07167c2382d Mon Sep 17 00:00:00 2001
From: Silver Kuusik <silver.kuusik@gmail.com>
Date: Thu, 17 May 2018 18:46:54 +0300
Subject: [PATCH] add reconnect to WiFi and WebSocket

---
 main.py | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index cbdd363..747aea8 100644
--- a/main.py
+++ b/main.py
@@ -2,6 +2,8 @@ import _thread
 import ubinascii
 import uwebsockets
 
+# to remember WiFi disconnects
+has_wifi_connection = False
 # WebSocket connection
 conn = None
 # blockly highlihgt WebSocket connection
@@ -53,13 +55,30 @@ def step():
 def ws_handler():
     global ast
     global conn
+    global has_wifi_connection
 
     while True:
+        # when WiFi is connected
+        if wlan.isconnected():
+            # when WiFi has just been reconnected
+            if not has_wifi_connection:
+                conn = uwebsockets.connect(url)
+                sumorobot.set_led(STATUS, True)
+                has_wifi_connection = True
+        else: # when WiFi is not connected
+            # when WiFi has just been disconnected
+            if has_wifi_connection:
+                sumorobot.set_led(STATUS, False)
+                has_wifi_connection = False
+            # continue to wait for a WiFi connection
+            continue
+
         try:
             fin, opcode, data = conn.read_frame()
         except: # urror
-            print("Exception while reading from socket, attempting reconnect")
-            conn = uwebsockets.connect(url)
+            # socket timeout, no data received
+            print("socket timeout")
+            # continue to read again from socket
             continue
 
         if data == b"forward":
@@ -108,6 +127,9 @@ while not wlan.isconnected():
 print("Connecting to:", url)
 conn = uwebsockets.connect(url)
 
+# set X seconds timeout for socket reads
+conn.settimeout(1)
+
 # send a ping to the robot
 print("Sending ping")
 conn.send("{'ping': true}")