import mysql.connector
import os

# Load database credentials from environment variables
DB_HOST = os.getenv("DB_HOST", "localhost")  # Your server's IP
DB_PORT = int(os.getenv("DB_PORT", "3306"))  # Convert port to integer
DB_USER = os.getenv("DB_USER", "emailuser")
DB_PASSWORD = os.getenv("DB_PASSWORD", "miguel33020")
DB_NAME = os.getenv("DB_NAME", "emailassistant")

def connect_db():
    try:
        conn = mysql.connector.connect(
            host=DB_HOST,
            port=DB_PORT,  # Now it's an integer
            user=DB_USER,
            password=DB_PASSWORD,
            database=DB_NAME
        )
        print("✅ Connected to MariaDB successfully!")
        return conn
    except mysql.connector.Error as err:
        print(f"❌ Error: {err}")
        return None

# Test connection
if __name__ == "__main__":
    conn = connect_db()
    if conn:
        conn.close()