import subprocess
subprocess.check_call(['pip', 'install', 'pymysql', '-q'])

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'],
)

with conn.cursor() as cur:
    # Corrige usuários com role vazio que têm tenant_id (são admin_tenant)
    cur.execute(
        "UPDATE portal_users SET role = 'tenant_admin' WHERE role = '' AND tenant_id IS NOT NULL"
    )
    affected = cur.rowcount
    conn.commit()
    print(f'[v0] Registros corrigidos: {affected}')

    cur.execute("SELECT id, name, email, role, tenant_id FROM portal_users")
    for row in cur.fetchall():
        print(f'  id={str(row[0])[:8]}... name={row[1]} email={row[2]} role={row[3]} tenant_id={str(row[4])[:8] if row[4] else "NULL"}')

conn.close()