"""
Migração: adiciona a coluna 'aptidao' à tabela marcacoes.
Corre este script UMA VEZ na pasta do projeto:
    python migrar_aptidao.py
"""
import sqlite3
from pathlib import Path

DB_PATH = Path(__file__).parent / "database" / "cofihst.sqlite3"

if not DB_PATH.exists():
    print(f"ERRO: Base de dados não encontrada em {DB_PATH}")
    exit(1)

conn = sqlite3.connect(DB_PATH)
cur  = conn.cursor()

# Verificar se a coluna já existe
cur.execute("PRAGMA table_info(marcacoes)")
colunas = [row[1] for row in cur.fetchall()]

if "aptidao" in colunas:
    print("✅ Coluna 'aptidao' já existe — nada a fazer.")
else:
    cur.execute("""
        ALTER TABLE marcacoes
        ADD COLUMN aptidao TEXT CHECK(aptidao IN ('Apto','Condicionado','Inapto')) DEFAULT NULL
    """)
    conn.commit()
    print("✅ Coluna 'aptidao' adicionada com sucesso!")

conn.close()