Added dockerfile and basic looping main.py to static IP
This commit is contained in:
parent
e7c03e06e4
commit
7201cc5371
8
Dockerfile
Normal file
8
Dockerfile
Normal file
@ -0,0 +1,8 @@
|
||||
FROM python:3
|
||||
|
||||
ADD main.py /
|
||||
|
||||
run pip install speedtest-cli
|
||||
run pip install influxdb
|
||||
|
||||
CMD [ "python", "./main.py" ]
|
||||
64
main.py
Normal file
64
main.py
Normal file
@ -0,0 +1,64 @@
|
||||
# Sean Corrigan 2020
|
||||
# Script sends data to influx in bits/sec
|
||||
|
||||
import speedtest
|
||||
import json
|
||||
from influxdb import InfluxDBClient
|
||||
from time import time, sleep
|
||||
|
||||
serverip = "192.168.0.13"
|
||||
serverport = 8086
|
||||
serverdatabase = "telegraf"
|
||||
servernickname = 'Yoo01pn.ddns.net'
|
||||
|
||||
def speed(): # Actual Speedtest using speedtest-cli
|
||||
servers = ['19249'] # If you want to test against a specific server eg. ['13030'] or [] for closest 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()
|
||||
|
||||
|
||||
result = {}
|
||||
result["UploadSpeed"] = results["upload"]
|
||||
result["DownloadSpeed"] = results["download"]
|
||||
result["Ping"] = results["ping"]
|
||||
|
||||
print(results["share"]) # Show share link
|
||||
return(result)
|
||||
|
||||
def uploadInfluxdata(host='192.168.0.13', port=8086): # Main upload section
|
||||
query = 'select Float_value from cpu_load_short;'
|
||||
query_where = 'select Int_value from cpu_load_short where host=$host;'
|
||||
bind_params = {'host': servernickname}
|
||||
testdata = speed()
|
||||
json_body = [
|
||||
{
|
||||
"measurement": "PythonSpeedTest",
|
||||
"tags": {
|
||||
"host": servernickname,
|
||||
},
|
||||
|
||||
"fields": {
|
||||
"Upload": testdata["UploadSpeed"],
|
||||
"Download": testdata["DownloadSpeed"],
|
||||
"Ping": testdata["Ping"]
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
client = InfluxDBClient(host, port, database = serverdatabase) # Init connection to Influx Server
|
||||
client.write_points(json_body) # Write Speedtest results
|
||||
|
||||
while True:
|
||||
print("Running Test")
|
||||
uploadInfluxdata()
|
||||
sleep(600)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user