init_db.py

from banco import engine, SessionLocal, Base
import models
from models import Usuario
from utils_seguranca import gerar_hash_senha

# Cria todas as tabelas definidas nos modelos
Base.metadata.create_all(bind=engine)

db = SessionLocal()

try:
    usuarios_padrao = [
        ("admin", "admin123", "admin"),
        ("usuario", "usuario123", "usuario"),
        ("consulta", "consulta123", "consulta"),
    ]

    for nome, senha, perfil in usuarios_padrao:
        existe = db.query(Usuario).filter(Usuario.usuario == nome).first()

        if not existe:
            novo = Usuario(
                usuario=nome,
                senha=gerar_hash_senha(senha),
                perfil=perfil,
                ativo=True
            )
            db.add(novo)
            print(f"✅ Usuário '{nome}' criado")
        else:
            print(f"ℹ️ Usuário '{nome}' já existe")

    db.commit()

finally:
    db.close()

print("✅ Banco inicializado com sucesso!")