diff options
Diffstat (limited to 'inspect_db.py')
| -rw-r--r-- | inspect_db.py | 34 |
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() |