upated

db.py

@@ -0,0 +1,89 @@
+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()
+