import sqlite3
import os

def main():
    print("Iniciando migración de base de datos para exámenes de repaso...")
    dbs = ["musical_master.db", "melodix.db"]
    
    for db_file in dbs:
        if os.path.exists(db_file):
            print(f"Migrando base de datos: {db_file}...")
            conn = sqlite3.connect(db_file)
            cursor = conn.cursor()
            try:
                # Agregar la columna
                cursor.execute("ALTER TABLE progresos_plan ADD COLUMN repaso_completado_hasta INTEGER DEFAULT 0")
                conn.commit()
                print("  -> ¡Éxito! Columna 'repaso_completado_hasta' agregada exitosamente.")
            except sqlite3.OperationalError as e:
                # Ignorar si la columna ya existe
                if "duplicate column name" in str(e).lower():
                    print("  -> La columna ya existe en la base de datos. Saltando.")
                else:
                    print(f"  -> Error al agregar columna: {e}")
            finally:
                conn.close()
        else:
            print(f"Base de datos {db_file} no encontrada. Saltando.")

if __name__ == "__main__":
    main()