# Sean Corrigan 2020
import speedtest
import json
from influxdb import InfluxDBClient

serverip = "192.168.0.13"
serverport = 8086
serverdatabase = "telegraf"

def speed():    # To be called on schedule
    servers = [] # If you want to test against a specific server
    threads = 4
    # Choose the amount of threads to use for the test
    test = speedtest.Speedtest()
    test.get_servers(servers)
    test.get_best_server()
    test.download(threads=threads)
    test.upload(threads=threads)
    test.results.share()

    results = test.results.dict()
    

    resultlist = []
    resultupload = {}
    resultupload["UploadSpeed"] = results["upload"]
    resultdownload = {}
    resultdownload["DownloadSpeed"] = results["download"]
    resultlatency = {}
    resultlatency["Latency"] = results["ping"]
    resultlist.append(resultupload)
    resultlist.append(resultdownload)
    resultlist.append(resultlatency)

    resultforupload = json.dumps(resultlist)
    print(resultforupload)

    return(resultforupload)

def influxupload(): # Still having errors in uploading the data..
    jsondata = speed()
    client = InfluxDBClient(host=serverip, port=8086, database=serverdatabase)
    client.write(str(jsondata), protocol='json')

influxupload()