|
|
""" |
|
|
Database Migration Script: Add Transcript Support |
|
|
|
|
|
This script adds the transcript_filename field to the User model |
|
|
to support transcript file uploads. |
|
|
""" |
|
|
|
|
|
from app import app, db |
|
|
from sqlalchemy import text |
|
|
|
|
|
def migrate_transcript_field(): |
|
|
"""Add transcript_filename column to user table if it doesn't exist.""" |
|
|
|
|
|
with app.app_context(): |
|
|
try: |
|
|
|
|
|
with db.engine.connect() as conn: |
|
|
result = conn.execute(text("PRAGMA table_info(user)")) |
|
|
columns = [row[1] for row in result] |
|
|
|
|
|
if 'transcript_filename' not in columns: |
|
|
print("Adding transcript_filename column to user table...") |
|
|
conn.execute(text( |
|
|
"ALTER TABLE user ADD COLUMN transcript_filename VARCHAR(200) DEFAULT ''" |
|
|
)) |
|
|
conn.commit() |
|
|
print("[OK] Successfully added transcript_filename column") |
|
|
else: |
|
|
print("[OK] transcript_filename column already exists") |
|
|
|
|
|
except Exception as e: |
|
|
print(f"[ERROR] Error during migration: {str(e)}") |
|
|
raise |
|
|
|
|
|
if __name__ == '__main__': |
|
|
print("="*60) |
|
|
print("Database Migration: Transcript Support") |
|
|
print("="*60) |
|
|
migrate_transcript_field() |
|
|
print("\n" + "="*60) |
|
|
print("Migration completed successfully!") |
|
|
print("="*60) |
|
|
|
