import subprocess, sys
subprocess.check_call([sys.executable, "-m", "pip", "install", "pymysql", "--quiet"])

import pymysql, os

conn = pymysql.connect(
    host=os.environ['MYSQL_HOST'],
    port=int(os.environ.get('MYSQL_PORT', 3306)),
    user=os.environ['MYSQL_USER'],
    password=os.environ['MYSQL_PASSWORD'],
    database=os.environ['MYSQL_DATABASE'],
    charset='utf8mb4',
)

with conn.cursor() as cur:
    # Adiciona coluna allow_global_admin_access se não existir
    cur.execute("""
        ALTER TABLE tenants
        ADD COLUMN IF NOT EXISTS allow_global_admin_access TINYINT(1) NOT NULL DEFAULT 1
        COMMENT 'Se 1, o admin global (ConsultarTI) pode ver e gerenciar dados desta empresa'
    """)
    conn.commit()
    print("[v0] Coluna allow_global_admin_access adicionada (ou já existia).")

    # Confirma
    cur.execute("DESCRIBE tenants")
    cols = [r[0] for r in cur.fetchall()]
    print(f"[v0] Colunas em tenants: {cols}")

conn.close()
print("[v0] Concluido.")
