hamba-ho's picture
Préparation pour déploiement Hugging Face (Docker + UI)
fab35c8
Raw
History Blame Contribute Delete
674 Bytes
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from .settings import settings
import os
# Créer le dossier data/ si nécessaire (pour SQLite)
db_path = settings.DATABASE_URL.replace("sqlite:///", "")
os.makedirs(os.path.dirname(db_path), exist_ok=True)
# SQLite nécessite connect_args pour le multi-threading FastAPI
connect_args = {"check_same_thread": False} if "sqlite" in settings.DATABASE_URL else {}
engine = create_engine(settings.DATABASE_URL, connect_args=connect_args)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()