File: speedtest.py

package info (click to toggle)
bumblebee-status 2.2.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,844 kB
  • sloc: python: 13,430; sh: 68; makefile: 29
file content (65 lines) | stat: -rw-r--r-- 1,664 bytes parent folder | download | duplicates (2)
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
60
61
62
63
64
65
# pylint: disable=C0111,R0903

"""Performs a speedtest - only updates when the "play" button is clicked

Requires the following python module:
    * speedtest-cli

"""

import sys

import core.module
import core.widget
import core.input
import core.event
import core.decorators

import speedtest


class Module(core.module.Module):
    @core.decorators.never
    def __init__(self, config, theme):
        super().__init__(config, theme, [])

        self.background = True
        self.__result = "<speedtest>"
        self.__running = False

        start = self.add_widget(name="start")
        main = self.add_widget(name="main", full_text=self.result)

        core.input.register(start, button=core.input.LEFT_MOUSE, cmd=self.update_event)

    def result(self, _):
        return self.__result

    def update_event(self, _):
        self.__running = True
        self.update()

    def update(self):
        if not self.__running:
            return
        core.event.trigger("update", [self.id], redraw_only=True)
        s = speedtest.Speedtest()
        s.get_best_server()
        s.download(threads=None)
        s.upload(threads=None)

        self.__result = "ping: {:.2f}ms down: {:.2f}Mbps up: {:.2f}Mbps".format(
            s.results.ping,
            s.results.download / 1024 / 1024,
            s.results.upload / 1024 / 1024,
        )
        self.__running = False
        core.event.trigger("update", [self.id], redraw_only=True)

    def state(self, widget):
        if widget.name == "start":
            return "running" if self.__running else "not-running"
        return None


# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4