From bdf06eb598059e45f6e8737509932f5a2b94f34c Mon Sep 17 00:00:00 2001
From: Nixellion <nixesvfx@gmail.com>
Date: Tue, 23 Jun 2020 02:18:45 +0300
Subject: [PATCH] Update speedtester.py

---
 speedtester.py | 29 +++++++++++++++++------------
 1 file changed, 17 insertions(+), 12 deletions(-)

diff --git a/speedtester.py b/speedtester.py
index 60d0230..c897d0d 100644
--- a/speedtester.py
+++ b/speedtester.py
@@ -55,21 +55,26 @@ if __name__ == "__main__":
         servers = []
         threads = None
 
-        log.debug("Initializing speedtest...")
-        s = speedtest.Speedtest()
+        for i in range (0, 3):
+            try:
+                log.debug("Initializing speedtest...")
+                s = speedtest.Speedtest()
 
 
-        log.debug(f"Running test...")
-        s.get_servers(servers)
-        s.get_best_server()
-        s.download(threads=threads)
-        s.upload(threads=threads, pre_allocate=False)
+                log.debug(f"Running test...")
+                s.get_servers(servers)
+                s.get_best_server()
+                s.download(threads=threads)
+                s.upload(threads=threads, pre_allocate=False)
 
-        results_dict = s.results.dict()
-        download = round(results_dict['download']/1000000, 2)
-        upload = round(results_dict['upload']/1000000, 2)
-        # download = uniform(0,2)
-        # upload = uniform(0,2)
+                results_dict = s.results.dict()
+                download = round(results_dict['download']/1000000, 2)
+                upload = round(results_dict['upload']/1000000, 2)
+                # download = uniform(0,2)
+                # upload = uniform(0,2)
+                break
+            except:
+                log.error(f"Test failed, try {i+1}/3", exc_info=True)
 
         log.debug(f"{download}mbps, {upload}mbps")