Judge / backend /scripts /migrate_corpus_version.py
GonzaViss's picture
chore(corpus): add parser and migration scripts for v1.3.0
daa6630
Raw
History Blame Contribute Delete
691 Bytes
from dotenv import load_dotenv
load_dotenv()
import os, psycopg2
conn = psycopg2.connect(os.getenv("DATABASE_URL"))
cur = conn.cursor()
cur.execute("SELECT COUNT(*) FROM corpus_chunks WHERE corpus_version = 'v1.2.0'")
count = cur.fetchone()[0]
print(f"Chunks en v1.2.0: {count}")
cur.execute("UPDATE corpus_chunks SET corpus_version = 'v1.3.0' WHERE corpus_version = 'v1.2.0'")
print(f"Rows actualizados: {cur.rowcount}")
conn.commit()
cur.execute("SELECT source_type, COUNT(*) FROM corpus_chunks WHERE corpus_version = 'v1.3.0' GROUP BY source_type ORDER BY source_type")
print("Corpus v1.3.0 final:")
for row in cur.fetchall():
print(f" {row[0]}: {row[1]} chunks")
conn.close()