21 lines
525 B
Python
21 lines
525 B
Python
import mysql.connector
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
# Get MySQL credentials from environment variables
|
|
MYSQL_HOST = os.getenv("MYSQL_HOST")
|
|
MYSQL_USER = os.getenv("MYSQL_USER")
|
|
MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD")
|
|
MYSQL_DB = os.getenv("MYSQL_DB")
|
|
|
|
# Connect to MySQL
|
|
def get_db_connection():
|
|
return mysql.connector.connect(
|
|
host=MYSQL_HOST,
|
|
user=MYSQL_USER,
|
|
password=MYSQL_PASSWORD,
|
|
database=MYSQL_DB
|
|
) |