"""
Migration script to add avatar_url field to users table
"""
import sqlite3
import os

def migrate_avatar_field():
    """Add avatar_url column to users table"""
    db_path = "vfx_project_management.db"
    
    if not os.path.exists(db_path):
        print(f"Database file {db_path} not found!")
        return
    
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    
    try:
        # Check if column already exists
        cursor.execute("PRAGMA table_info(users)")
        columns = [column[1] for column in cursor.fetchall()]
        
        if 'avatar_url' in columns:
            print("✓ avatar_url column already exists in users table")
        else:
            # Add avatar_url column
            cursor.execute("ALTER TABLE users ADD COLUMN avatar_url TEXT")
            conn.commit()
            print("✓ Added avatar_url column to users table")
        
        print("\nMigration completed successfully!")
        
    except Exception as e:
        print(f"Error during migration: {e}")
        conn.rollback()
    finally:
        conn.close()

if __name__ == "__main__":
    migrate_avatar_field()