77import pymysql
88
99from app .exceptions import ConfigurationException
10+ from app .logging_config import get_logger , setup_logging
1011from app .settings import (
1112 MYSQL_CHARSET ,
1213 MYSQL_DATABASE ,
1718)
1819
1920
21+ setup_logging ()
22+ logger = get_logger (__name__ )
23+
24+
2025class DatabaseConnection :
2126 """MySQL database connection manager."""
2227
@@ -46,6 +51,7 @@ def get_connection(self) -> pymysql.Connection:
4651 Raises:
4752 pymysql.Error: If connection fails
4853 """
54+ logger .debug ("Connecting to MySQL database at %s:%s" , self .host , self .port )
4955 try :
5056 connection = pymysql .connect (
5157 host = self .host ,
@@ -99,6 +105,7 @@ def execute_query(self, query: str, params: Optional[tuple] = None) -> List[Dict
99105 Returns:
100106 List[Dict[str, Any]]: Query results as list of dictionaries
101107 """
108+ logger .debug ("Executing query: %s with params: %s" , query , params )
102109 with self .get_cursor () as cursor :
103110 cursor .execute (query , params or ())
104111 return cursor .fetchall ()
@@ -114,6 +121,7 @@ def execute_update(self, query: str, params: Optional[tuple] = None) -> int:
114121 Returns:
115122 int: Number of affected rows
116123 """
124+ logger .debug ("Executing update: %s with params: %s" , query , params )
117125 with self .get_cursor () as cursor :
118126 cursor .execute (query , params or ())
119127 return cursor .rowcount
@@ -131,7 +139,7 @@ def test_connection(self) -> bool:
131139 return True
132140 # pylint: disable=broad-except
133141 except Exception as e :
134- print ( f "Database connection test failed: { e } " )
142+ logger . error ( "Database connection test failed: %s" , e )
135143 return False
136144
137145
0 commit comments