24 lines
571 B
Python
24 lines
571 B
Python
# 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',
|
|
cursorclass=MySQLdb.cursors.DictCursor
|
|
)
|
|
yield conn
|
|
except MySQLdb.Error as e:
|
|
print(f"Erreur de base de données : {e}")
|
|
finally:
|
|
if conn:
|
|
conn.close() |