Update banco.py

banco.py

@@ -5,6 +5,7 @@ Compatível com:
  - Roteamento por ambiente (db_router.py): produção/teste/treinamento
  - 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')
 """
 
 import os
@@ -20,7 +21,7 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 load_dotenv()
 
 # =========================================================
-# 1) Correção de case para SQLite (Load.db → load.db)
+# 1) Correção de case para SQLite (Load.db → load.db) — opcional
 # =========================================================
 def _ensure_sqlite_case_alias() -> str:
     """
@@ -47,10 +48,42 @@ def _ensure_sqlite_case_alias() -> str:
 
 
 # =========================================================
-# 2) Suporte a roteador (db_router.py) — opcional
+# 2) Helper: garantir diretório pai do arquivo SQLite
+# =========================================================
+def _ensure_parent_dir_sqlite(sqlite_url: str) -> str:
+    """
+    Garante que o diretório pai do arquivo SQLite exista.
+    Caso não consiga criar (permissão), cai para ~/.ioirun/<arquivo>.db
+    Retorna a URL (que pode ser ajustada para fallback).
+    """
+    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)
+    parent = os.path.dirname(file_path)
+
+    try:
+        os.makedirs(parent, exist_ok=True)
+        return sqlite_url
+    except Exception:
+        # fallback para HOME (gravável no Spaces)
+        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}"
+
+
+# =========================================================
+# 3) Suporte a roteador (db_router.py) — opcional
 # =========================================================
 # Se existir um roteador, delegamos a criação da engine e da SessionLocal
-# conforme o "banco atual" selecionado (ex.: prod/test).
+# conforme o "banco atual" selecionado (ex.: prod/test/treinamento).
 try:
     from db_router import (
         get_engine as _router_get_engine,
@@ -61,8 +94,9 @@ try:
 except Exception:
     _HAS_ROUTER = False
 
+
 # =========================================================
-# 3) Fallback quando NÃO há roteador: construir a URI
+# 4) Fallback quando NÃO há roteador: construir a URI
 # =========================================================
 def _build_fallback_uri() -> str:
     """
@@ -72,12 +106,13 @@ def _build_fallback_uri() -> str:
       2. Variáveis separadas: DB_DRIVER, DB_HOST, DB_PORT, DB_USER, DB_PASS, DB_NAME
       3. SQLite local em 'load.db'
     """
-    # 3.1 DATABASE_URL completo
+    # 4.1 DATABASE_URL completo
     url = os.getenv("DATABASE_URL")
     if url:
-        return url
+        # Garante diretório pai no caso de sqlite
+        return _ensure_parent_dir_sqlite(url)
 
-    # 3.2 Campos separados
+    # 4.2 Campos separados
     driver = (os.getenv("DB_DRIVER") or "").strip().lower()  # "postgresql", "mysql"
     host = os.getenv("DB_HOST")
     port = os.getenv("DB_PORT")
@@ -95,13 +130,13 @@ def _build_fallback_uri() -> str:
         elif driver.startswith("mysql"):  # MySQL/MariaDB
             return f"mysql+pymysql://{user}:{pwd}@{host}:{port}/{name}"
 
-    # 3.3 SQLite local (fallback)
+    # 4.3 SQLite local (fallback)
     sqlite_path = _ensure_sqlite_case_alias()
-    return f"sqlite:///{sqlite_path}"
+    return _ensure_parent_dir_sqlite(f"sqlite:///{sqlite_path}")
 
 
 # =========================================================
-# 4) Engine / SessionLocal
+# 5) Engine / SessionLocal
 # =========================================================
 # Observação importante:
 # - Se houver db_router, usamos as fábricas do roteador (engine/sessões por ambiente).
@@ -132,7 +167,6 @@ else:
     engine_args = {
         "echo": False,          # defina True para depuração de SQL
         "pool_pre_ping": True,  # valida conexões antes de usar
-        # "future": True,       # opcional (SQLAlchemy 2.x APIs)
     }
 
     if DATABASE_URL.startswith("sqlite"):
@@ -162,7 +196,7 @@ else:
 
 
 # =========================================================
-# 5) Expor 'engine' e Base ORM
+# 6) Expor 'engine' 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()
@@ -172,7 +206,7 @@ Base = declarative_base()
 
 
 # =========================================================
-# 6) Utilitários (opcionais)
+# 7) Utilitários (opcionais)
 # =========================================================
 def init_schema():
     """
@@ -202,7 +236,6 @@ def db_info() -> dict:
     try:
         url = str(eng.url)
     except Exception:
-        # Quando sem roteador, se falhar, tenta DATABASE_URL/DB_* do fallback
         try:
             url = DATABASE_URL  # type: ignore[name-defined]
         except Exception: