11"""Database Module Initialization"""
22
3+ import logging
4+
5+ logger = logging .getLogger (__name__ )
6+
37try :
48 from .mongo_connection import (
59 enhanced_mongo_manager ,
@@ -20,7 +24,7 @@ def get_mongo_connection():
2024 MONGO_AVAILABLE = True
2125
2226except ImportError as e :
23- print (f"MongoDB connection not available: { e } " )
27+ logger . warning (f"MongoDB connection not available: { e } " )
2428
2529 class MockMongoManager :
2630 def __init__ (self ):
@@ -59,7 +63,7 @@ def get_mongo_connection():
5963 POSTGRES_AVAILABLE = True
6064
6165except ImportError as e :
62- print (f"PostgreSQL not available: { e } " )
66+ logger . warning (f"PostgreSQL not available: { e } " )
6367 postgres_manager = None
6468 get_postgres_session = None
6569 POSTGRES_AVAILABLE = False
@@ -74,7 +78,7 @@ def get_mongo_connection():
7478 SCYLLA_AVAILABLE = True
7579
7680except ImportError as e :
77- print (f"ScyllaDB not available: { e } " )
81+ logger . warning (f"ScyllaDB not available: { e } " )
7882
7983 class MockScyllaManager :
8084 def __init__ (self ):
@@ -100,7 +104,7 @@ def disconnect(self):
100104 REDIS_AVAILABLE = True
101105
102106except ImportError as e :
103- print (f"Redis not available: { e } " )
107+ logger . warning (f"Redis not available: { e } " )
104108 redis_manager = None
105109 get_redis = None
106110 REDIS_AVAILABLE = False
@@ -112,7 +116,7 @@ def get_seed_function():
112116 from app .utils .seed_data import main as seed_main
113117 return seed_main
114118 except ImportError as e :
115- print (f"Seed data module not available: { e } " )
119+ logger . warning (f"Seed data module not available: { e } " )
116120 return None
117121
118122
@@ -125,10 +129,10 @@ def seed_knowledge_base():
125129 try :
126130 return asyncio .run (seed_func ())
127131 except Exception as e :
128- print (f"Seeding failed: { e } " )
132+ logger . error (f"Seeding failed: { e } " )
129133 return False
130134 else :
131- print ("Seeding not available - seed_data module not found" )
135+ logger . warning ("Seeding not available - seed_data module not found" )
132136 return False
133137
134138
0 commit comments