Update db_router.py

db_router.py

@@ -6,17 +6,15 @@ db_router.py — Roteia Engine/SessionLocal para:
  - 'treinamento'  → Load_treinamento.db
 
 Mantém a escolha do usuário em st.session_state (quando disponível).
-Exibe uma API simples esperada pelo banco.py:
-
-• set_db_choice("prod"|"test"|"treinamento")  → define o banco ativo
-• current_db_choice()                         → retorna 'prod' | 'test' | 'treinamento'
-• get_engine()                                → engine do banco ativo (SQLite)
-• get_session_factory()                       → sessionmaker do banco ativo
-• SessionLocal()                              → sessão no banco ativo
-
-Compatibilidade:
-- Se Streamlit não estiver disponível (execução fora do app), usa cache global interno (fallback).
-- Pode ler URLs do .env/Secrets (opcional) para apontar bancos em outro local/driver.
+Expõe:
+• set_db_choice("prod"|"test"|"treinamento")
+• current_db_choice()
+• get_engine()
+• get_session_factory()
+• SessionLocal()
+
+Inclui garantia de criação do diretório pai do SQLite.
+Compatível com execução fora do Streamlit (fallback em estado global).
 """
 
 from __future__ import annotations
@@ -44,12 +42,7 @@ PROD_DB_NAME         = "Load.db"
 TEST_DB_NAME         = "Load_teste.db"
 TREINAMENTO_DB_NAME  = "Load_treinamento.db"
 
-# ============================
-# (Opcional) uso de .env/Secrets
-# ============================
-# Se quiser apontar para outros caminhos/URIs via Secrets do Spaces:
-#   DB1_PROD_URL, DB2_TEST_URL, DB3_TREINAMENTO_URL
-# Se não definir, usamos o SQLite no diretório do projeto.
+# (Opcional) uso de .env/Secrets para apontar outras URIs (Postgres/MySQL/SQLite abs.)
 DB1_PROD_URL = os.getenv("DB1_PROD_URL", f"sqlite:///{os.path.join(BASE_DIR, PROD_DB_NAME)}")
 DB2_TEST_URL = os.getenv("DB2_TEST_URL", f"sqlite:///{os.path.join(BASE_DIR, TEST_DB_NAME)}")
 DB3_TREINAMENTO_URL = os.getenv("DB3_TREINAMENTO_URL", f"sqlite:///{os.path.join(BASE_DIR, TREINAMENTO_DB_NAME)}")
@@ -67,34 +60,28 @@ DB_LABELS: Dict[str, str] = {
 }
 
 # ============================
-# (Opcional) normalização de case para SQLite
+# Garantir diretório do SQLite
 # ============================
-# Se você quiser garantir que exista também uma cópia "load.db" (minúsculo)
-# a partir de "Load.db" (maiúsculo) — útil se outros módulos esperarem minúsculas:
-_ENABLE_SQLITE_LOWERCASE_ALIAS = False  # mude para True se quiser habilitar
-
-def _ensure_sqlite_lowercase_alias():
-    import shutil
-    # Para cada caminho SQLite "Load*.db", garantir um "load*.db"
-    for key, url in DB_URLS.items():
-        if not url.startswith("sqlite:///"):
-            continue
-        path = url.replace("sqlite:///", "", 1)
-        # Se usar caminho relativo da BASE_DIR, resolva:
-        if not os.path.isabs(path):
-            path = os.path.join(BASE_DIR, path)
-        dirn, fname = os.path.split(path)
-        lower_name = fname.lower()
-        lower_path = os.path.join(dirn, lower_name)
-        if fname != lower_name and os.path.exists(path) and not os.path.exists(lower_path):
-            try:
-                shutil.copy(path, lower_path)
-            except Exception:
-                # Se falhar, ignora (não é crítico)
-                pass
-
-if _ENABLE_SQLITE_LOWERCASE_ALIAS:
-    _ensure_sqlite_lowercase_alias()
+def _ensure_parent_dir_sqlite(url: str) -> str:
+    """
+    Garante que a pasta do arquivo SQLite exista. Se não conseguir,
+    direciona para ~/.ioirun/<arquivo>.db (gravável no Spaces).
+    """
+    if not url or not url.startswith("sqlite"):
+        return url
+    path = url.replace("sqlite:///", "", 1)
+    if path.startswith("//"):
+        path = path[1:]
+    file_path = os.path.abspath(path)
+    parent = os.path.dirname(file_path)
+    try:
+        os.makedirs(parent, exist_ok=True)
+        return url
+    except Exception:
+        home_dir = os.path.join(os.path.expanduser("~"), ".ioirun")
+        os.makedirs(home_dir, exist_ok=True)
+        alt = os.path.join(home_dir, os.path.basename(file_path))
+        return f"sqlite:///{alt}"
 
 # ============================
 # Helpers de UI
@@ -188,6 +175,8 @@ def get_engine():
         return cached
 
     url = _url_for_choice(choice)
+    url = _ensure_parent_dir_sqlite(url)  # ⬅️ garante diretório pai
+
     eng = create_engine(url, **_engine_args_for_url(url))
     setattr(eng, "__db_choice__", choice)
     _session_set(SESSION_DB_ENGINE_KEY, eng)