# 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 )