Initialisation depuis le code en production
This commit is contained in:
23
scripts/db_utils.py
Normal file
23
scripts/db_utils.py
Normal file
@@ -0,0 +1,23 @@
|
||||
import mysql.connector
|
||||
import os
|
||||
import sys
|
||||
import logging
|
||||
|
||||
def connect_to_mysql(db_name: str = None):
|
||||
"""
|
||||
Se connecte à MySQL. Par défaut, utilise la base spécifiée dans DB_NAME,
|
||||
mais peut être redirigé vers une autre base en passant db_name.
|
||||
"""
|
||||
try:
|
||||
sys.path.insert(0, "/myenv/lib/python3.11.2/site-packages") # à ajuster si inutile
|
||||
mydb = mysql.connector.connect(
|
||||
host=os.getenv("DB_HOST"),
|
||||
user=os.getenv("DB_USER"),
|
||||
password=os.getenv("DB_PASSWORD"),
|
||||
database=db_name if db_name else os.getenv("DB_NAME")
|
||||
)
|
||||
logging.info(f"Connexion MySQL à la base {mydb.database} réussie.")
|
||||
return mydb
|
||||
except mysql.connector.Error as err:
|
||||
logging.error(f"Erreur de connexion MySQL : {err}")
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user