Update init_db.py

init_db.py

@@ -1,38 +1,120 @@
-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!")
+# -*- coding: utf-8 -*-
+"""
+init_db.py — Inicializa o banco ATUAL (roteado pelo db_router/banco).
+- Cria/atualiza o schema (Base.metadata.create_all) no banco selecionado.
+- Realiza seed idempotente dos usuários padrão com hash via utils_seguranca.
+- Não usa caminhos fixos; respeita o banco escolhido na UI (prod/test/treinamento).
+
+Uso:
+    - Execute este arquivo diretamente, ou
+    - Importe e chame run() a partir do app, após selecionar o banco.
+"""
+
+from __future__ import annotations
+
+from datetime import datetime
+from sqlalchemy import text
+from sqlalchemy.exc import SQLAlchemyError
+
+# 👇 Usa as fábricas dinâmicas do módulo banco (que delegam ao db_router)
+from banco import get_engine, init_schema, SessionLocal, db_info
+
+# Importa seus modelos para garantir que o metadata conheça todas as tabelas
+import models  # noqa: F401  (garante o registro das classes no metadata)
+
+# Utilitário de hash (bcrypt) já existente no seu projeto
+from utils_seguranca import gerar_hash_senha
+
+
+def _print_header():
+    try:
+        info = db_info()
+        print("===== Application Startup — init_db.run() =====")
+        print(f"📦 Router ativo? {info.get('using_router')}")
+        label = info.get("label") or info.get("choice") or "(desconhecido)"
+        print(f"🏷️ Banco selecionado: {label}")
+        print(f"🔗 URL: {info.get('url')}")
+    except Exception as e:
+        print(f"ℹ️ db_info indisponível: {e}")
+
+
+def _seed_usuarios(db):
+    """
+    Insere usuários padrão se não existirem.
+    Usa gerar_hash_senha(senha_plana) -> hash (bcrypt).
+    """
+    from models import Usuario  # import interno para evitar custos em startup
+
+    usuarios_padrao = [
+        ("admin",    "admin123",    "admin"),
+        ("usuario",  "usuario123",  "usuario"),
+        ("consulta", "consulta123", "consulta"),
+    ]
+
+    for nome, senha_plana, perfil in usuarios_padrao:
+        try:
+            existe = db.query(Usuario).filter(Usuario.usuario == nome).first()
+        except Exception:
+            # Se a tabela ainda não estiver criada (latência em provedor), reforça schema
+            init_schema()
+            existe = db.query(Usuario).filter(Usuario.usuario == nome).first()
+
+        if not existe:
+            try:
+                senha_hash = gerar_hash_senha(senha_plana)
+            except Exception as e:
+                print(f"❌ Falha ao gerar hash para '{nome}': {e}")
+                continue
+
+            novo = Usuario(
+                usuario=nome,
+                senha=senha_hash,  # ⚠️ Armazena HASH, nunca em texto puro
+                perfil=perfil,
+                ativo=True,
+                data_criacao=datetime.utcnow(),
+                nome=nome.capitalize(),
+                email=None,  # ajuste se quiser e-mail padrão
+            )
+            db.add(novo)
+            print(f"✅ Usuário '{nome}' criado")
+        else:
+            print(f"ℹ️ Usuário '{nome}' já existe")
+
+
+def run():
+    """
+    Executa a inicialização do banco atual:
+    - Cria/atualiza schema
+    - Insere usuários padrão (idempotente)
+    """
+    _print_header()
+
+    # 1) Cria/atualiza schema no banco ATUAL
+    try:
+        init_schema()
+    except Exception as e:
+        print(f"❌ Falha ao criar/atualizar schema: {e}")
+        # ainda assim tenta seguir, pois alguns provedores aplicam lazy
+        pass
+
+    # 2) Seed de usuários
+    db = SessionLocal()
+    try:
+        _seed_usuarios(db)
+        db.commit()
+        print("✅ Banco inicializado com sucesso!")
+    except SQLAlchemyError as e:
+        db.rollback()
+        print(f"❌ Erro SQL ao inicializar banco: {e}")
+    except Exception as e:
+        db.rollback()
+        print(f"❌ Erro ao inicializar banco: {e}")
+    finally:
+        try:
+            db.close()
+        except Exception:
+            pass
+
+
+if __name__ == "__main__":
+    run()