aboutsummaryrefslogtreecommitdiff
path: root/inspect_db.py
diff options
context:
space:
mode:
Diffstat (limited to 'inspect_db.py')
-rw-r--r--inspect_db.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/inspect_db.py b/inspect_db.py
new file mode 100644
index 0000000..0419f26
--- /dev/null
+++ b/inspect_db.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import sqlite3
+from pathlib import Path
+
+DB_PATH = Path("storage/cmdhelp.db")
+
+
+def main() -> None:
+ conn = sqlite3.connect(DB_PATH)
+ try:
+ card_count = conn.execute("SELECT COUNT(*) FROM cards").fetchone()[0]
+ emb_count = conn.execute("SELECT COUNT(*) FROM card_embeddings").fetchone()[0]
+
+ print(f"cards: {card_count}")
+ print(f"embeddings: {emb_count}")
+ print()
+
+ rows = conn.execute("""
+ SELECT id, command, safety
+ FROM cards
+ ORDER BY id
+ LIMIT 10
+ """).fetchall()
+
+ for row in rows:
+ print(row)
+ finally:
+ conn.close()
+
+
+if __name__ == "__main__":
+ main()