90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
import sqlite3
|
|
from config import DB_PATH
|
|
|
|
|
|
def init_db(db_path=DB_PATH):
|
|
conn = sqlite3.connect(db_path)
|
|
c = conn.cursor()
|
|
c.execute('''
|
|
CREATE TABLE IF NOT EXISTS speed_tests (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
timestamp REAL NOT NULL,
|
|
failed BOOLEAN NOT NULL,
|
|
isp TEXT,
|
|
ip TEXT,
|
|
location_code TEXT,
|
|
location_city TEXT,
|
|
location_region TEXT,
|
|
latency REAL,
|
|
jitter REAL,
|
|
down_100kB REAL,
|
|
down_1MB REAL,
|
|
down_10MB REAL,
|
|
down_25MB REAL,
|
|
down_90th REAL,
|
|
up_100kB REAL,
|
|
up_1MB REAL,
|
|
up_10MB REAL,
|
|
up_90th REAL
|
|
)
|
|
''')
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
def insert_result(results: dict|None, db_path=DB_PATH):
|
|
conn = sqlite3.connect(db_path)
|
|
c = conn.cursor()
|
|
|
|
# If the test failed entirely, store it as a failure with timestamp now
|
|
if results is None or "tests" not in results:
|
|
from time import time
|
|
c.execute("INSERT INTO speed_tests (timestamp, failed) VALUES (?, ?)", (time(), True))
|
|
conn.commit()
|
|
conn.close()
|
|
return
|
|
|
|
tests = results.get("tests", {})
|
|
meta = results.get("meta", {})
|
|
|
|
# Get a consistent timestamp from any TestResult (or fallback to now)
|
|
from time import time as now
|
|
sample_test = next(iter(tests.values()), None)
|
|
timestamp = sample_test.time if sample_test else now()
|
|
|
|
print(tests)
|
|
print(meta)
|
|
|
|
def get(tests, key):
|
|
return tests[key].value if key in tests else None
|
|
|
|
c.execute('''
|
|
INSERT INTO speed_tests (
|
|
timestamp, failed, isp, ip, location_code, location_city, location_region,
|
|
latency, jitter,
|
|
down_100kB, down_1MB, down_10MB, down_25MB, down_90th,
|
|
up_100kB, up_1MB, up_10MB, up_90th
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
''', (
|
|
timestamp, False,
|
|
get(tests, "isp"),
|
|
get(meta, "ip"),
|
|
get(meta, "location_code"),
|
|
get(meta, "location_city"),
|
|
get(meta, "location_region"),
|
|
get(tests, "latency"),
|
|
get(tests, "jitter"),
|
|
get(tests, "100kB_down_mbps"),
|
|
get(tests, "1MB_down_mbps"),
|
|
get(tests, "10MB_down_mbps"),
|
|
get(tests, "25MB_down_mbps"),
|
|
get(tests, "90th_percentile_down_mbps"),
|
|
get(tests, "100kB_up_mbps"),
|
|
get(tests, "1MB_up_mbps"),
|
|
get(tests, "10MB_up_mbps"),
|
|
get(tests, "90th_percentile_up_mbps")
|
|
))
|
|
conn.commit()
|
|
conn.close()
|
|
|