add make command make reformat

reformat python files with black
This commit is contained in:
Johan Lundberg 2025-05-01 20:41:53 +02:00
parent 66ed711fec
commit 431ce1c8d0
3 changed files with 82 additions and 58 deletions

84
db.py
View file

@ -3,13 +3,15 @@ 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("""
c.execute(
"""
CREATE TABLE IF NOT EXISTS telemetry_log (
id INTEGER PRIMARY KEY AUTOINCREMENT,
character_name TEXT NOT NULL,
@ -27,10 +29,12 @@ def init_db() -> None:
prismatic_taper_count INTEGER,
vt_state TEXT
)
""")
"""
)
# Live snapshot (upsert)
c.execute("""
c.execute(
"""
CREATE TABLE IF NOT EXISTS live_state (
character_name TEXT PRIMARY KEY,
char_tag TEXT,
@ -47,41 +51,49 @@ def init_db() -> None:
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("""
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"),
))
""",
(
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("""
c.execute(
"""
INSERT INTO live_state (
character_name, char_tag, session_id, timestamp,
ew, ns, z,
@ -102,20 +114,24 @@ def save_snapshot(data: Dict) -> None:
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"),
))
""",
(
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()