9c246bc60f
This module provides a glimpse/snapshot of current internet based speeds. To keep things simple (and due to bugs in speedtest_cli), this module simply uses requests to download a small(ish) image and times the download. This is not a perfect/definitive test, but it should be a decent indicator of how the internet bound traffic looks on the clients network.
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
from i3pystatus import IntervalModule
|
|
import speedtest_cli
|
|
import requests
|
|
import time
|
|
import os
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
class NetSpeed(IntervalModule):
|
|
"""
|
|
Attempts to provide an estimation of internet speeds.
|
|
Requires: speedtest_cli
|
|
"""
|
|
|
|
settings = (
|
|
("url", "Target URL to download a file from. Uses speedtest_cli to "
|
|
"find the 'best' server if none is supplied."),
|
|
"format"
|
|
)
|
|
color = "#FFFFFF"
|
|
interval = 300
|
|
url = None
|
|
format = "{speed} ({hosting_provider})"
|
|
|
|
def run(self):
|
|
if not self.url:
|
|
config = speedtest_cli.getConfig()
|
|
servers = speedtest_cli.closestServers(config['client'])
|
|
best = speedtest_cli.getBestServer(servers)
|
|
# 1500x1500 is about 4.3MB, which seems like a reasonable place to
|
|
# start, i guess...
|
|
url = '%s/random1500x1500.jpg' % os.path.dirname(best['url'])
|
|
|
|
with open('/dev/null', 'wb') as devnull:
|
|
start = time.time()
|
|
req = requests.get(url, stream=True)
|
|
devnull.write(req.content)
|
|
end = time.time()
|
|
total_length = int(req.headers.get('content-length'))
|
|
devnull.close()
|
|
|
|
# chop off the float after the 4th decimal point
|
|
# note: not rounding, simply cutting
|
|
# note: dl_time is in seconds
|
|
dl_time = float(end - start)
|
|
|
|
if total_length < 999:
|
|
unit = "Bps"
|
|
bps = total_length / dl_time
|
|
|
|
if total_length >= 1000 < 999999:
|
|
unit = "KBps"
|
|
bps = (total_length / 1024.0) / dl_time
|
|
|
|
if total_length >= 1000000 < 999999999:
|
|
unit = "MBps"
|
|
bps = (total_length / (1024.0 * 1024.0)) / dl_time
|
|
|
|
if total_length >= 10000000:
|
|
unit = "GBps"
|
|
bps = (total_length / (1024.0 * 1024.0 * 1024.0)) / dl_time
|
|
|
|
bps = "%.2f" % bps
|
|
speed = "%s %s" % (bps, unit)
|
|
hosting_provider = '.'.join(urlparse(url).hostname.split('.')[-2:])
|
|
|
|
cdict = {
|
|
"speed": speed,
|
|
"hosting_provider": hosting_provider,
|
|
}
|
|
|
|
self.output = {
|
|
"full_text": self.format.format(**cdict),
|
|
"color": self.color
|
|
}
|