44 lines
1.1 KiB
Python
44 lines
1.1 KiB
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
|
|
# )
|
|
|
|
|
|
import mysql.connector
|
|
import os
|
|
|
|
|
|
class DBConfig:
|
|
@staticmethod
|
|
def get_db_connection():
|
|
"""
|
|
Create and return a MySQL database connection
|
|
using environment variables.
|
|
"""
|
|
|
|
return mysql.connector.connect(
|
|
host=os.getenv("DB_HOST", "db"), # Docker service name
|
|
port=int(os.getenv("DB_PORT", 3306)),
|
|
user=os.getenv("DB_USER", "root"),
|
|
password=os.getenv("DB_PASSWORD", "root"),
|
|
database=os.getenv("DB_NAME", "test_income_taxdb"),
|
|
autocommit=False
|
|
)
|