68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
import os
|
|
import glob
|
|
import matplotlib.pyplot as plt
|
|
import math
|
|
import csv
|
|
from datetime import datetime
|
|
from cfspeedtest import CloudflareSpeedtest
|
|
|
|
def run_test():
|
|
suite = CloudflareSpeedtest()
|
|
results = suite.run_all()
|
|
tests = results['tests']
|
|
time = tests['isp'].time
|
|
latency = tests['latency'].value
|
|
jitter = tests['jitter'].value
|
|
h_down = tests['100kB_down_bps'].value
|
|
o_down = tests['1MB_down_bps'].value
|
|
t_down = tests['10MB_down_bps'].value
|
|
q_down = tests['25MB_down_bps'].value
|
|
np_down = tests['90th_percentile_down_bps'].value
|
|
h_up = tests['100kB_up_bps'].value
|
|
o_up = tests['1MB_up_bps'].value
|
|
t_up = tests['10MB_up_bps'].value
|
|
np_up = tests['90th_percentile_up_bps'].value
|
|
return [time, latency, jitter, h_down, o_down, t_down, q_down, h_up, o_up, t_up, np_down, np_up]
|
|
|
|
def write_data(filename, data):
|
|
with open(filename, 'a', newline='\n') as csvfile:
|
|
writer = csv.writer(csvfile)
|
|
if os.path.getsize(filename) == 0:
|
|
writer.writerow(['time','latency','jitter','100kB_down','1MB_down','10MB_down','25MB_down','100kB_up','1MB_up','10MB_up','90th_percentile_down','90th_percentile_up'])
|
|
writer.writerow(data)
|
|
|
|
def plot(time, data1, data2, label1, label2, filename, autofmt=True, xticks=None):
|
|
fig_length = -11.36 + 5.46 * math.log(len(time))
|
|
fig_length = max(6.0, fig_length)
|
|
plt.figure(figsize=[int(fig_length), 6])
|
|
plt.plot(time, data1)
|
|
plt.plot(time, data2)
|
|
plt.legend([label1, label2])
|
|
if autofmt:
|
|
plt.gcf().autofmt_xdate()
|
|
if xticks is not None:
|
|
plt.xticks(xticks)
|
|
plt.savefig(filename)
|
|
|
|
def import_csv(filenames):
|
|
time_data = []
|
|
ping_data = []
|
|
jitter_data = []
|
|
up_data = []
|
|
down_data = []
|
|
|
|
for i in range(len(filenames)):
|
|
files = glob.glob(filenames[i])
|
|
for j in range(len(files)):
|
|
with open(files[j], 'r', newline='\n') as csvfile:
|
|
reader = csv.reader(csvfile)
|
|
line_count = 0
|
|
for row in reader:
|
|
if line_count != 0:
|
|
time_data.append(datetime.fromtimestamp(float(row[0])))
|
|
ping_data.append(float(row[1]))
|
|
jitter_data.append(float(row[2]))
|
|
down_data.append(int(row[10]) / 1000000)
|
|
up_data.append(int(row[11]) / 1000000)
|
|
line_count += 1
|
|
return time_data, ping_data, jitter_data, up_data, down_data |