File size: 1,439 Bytes
5e54e45 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
"""
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:
# Check if column already exists
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)
|