import subprocess
import sys
import os

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

conn = pymysql.connect(
    host=os.environ.get('MYSQL_HOST', 'localhost'),
    port=int(os.environ.get('MYSQL_PORT', 3306)),
    user=os.environ.get('MYSQL_USER', 'root'),
    password=os.environ.get('MYSQL_PASSWORD', ''),
    database=os.environ.get('MYSQL_DATABASE', 'admanager'),
    charset='utf8mb4',
    connect_timeout=15,
)

with conn.cursor() as cur:
    cur.execute("SELECT COUNT(*) FROM portal_users WHERE role = 'admin_tenant'")
    count = cur.fetchone()[0]
    print(f"[v0] Usuários com role incorreto 'admin_tenant': {count}")

    cur.execute("UPDATE portal_users SET role = 'tenant_admin' WHERE role = 'admin_tenant'")
    conn.commit()
    print(f"[v0] Corrigidos {cur.rowcount} usuário(s): role 'admin_tenant' -> 'tenant_admin'")

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