notebookVb

run_migration.py

@@ -0,0 +1,43 @@
+#!/usr/bin/env python3
+"""One-shot migration: add thumbnail_path column + partial index."""
+import os
+import sys
+from pathlib import Path
+
+import psycopg2
+from dotenv import load_dotenv
+
+load_dotenv(Path(__file__).parent / ".env")
+
+conn = psycopg2.connect(
+    host=os.getenv("DB_HOST"),
+    port=int(os.getenv("DB_PORT", 5432)),
+    dbname=os.getenv("DB_NAME"),
+    user=os.getenv("DB_USER"),
+    password=os.getenv("DB_PASSWORD"),
+    connect_timeout=10,
+)
+conn.autocommit = True
+cur = conn.cursor()
+
+print("Step 1: ALTER TABLE ...", flush=True)
+cur.execute("ALTER TABLE photos ADD COLUMN IF NOT EXISTS thumbnail_path VARCHAR(2000)")
+print("  Done.", flush=True)
+
+print("Step 2: CREATE INDEX CONCURRENTLY ...", flush=True)
+cur.execute(
+    "CREATE INDEX CONCURRENTLY IF NOT EXISTS idx_photos_thumbnail_path_null "
+    "ON photos (id) WHERE thumbnail_path IS NULL"
+)
+print("  Done.", flush=True)
+
+cur.execute(
+    "SELECT column_name FROM information_schema.columns "
+    "WHERE table_name='photos' AND column_name='thumbnail_path'"
+)
+row = cur.fetchone()
+print(f"Verified column exists: {row is not None}", flush=True)
+
+cur.close()
+conn.close()
+print("Migration complete.", flush=True)