Use a timeout on fetching data, and handle failures gracefully.

gavanfantom · gavanfantom · commit 37b7f1d35410 · 2017-08-05T18:12:00.000+01:00
Prevents a crash in the installer when the wifi connection is flaky.
diff --git a/esp32/modules/installer.py b/esp32/modules/installer.py
@@ -1,4 +1,4 @@
-import ugfx, badge, network, gc, time, urequests, appglue
+import ugfx, badge, network, gc, time, urequests, appglue, sys
 
 # SHA2017 Badge installer
 # V2 Thomas Roos
@@ -71,10 +71,19 @@ def list_apps(slug):
 	ugfx.flush(ugfx.LUT_FULL)
 
 	try:
-		f = urequests.get("https://badge.sha2017.org/eggs/category/%s/json" % slug)
-		packages = f.json()
-	finally:
-		f.close()
+		f = urequests.get("https://badge.sha2017.org/eggs/category/%s/json" % slug, timeout=30)
+		try:
+			packages = f.json()
+		finally:
+			f.close()
+	except BaseException as e:
+		print("[Installer] Failed to download list of eggs:")
+		sys.print_exception(e)
+		text.text("Download failed")
+		ugfx.flush(ugfx.LUT_FULL)
+		list_categories()
+		gc.collect()
+		return
 
 	for package in packages:
 		options.add_item("%s rev. %s" % (package["name"], package["revision"]))
@@ -87,6 +96,7 @@ def list_apps(slug):
 	ugfx.input_attach(ugfx.BTN_START, lambda pushed: appglue.start_app('') if pushed else False)
 
 	show_description(True)
+	ugfx.flush(ugfx.LUT_FULL)
 	gc.collect()
 
 def start_categories(pushed):
@@ -145,7 +155,7 @@ def list_categories():
 		ugfx.input_init()
 		draw_msg('Getting categories')
 		try:
-			f = urequests.get("https://badge.sha2017.org/eggs/categories/json")
+			f = urequests.get("https://badge.sha2017.org/eggs/categories/json", timeout=30)
 			categories = f.json()
 		except:
 			draw_msg('Failed!')
