22 lines
553 B
Python
22 lines
553 B
Python
import mysql.connector
|
|
import os
|
|
|
|
# Database Config
|
|
class DBConfig:
|
|
MYSQL_HOST = os.getenv("DB_HOST")
|
|
MYSQL_USER = os.getenv("DB_USER")
|
|
MYSQL_PASSWORD = os.getenv("DB_PASSWORD")
|
|
MYSQL_DB = os.getenv("DB_NAME")
|
|
|
|
@staticmethod
|
|
def get_db_connection():
|
|
"""
|
|
Returns a MySQL connection object.
|
|
"""
|
|
return mysql.connector.connect(
|
|
host=DBConfig.MYSQL_HOST,
|
|
user=DBConfig.MYSQL_USER,
|
|
password=DBConfig.MYSQL_PASSWORD,
|
|
database=DBConfig.MYSQL_DB
|
|
)
|