121 lines
4.3 KiB
Python
121 lines
4.3 KiB
Python
import sqlite3
|
|
from typing import Dict
|
|
|
|
DB_FILE = "dereth.db"
|
|
|
|
def init_db() -> None:
|
|
"""Create tables if they do not exist (extended with kills_per_hour and onlinetime)."""
|
|
conn = sqlite3.connect(DB_FILE)
|
|
c = conn.cursor()
|
|
|
|
# History log
|
|
c.execute("""
|
|
CREATE TABLE IF NOT EXISTS telemetry_log (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
character_name TEXT NOT NULL,
|
|
char_tag TEXT,
|
|
session_id TEXT NOT NULL,
|
|
timestamp TEXT NOT NULL,
|
|
ew REAL,
|
|
ns REAL,
|
|
z REAL,
|
|
kills INTEGER,
|
|
kills_per_hour TEXT,
|
|
onlinetime TEXT,
|
|
deaths INTEGER,
|
|
rares_found INTEGER,
|
|
prismatic_taper_count INTEGER,
|
|
vt_state TEXT
|
|
)
|
|
""")
|
|
|
|
# Live snapshot (upsert)
|
|
c.execute("""
|
|
CREATE TABLE IF NOT EXISTS live_state (
|
|
character_name TEXT PRIMARY KEY,
|
|
char_tag TEXT,
|
|
session_id TEXT,
|
|
timestamp TEXT,
|
|
ew REAL,
|
|
ns REAL,
|
|
z REAL,
|
|
kills INTEGER,
|
|
kills_per_hour TEXT,
|
|
onlinetime TEXT,
|
|
deaths INTEGER,
|
|
rares_found INTEGER,
|
|
prismatic_taper_count INTEGER,
|
|
vt_state TEXT
|
|
)
|
|
""")
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def save_snapshot(data: Dict) -> None:
|
|
"""Insert snapshot into history and upsert into live_state (with new fields)."""
|
|
conn = sqlite3.connect(DB_FILE)
|
|
c = conn.cursor()
|
|
|
|
# Insert full history row
|
|
c.execute("""
|
|
INSERT INTO telemetry_log (
|
|
character_name, char_tag, session_id, timestamp,
|
|
ew, ns, z,
|
|
kills, kills_per_hour, onlinetime,
|
|
deaths, rares_found, prismatic_taper_count, vt_state
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
""", (
|
|
data["character_name"],
|
|
data.get("char_tag", ""),
|
|
data["session_id"],
|
|
data["timestamp"],
|
|
data["ew"], data["ns"], data.get("z", 0.0),
|
|
data["kills"],
|
|
data.get("kills_per_hour", ""),
|
|
data.get("onlinetime", ""),
|
|
data.get("deaths", 0),
|
|
data.get("rares_found", 0),
|
|
data.get("prismatic_taper_count", 0),
|
|
data.get("vt_state", "Unknown"),
|
|
))
|
|
|
|
# Upsert into live_state
|
|
c.execute("""
|
|
INSERT INTO live_state (
|
|
character_name, char_tag, session_id, timestamp,
|
|
ew, ns, z,
|
|
kills, kills_per_hour, onlinetime,
|
|
deaths, rares_found, prismatic_taper_count, vt_state
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
ON CONFLICT(character_name) DO UPDATE SET
|
|
char_tag = excluded.char_tag,
|
|
session_id = excluded.session_id,
|
|
timestamp = excluded.timestamp,
|
|
ew = excluded.ew,
|
|
ns = excluded.ns,
|
|
z = excluded.z,
|
|
kills = excluded.kills,
|
|
kills_per_hour = excluded.kills_per_hour,
|
|
onlinetime = excluded.onlinetime,
|
|
deaths = excluded.deaths,
|
|
rares_found = excluded.rares_found,
|
|
prismatic_taper_count = excluded.prismatic_taper_count,
|
|
vt_state = excluded.vt_state
|
|
""", (
|
|
data["character_name"],
|
|
data.get("char_tag", ""),
|
|
data["session_id"],
|
|
data["timestamp"],
|
|
data["ew"], data["ns"], data.get("z", 0.0),
|
|
data["kills"],
|
|
data.get("kills_per_hour", ""),
|
|
data.get("onlinetime", ""),
|
|
data.get("deaths", 0),
|
|
data.get("rares_found", 0),
|
|
data.get("prismatic_taper_count", 0),
|
|
data.get("vt_state", "Unknown"),
|
|
))
|
|
|
|
conn.commit()
|
|
conn.close()
|