IncomeTaxSystem/AppCode/Config.py

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