102 lines
3.4 KiB
Python
Executable File
102 lines
3.4 KiB
Python
Executable File
import time
|
|
import json
|
|
import subprocess
|
|
import os
|
|
from influxdb import InfluxDBClient
|
|
|
|
# InfluxDB Settings
|
|
DB_ADDRESS = os.environ.get('INFLUX_DB_ADDRESS')
|
|
DB_PORT = int(os.environ.get('INFLUX_DB_PORT'))
|
|
DB_USER = os.environ.get('INFLUX_DB_USER')
|
|
DB_PASSWORD = os.environ.get('INFLUX_DB_PASSWORD')
|
|
DB_DATABASE = os.environ.get('INFLUX_DB_DATABASE')
|
|
|
|
# Speedtest Settings
|
|
TEST_INTERVAL = int(os.environ.get('SPEEDTEST_INTERVAL')) # Time between tests (in seconds).
|
|
TEST_FAIL_INTERVAL = int(os.environ.get('SPEEDTEST_FAIL_INTERVAL')) # Time before retrying a failed Speedtest (in seconds).
|
|
|
|
influxdb_client = InfluxDBClient(
|
|
DB_ADDRESS, DB_PORT, DB_USER, DB_PASSWORD, None)
|
|
|
|
|
|
def init_db():
|
|
databases = influxdb_client.get_list_database()
|
|
|
|
if len(list(filter(lambda x: x['name'] == DB_DATABASE, databases))) == 0:
|
|
influxdb_client.create_database(
|
|
DB_DATABASE) # Create if does not exist.
|
|
else:
|
|
influxdb_client.switch_database(DB_DATABASE) # Switch to if does exist.
|
|
def pkt_loss(data):
|
|
if 'packetLoss' in data.keys():
|
|
return data['packetLoss']
|
|
else:
|
|
return 0
|
|
|
|
def format_for_influx(cliout):
|
|
data = json.loads(cliout)
|
|
# There is additional data in the speedtest-cli output but it is likely not necessary to store.
|
|
influx_data = [
|
|
{
|
|
'measurement': 'ping',
|
|
'time': data['timestamp'],
|
|
'fields': {
|
|
'jitter': data['ping']['jitter'],
|
|
'latency': data['ping']['latency']
|
|
}
|
|
},
|
|
{
|
|
'measurement': 'download',
|
|
'time': data['timestamp'],
|
|
'fields': {
|
|
# Byte to Megabit
|
|
'bandwidth': data['download']['bandwidth'] / 125000,
|
|
'bytes': data['download']['bytes'],
|
|
'elapsed': data['download']['elapsed']
|
|
}
|
|
},
|
|
{
|
|
'measurement': 'upload',
|
|
'time': data['timestamp'],
|
|
'fields': {
|
|
# Byte to Megabit
|
|
'bandwidth': data['upload']['bandwidth'] / 125000,
|
|
'bytes': data['upload']['bytes'],
|
|
'elapsed': data['upload']['elapsed']
|
|
}
|
|
},
|
|
{
|
|
'measurement': 'packetLoss',
|
|
'time': data['timestamp'],
|
|
'fields': {
|
|
'packetLoss': pkt_loss(data)
|
|
}
|
|
}
|
|
]
|
|
|
|
return influx_data
|
|
|
|
|
|
def main():
|
|
init_db() # Setup the database if it does not already exist.
|
|
|
|
while (1): # Run a Speedtest and send the results to influxDB indefinitely.
|
|
speedtest = subprocess.run(
|
|
["speedtest", "--accept-license", "--accept-gdpr", "-f", "json"], capture_output=True)
|
|
|
|
if speedtest.returncode == 0: # Speedtest was successful.
|
|
data = format_for_influx(speedtest.stdout)
|
|
print("Speedtest Successful:")
|
|
if influxdb_client.write_points(data) == True:
|
|
print("Data written to DB successfully")
|
|
time.sleep(TEST_INTERVAL)
|
|
else: # Speedtest failed.
|
|
print("Speedtest Failed:")
|
|
print(speedtest.stderr)
|
|
print(speedtest.stdout)
|
|
time.sleep(TEST_FAIL_INTERVAL)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print('Speedtest CLI Data Logger to InfluxDB')
|
|
main() |