"""
Add orientation column to playlist table
Run this script to update the database schema
"""
import sys
import os

# Add parent directory to path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.app import app
from app.extensions import db
from sqlalchemy import text

def add_orientation_column():
    """Add orientation column to playlist table."""
    with app.app_context():
        try:
            # Check if column exists
            result = db.session.execute(text("PRAGMA table_info(playlist)"))
            columns = [row[1] for row in result]
            
            if 'orientation' in columns:
                print("✅ Column 'orientation' already exists in playlist table")
                return
            
            # Add the column
            print("Adding 'orientation' column to playlist table...")
            db.session.execute(text("""
                ALTER TABLE playlist 
                ADD COLUMN orientation VARCHAR(20) DEFAULT 'Landscape' NOT NULL
            """))
            db.session.commit()
            
            print("✅ Successfully added 'orientation' column to playlist table")
            print("   Default value: 'Landscape'")
            
        except Exception as e:
            print(f"❌ Error adding column: {str(e)}")
            db.session.rollback()
            raise

if __name__ == '__main__':
    add_orientation_column()