Update modules/database/database_init.py

modules/database/database_init.py

@@ -21,6 +21,7 @@ user_sessions_container = None
 mongo_client = None
 mongo_db = None
 
+###################################################################
 def verify_container_partition_key(container, expected_path):
     """Verifica la configuración de partition key de un contenedor"""
     try:
@@ -32,88 +33,62 @@ def verify_container_partition_key(container, expected_path):
         logger.error(f"Error verificando partition key en {container.id}: {str(e)}")
         return False
 
+###################################################################
 def get_container(container_name):
     """Obtiene un contenedor específico"""
-    global containers
+    global user_container, user_sessions_container
     
-    container_config = {
-        "users": "/id",
-        "application_requests": "/email",
-        "user_feedback": "/username",
-        "users_sessions": "/username"
-    }
-    
-    if container_name not in container_config:
-        logger.error(f"Contenedor no válido: {container_name}")
-        return None
-    
-    # Si el contenedor ya está en caché, devolverlo
-    if container_name in containers and containers[container_name] is not None:
-        return containers[container_name]
-    
-    # Si no está en caché, intentar inicializar la conexión
     if not initialize_cosmos_sql_connection():
         logger.error("No se pudo inicializar la conexión")
         return None
-        
+    
+    containers = {
+        "users": user_container,
+        "users_sessions": user_sessions_container
+    }
+    
     return containers.get(container_name)
 
+###################################################################
 def initialize_cosmos_sql_connection():
-    """Inicializa y verifica la conexión a Cosmos DB SQL API"""
-    global cosmos_client, user_database, containers
+    """Inicializa la conexión a Cosmos DB SQL API"""
+    global cosmos_client, user_database, user_container, user_sessions_container
 
     try:
-        # Si ya está inicializado, retornar True
-        if cosmos_client and user_database and containers:
+        if cosmos_client and user_database and user_container and user_sessions_container:
             return True
 
         cosmos_endpoint = os.environ.get("COSMOS_ENDPOINT")
         cosmos_key = os.environ.get("COSMOS_KEY")
         
         if not cosmos_endpoint or not cosmos_key:
-            raise ValueError("COSMOS_ENDPOINT and COSMOS_KEY environment variables must be set")
+            raise ValueError("COSMOS_ENDPOINT y COSMOS_KEY deben estar configurados")
 
         cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
         user_database = cosmos_client.get_database_client("user_database")
         
-        # Configuración de contenedores
-        container_config = {
-            "users": "/id",
-            "application_requests": "/email",
-            "user_feedback": "/username",
-            "users_sessions": "/username"
-        }
-
-        # Inicializar contenedores
-        for container_id, partition_path in container_config.items():
-            try:
-                container = user_database.get_container_client(container_id)
-                if not verify_container_partition_key(container, partition_path):
-                    logger.error(f"Verificación de partition key falló para {container_id}")
-                    return False
-                containers[container_id] = container
-                logger.info(f"Contenedor {container_id} inicializado correctamente")
-            except Exception as e:
-                logger.error(f"Error inicializando contenedor {container_id}: {str(e)}")
-                return False
-
+        # Inicializar contenedores manteniendo la estructura existente
+        user_container = user_database.get_container_client("users")
+        user_sessions_container = user_database.get_container_client("users_sessions")
+        
         logger.info("Conexión a Cosmos DB SQL API exitosa")
         return True
 
     except Exception as e:
         logger.error(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
         return False
-
+        
+###################################################################
 def initialize_mongodb_connection():
-    """Inicializa la conexión a Cosmos DB MongoDB API"""
+    """Inicializa la conexión a MongoDB"""
     global mongo_client, mongo_db
     try:
-        cosmos_mongodb_connection_string = os.getenv("MONGODB_CONNECTION_STRING")
-        if not cosmos_mongodb_connection_string:
-            raise ValueError("MONGODB_CONNECTION_STRING environment variable is not set")
+        connection_string = os.getenv("MONGODB_CONNECTION_STRING")
+        if not connection_string:
+            raise ValueError("MONGODB_CONNECTION_STRING debe estar configurado")
 
         mongo_client = MongoClient(
-            cosmos_mongodb_connection_string,
+            connection_string,
             tls=True,
             tlsCAFile=certifi.where(),
             retryWrites=False,
@@ -122,19 +97,18 @@ def initialize_mongodb_connection():
             socketTimeoutMS=10000
         )
         
-        mongo_client.admin.command('ping')
         mongo_db = mongo_client['aideatext_db']
-        logger.info("Conexión a Cosmos DB MongoDB API exitosa")
         return True
-        
     except Exception as e:
-        logger.error(f"Error al conectar con Cosmos DB MongoDB API: {str(e)}")
+        logger.error(f"Error conectando a MongoDB: {str(e)}")
         return False
 
+###################################################################
 def initialize_database_connections():
-    """Inicializa todas las conexiones a bases de datos"""
+    """Inicializa todas las conexiones"""
     return initialize_cosmos_sql_connection() and initialize_mongodb_connection()
 
+###################################################################
 def get_mongodb():
     """Obtiene la conexión MongoDB"""
     if mongo_db is None: