# database.py
import os
import MySQLdb
from dotenv import load_dotenv

load_dotenv()

def get_db_connection():
    conn = None
    try:
        conn = MySQLdb.connect(
            user=os.getenv("DB_USER"),
            passwd=os.getenv("DB_PASS"),
            host=os.getenv("DB_HOST"),
            db=os.getenv("DB_NAME"),
            charset='utf8mb4'
        )
        yield conn
    except MySQLdb.Error as e:
        print(f"Erreur de base de données : {e}")
    finally:
        if conn:
            conn.close()