From edb122edae293fd666ca66ff786954442ce24287 Mon Sep 17 00:00:00 2001
From: NickSavino <nicksavino2@gmail.com>
Date: Mon, 24 Feb 2025 13:37:10 -0700
Subject: [PATCH] added error failsafe to netspeed command

---
 src/tasks.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/tasks.py b/src/tasks.py
index b993590..88a6fea 100644
--- a/src/tasks.py
+++ b/src/tasks.py
@@ -89,11 +89,15 @@ def netspeed(c, verbose=0):
     # If you want to use a single threaded test
     # threads = 1
 
-    s = speedtest.Speedtest()
-    s.get_servers(servers)
-    s.get_best_server()
-    s.download(threads=threads)
-    s.upload(threads=threads)
+    try:
+        s = speedtest.Speedtest()
+        s.get_servers(servers)
+        s.get_best_server()
+        s.download(threads=threads)
+        s.upload(threads=threads)
+    except Exception as e:
+        raise invoke.Exit("ERROR: Speedtest failed.", e)
+   
 
     results_dict = s.results.dict()