forked from luke5sky/discontinued-speedtest-skill
-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path__init__.py
59 lines (53 loc) · 2.32 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# Copyright 2018 Lukas Gangel
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import time
import speedtest
from ovos_utils.log import LOG
from ovos_workshop.decorators import intent_handler
from ovos_workshop.intents import IntentBuilder
from ovos_workshop.skills import OVOSSkill
class SpeedTestSkill(OVOSSkill):
@intent_handler(IntentBuilder("SpeedtestIntent").require("Run").require("Speedtest"))
def handle_speedtest_intent(self, message):
LOG.info("speedtest started")
try:
self.speak_dialog('running')
self.enclosure.deactivate_mouth_events()
self.enclosure.mouth_think()
self.gui.show_animated_image("wifi-speed.gif")
servers = []
speed = speedtest.Speedtest()
speed.get_servers(servers)
speed.get_best_server()
self.enclosure.eyes_look("d")
speed.download()
self.enclosure.eyes_look("u")
speed.upload(pre_allocate=False)
speed.results.share()
result = speed.results.dict()
LOG.info("speedtest finished")
self.enclosure.eyes_narrow()
downspeed = ('%.2f' % float((result["download"]) / 1000000))
upspeed = ('%.2f' % float((result["upload"]) / 1000000))
self.gui.show_text(f"UP: {upspeed} MB/S\nDOWN: {downspeed} MB/S")
self.enclosure.mouth_text(f"UP: {upspeed} MB/S DOWN: {downspeed} MB/S ")
self.speak_dialog('result', {'DOWN': downspeed, 'UP': upspeed}, wait=True)
time.sleep(2) # let speed test results scroll for a bit
self.enclosure.eyes_fill(100)
except:
self.speak_dialog("error")
self.enclosure.activate_mouth_events()
self.enclosure.mouth_reset()
self.enclosure.eyes_reset()
self.gui.release()