Update banco.py

banco.py

@@ -6,10 +6,17 @@ Compatível com:
  - Fallback: um único DATABASE_URL vindo de env/Secrets
  - Postgres / MySQL / SQLite (c/ alias de case para load.db no Linux)
  - Garantia de criação do diretório pai do SQLite (evita 'unable to open database file')
+
+Principais melhorias:
+ - 'engine' agora é um PROXY dinâmico (quando houver db_router), refletindo a escolha atual.
+ - Mantém API compatível: engine, Base, SessionLocal, db_info(), get_engine(), etc.
 """
 
 import os
 import shutil
+import importlib
+from typing import Optional
+
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, declarative_base
 from dotenv import load_dotenv
@@ -59,10 +66,8 @@ def _ensure_parent_dir_sqlite(sqlite_url: str) -> str:
     if not sqlite_url or not sqlite_url.startswith("sqlite"):
         return sqlite_url
 
-    # Extrai caminho do arquivo a partir da URL
     # Formatos: sqlite:///relativo.db  |  sqlite:////abs/path/to.db
     path = sqlite_url.replace("sqlite:///", "", 1)
-    # se vier com // (pouco comum), normaliza
     if path.startswith("//"):
         path = path[1:]
     file_path = os.path.abspath(path)
@@ -80,19 +85,24 @@ def _ensure_parent_dir_sqlite(sqlite_url: str) -> str:
 
 
 # =========================================================
-# 3) Suporte a roteador (db_router.py) — opcional
+# 3) Suporte a roteador (db_router.py) — preferencial
 # =========================================================
-# Se existir um roteador, delegamos a criação da engine e da SessionLocal
-# conforme o "banco atual" selecionado (ex.: prod/test/treinamento).
 try:
     from db_router import (
         get_engine as _router_get_engine,
         get_session_factory as _router_get_session_factory,
         SessionLocal as _router_SessionLocal,
+        current_db_choice as _router_current_choice,
+        bank_label as _router_bank_label,
     )
     _HAS_ROUTER = True
 except Exception:
     _HAS_ROUTER = False
+    _router_get_engine = None
+    _router_get_session_factory = None
+    _router_SessionLocal = None
+    _router_current_choice = None
+    _router_bank_label = None
 
 
 # =========================================================
@@ -139,22 +149,17 @@ def _build_fallback_uri() -> str:
 # 5) Engine / SessionLocal
 # =========================================================
 # Observação importante:
-# - Se houver db_router, usamos as fábricas do roteador (engine/sessões por ambiente).
+# - Se houver db_router, usar SEMPRE as fábricas do roteador (engine/sessões por ambiente).
 # - Caso contrário, criamos uma engine única a partir de DATABASE_URL/DB_* ou SQLite.
 
 if _HAS_ROUTER:
     # =========== Com roteador ===========
     def get_engine():
-        """
-        Retorna a engine do banco ATUAL (prod/test/treinamento),
-        conforme implementado pelo seu db_router.get_engine().
-        """
+        """Retorna a engine do banco ATUAL (prod/test/treinamento)."""
         return _router_get_engine()
 
-    def _session_factory():
-        """
-        Fábrica de Session para o banco ATUAL (vinda do roteador).
-        """
+    def get_session_factory():
+        """Retorna um sessionmaker para o banco ATUAL (do roteador)."""
         return _router_get_session_factory()
 
     # SessionLocal fornecida pelo roteador (respeita o banco atual)
@@ -177,33 +182,47 @@ else:
 
     def get_engine():
         """
-        Engine fixa do fallback. Se quiser trocar de banco em runtime sem roteador,
-        você precisará recriar a engine manualmente (ou adotar db_router).
+        Engine fixa do fallback. Para trocar de banco em runtime sem roteador,
+        é necessário recriar a engine manualmente.
         """
         return _engine
 
-    _SessionFactory = sessionmaker(
-        autocommit=False,
-        autoflush=False,
-        bind=_engine,
-    )
-
-    def _session_factory():
-        return _SessionFactory
+    def get_session_factory():
+        """Fábrica de sessão fixa (fallback)."""
+        return sessionmaker(autocommit=False, autoflush=False, bind=_engine)
 
     # Para compatibilidade com código que usa SessionLocal()
-    SessionLocal = _SessionFactory
+    SessionLocal = get_session_factory()
 
 
 # =========================================================
-# 6) Expor 'engine' e Base ORM
+# 6) Expor 'engine' DINÂMICA e Base ORM
 # =========================================================
-# Atenção: 'engine' é resolvido no momento da importação.
-# Se você troca de banco após importar 'banco', prefira usar get_engine()
-# e criar a sessão com SessionLocal() para assegurar o banco ATUAL.
-engine = get_engine()
 Base = declarative_base()
 
+class _EngineProxy:
+    """
+    Proxy leve que encaminha atributos/operções para a engine ATUAL.
+    Isso permite manter 'engine' importável sem cristalizar a escolha de banco.
+    """
+    def __getattr__(self, name):
+        return getattr(get_engine(), name)
+
+    def __repr__(self):
+        eng = get_engine()
+        try:
+            url = str(eng.url)
+        except Exception:
+            url = "(url indisponível)"
+        if _HAS_ROUTER and _router_current_choice:
+            ch = _router_current_choice()
+            lbl = _router_bank_label(ch) if _router_bank_label else ch
+            return f"<EngineProxy choice={ch} label={lbl} url={url}>"
+        return f"<EngineProxy url={url}>"
+
+# 'engine' exportado como proxy (dinâmico)
+engine = _EngineProxy()
+
 
 # =========================================================
 # 7) Utilitários (opcionais)
@@ -211,13 +230,12 @@ Base = declarative_base()
 def init_schema():
     """
     Cria/atualiza as tabelas no banco ATUAL.
-    • Com roteador: aplica no banco escolhido (Produção/Teste/...).
+    • Com roteador: aplica no banco escolhido (Produção/Teste/Treinamento).
     • Sem roteador: aplica no DATABASE_URL padrão.
 
     Use em DEV/TESTE; em produção, prefira migrações (ex.: Alembic).
     """
     # Importa 'models' de forma tardia/segura para registrar mapeamentos
-    import importlib
     try:
         importlib.import_module("models")
     except ModuleNotFoundError:
@@ -240,7 +258,15 @@ def db_info() -> dict:
             url = DATABASE_URL  # type: ignore[name-defined]
         except Exception:
             url = "(não disponível)"
-    return {
-        "url": url,
-        "using_router": _HAS_ROUTER,
-    }
+
+    info = {"url": url, "using_router": _HAS_ROUTER}
+
+    if _HAS_ROUTER and _router_current_choice:
+        ch = _router_current_choice()
+        info["choice"] = ch
+        try:
+            info["label"] = _router_bank_label(ch)
+        except Exception:
+            info["label"] = ch
+
+    return info