database.py

import os
from dotenv import load_dotenv
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, DeclarativeBase

# 1. On charge l'URL depuis le .env
load_dotenv()
db_url = os.getenv("DATABASE_URL")

# SECURITÉ : On vérifie explicitement que l'URL n'est pas vide
# Cela rassure VS Code et empêche un crash au démarrage
if db_url is None:
    raise ImportError("❌ La variable DATABASE_URL est manquante dans le fichier .env")

SQLALCHEMY_DATABASE_URL: str = db_url

# 2. On crée le moteur de connexion
engine = create_engine(SQLALCHEMY_DATABASE_URL)

# 3. On prépare la fabrique de sessions
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

# 4. Version moderne (SQLAlchemy 2.0+) de la Base
class Base(DeclarativeBase):
    pass

# 5. Injection de dépendance pour FastAPI
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()