From e598f69843139197290b163897efad52c23ea7d1 Mon Sep 17 00:00:00 2001
From: Mathieu B <m4th1eu@protonmail.com>
Date: Sun, 1 Aug 2021 19:01:58 +0200
Subject: [PATCH] Improved speedtest skill

---
 .../skills/productivity/speedtest/__init__.py | 32 +++++++++++--------
 .../dialog/fr-fr/error_speedtest.dialog       |  2 ++
 .../dialog/fr-fr/result_speedtest.dialog      |  4 +--
 .../dialog/fr-fr/starting_speedtest.dialog    |  3 +-
 4 files changed, 25 insertions(+), 16 deletions(-)
 create mode 100644 jarvis/skills/productivity/speedtest/dialog/fr-fr/error_speedtest.dialog

diff --git a/jarvis/skills/productivity/speedtest/__init__.py b/jarvis/skills/productivity/speedtest/__init__.py
index 70b2fc9..ee033f1 100644
--- a/jarvis/skills/productivity/speedtest/__init__.py
+++ b/jarvis/skills/productivity/speedtest/__init__.py
@@ -11,21 +11,27 @@ class SpeedTestSkill(Skill, metaclass=SkillRegistering):
     @intent_file_handler("start_speedtest.intent", "StartSpeedTestIntent")
     def handle_speedtest(self, data):
         try:
-            self.speak_dialog('starting_speedtest')
+            self.speak_dialog_threaded('starting_speedtest')
+            result = start_speedtest()
 
-            speed = speedtest.Speedtest()
-            # speed.get_servers([])
-            speed.get_best_server()
-            speed.download()
-            speed.upload(pre_allocate=False)
-            # speed.results.share()
-            result = speed.results.dict()
-            down_speed = ('%.2f' % float((result["download"]) / 1000000))
-            up_speed = ('%.2f' % float((result["upload"]) / 1000000))
-
-            self.speak_dialog('result', {'DOWN': down_speed, 'UP': up_speed})
+            down_speed = result[0]
+            up_speed = result[1]
+            self.speak_dialog('result_speedtest', {'DOWN': down_speed, 'UP': up_speed})
         except:
-            self.speak_dialog("error")
+            self.speak("Error speedtest")
+
+
+def start_speedtest():
+    speed = speedtest.Speedtest(timeout=2)
+    speed.get_servers([])
+    speed.get_best_server()
+    speed.download()
+    speed.upload(pre_allocate=False)
+    result = speed.results.dict()
+    down_speed = ('%.2f' % float((result["download"]) / 1000000))
+    up_speed = ('%.2f' % float((result["upload"]) / 1000000))
+
+    return [down_speed, up_speed]
 
 
 def create_skill(data):
diff --git a/jarvis/skills/productivity/speedtest/dialog/fr-fr/error_speedtest.dialog b/jarvis/skills/productivity/speedtest/dialog/fr-fr/error_speedtest.dialog
new file mode 100644
index 0000000..2187f1d
--- /dev/null
+++ b/jarvis/skills/productivity/speedtest/dialog/fr-fr/error_speedtest.dialog
@@ -0,0 +1,2 @@
+Une erreur est survenue durant le test de débit, merci de réessayer plus tard...
+J'ai malheureusement fait face à une erreur durant le test de débit, merci de réessayer ultérieurement.
\ No newline at end of file
diff --git a/jarvis/skills/productivity/speedtest/dialog/fr-fr/result_speedtest.dialog b/jarvis/skills/productivity/speedtest/dialog/fr-fr/result_speedtest.dialog
index ac0c9a8..a250bdb 100644
--- a/jarvis/skills/productivity/speedtest/dialog/fr-fr/result_speedtest.dialog
+++ b/jarvis/skills/productivity/speedtest/dialog/fr-fr/result_speedtest.dialog
@@ -1,3 +1,3 @@
-Votre débit est de {{DOWN}} mégabits descendant et {{UP}} montant
-Vous atteignez des vitesses de {{DOWN}} mégabits en téléchargement et environ {{UP}} en envoi
+Votre débit est de {{DOWN}} mégabits descendant et {{UP}} mégabits montant
+Vous atteignez des vitesses de {{DOWN}} mégabits en téléchargement et environ {{UP}} mégabits en envoi
 Votre connexion atteint les {{DOWN}} mégabits en téléchargement et près de {{UP}} mégabits en envoi
\ No newline at end of file
diff --git a/jarvis/skills/productivity/speedtest/dialog/fr-fr/starting_speedtest.dialog b/jarvis/skills/productivity/speedtest/dialog/fr-fr/starting_speedtest.dialog
index 043c71a..37dbdb7 100644
--- a/jarvis/skills/productivity/speedtest/dialog/fr-fr/starting_speedtest.dialog
+++ b/jarvis/skills/productivity/speedtest/dialog/fr-fr/starting_speedtest.dialog
@@ -1 +1,2 @@
-Le test de débit commence, cela peut prendre un petit instant merci de patienter...
\ No newline at end of file
+Le test de débit commence, cela peut prendre un petit instant, je vous communique les résultats à la fin du test.
+Je viens de commencer le test de votre débit, ce test peut prendre un petit moment, je vous énoncerais les résultats dès que le test se termine.
\ No newline at end of file