import mysql.connector
import os

# Get MySQL credentials from environment variables
MYSQL_HOST = os.getenv("MYSQL_HOST", "127.0.0.1")
MYSQL_USER = os.getenv("MYSQL_USER", "root")
<<<<<<< HEAD
MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD", "root")
=======
MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD", "tiger")
>>>>>>> 1054d1f7e5aff7cb0993ba7919fe1e49ed8217f9
MYSQL_DB = os.getenv("MYSQL_DB", "test")

# Connect to MySQL
def get_db_connection():
    return mysql.connector.connect(
        host=MYSQL_HOST,
        user=MYSQL_USER,
        password=MYSQL_PASSWORD,
        database=MYSQL_DB
    )