82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""1회성 마이그레이션 — agent_office.db.tarot_readings → tarot.db.tarot_readings.
|
|
|
|
멱등성: 이미 존재하는 id는 SKIP.
|
|
|
|
실행:
|
|
docker exec agent-office python /app/scripts/migrate_tarot_to_lab.py
|
|
|
|
또는 호스트에서 직접:
|
|
AGENT_OFFICE_DB=/path/to/agent_office.db TAROT_DB=/path/to/tarot.db \\
|
|
python scripts/migrate_tarot_to_lab.py
|
|
"""
|
|
import os
|
|
import sqlite3
|
|
import sys
|
|
|
|
|
|
SRC = os.getenv("AGENT_OFFICE_DB", "/app/data/agent_office.db")
|
|
DST = os.getenv("TAROT_DB", "/app/data/tarot.db")
|
|
|
|
|
|
SCHEMA = """
|
|
CREATE TABLE IF NOT EXISTS tarot_readings (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
created_at TEXT NOT NULL DEFAULT (strftime('%Y-%m-%dT%H:%M:%fZ','now')),
|
|
spread_type TEXT NOT NULL,
|
|
category TEXT,
|
|
question TEXT,
|
|
cards TEXT NOT NULL,
|
|
interpretation_json TEXT,
|
|
summary TEXT,
|
|
model TEXT,
|
|
tokens_in INTEGER,
|
|
tokens_out INTEGER,
|
|
cost_usd REAL,
|
|
confidence TEXT,
|
|
favorite INTEGER NOT NULL DEFAULT 0,
|
|
note TEXT
|
|
);
|
|
"""
|
|
|
|
|
|
def migrate() -> int:
|
|
"""이관된 row 수 반환."""
|
|
src = sqlite3.connect(SRC)
|
|
src.row_factory = sqlite3.Row
|
|
dst = sqlite3.connect(DST)
|
|
dst.execute("PRAGMA journal_mode=WAL")
|
|
dst.executescript(SCHEMA)
|
|
|
|
rows = src.execute("SELECT * FROM tarot_readings").fetchall()
|
|
if not rows:
|
|
src.close(); dst.close()
|
|
return 0
|
|
|
|
all_cols = list(rows[0].keys())
|
|
|
|
moved = 0
|
|
for r in rows:
|
|
exists = dst.execute("SELECT 1 FROM tarot_readings WHERE id=?", (r["id"],)).fetchone()
|
|
if exists:
|
|
continue
|
|
# NULL 값은 INSERT에서 제외 → 목적지 스키마의 DEFAULT가 적용되도록 함
|
|
# (예: created_at이 NULL이면 strftime() 기본값 사용)
|
|
cols = [c for c in all_cols if r[c] is not None]
|
|
placeholders = ",".join("?" * len(cols))
|
|
cols_str = ",".join(cols)
|
|
dst.execute(
|
|
f"INSERT INTO tarot_readings ({cols_str}) VALUES ({placeholders})",
|
|
tuple(r[c] for c in cols),
|
|
)
|
|
moved += 1
|
|
dst.commit()
|
|
src.close(); dst.close()
|
|
return moved
|
|
|
|
|
|
if __name__ == "__main__":
|
|
moved = migrate()
|
|
total = sqlite3.connect(SRC).execute("SELECT COUNT(*) FROM tarot_readings").fetchone()[0]
|
|
print(f"migrated {moved} / {total} rows from {SRC} to {DST}")
|
|
sys.exit(0)
|