Update modules/database/database_init.py

modules/database/database_init.py

@@ -1,4 +1,4 @@
-##modules/database/database_init.py
+# 1. modules/database/database_init.py
 
 import os
 import logging
@@ -15,7 +15,7 @@ user_database = None
 user_container = None
 application_requests_container = None
 user_feedback_container = None
-user_sessions_container = None  # Agregado para sessions
+user_sessions_container = None
 
 # Variables globales para Cosmos DB MongoDB API
 mongo_client = None
@@ -26,20 +26,32 @@ def verify_container_partition_key(container, expected_path):
     try:
         container_props = container.read()
         partition_key_paths = container_props['partitionKey']['paths']
-        
         logger.info(f"Container: {container.id}, Partition Key Paths: {partition_key_paths}")
-        
-        if expected_path not in partition_key_paths:
-            logger.warning(
-                f"Partition key incorrecta en {container.id}. "
-                f"Actual: {partition_key_paths}, Esperada: {expected_path}"
-            )
-            return False
-        return True
+        return expected_path in partition_key_paths
     except Exception as e:
         logger.error(f"Error verificando partition key en {container.id}: {str(e)}")
         return False
 
+def get_container(container_name):
+    """Obtiene un contenedor específico"""
+    containers = {
+        "users": (user_container, "/id"),
+        "application_requests": (application_requests_container, "/email"),
+        "user_feedback": (user_feedback_container, "/username"),
+        "users_sessions": (user_sessions_container, "/username")
+    }
+    
+    if container_name not in containers:
+        logger.error(f"Contenedor no válido: {container_name}")
+        return None
+    
+    container, partition_path = containers[container_name]
+    if container is None:
+        initialize_cosmos_sql_connection()
+        container, _ = containers[container_name]
+    
+    return container
+
 def initialize_cosmos_sql_connection():
     """Inicializa y verifica la conexión a Cosmos DB SQL API"""
     global cosmos_client, user_database, user_container, \
@@ -53,11 +65,10 @@ def initialize_cosmos_sql_connection():
         if not cosmos_endpoint or not cosmos_key:
             raise ValueError("COSMOS_ENDPOINT and COSMOS_KEY environment variables must be set")
 
-        # Inicializar cliente y obtener contenedores
         cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
         user_database = cosmos_client.get_database_client("user_database")
         
-        # Configuración de contenedores y sus partition keys esperadas
+        # Configuración de contenedores
         containers_config = {
             "users": ("/id", user_container),
             "application_requests": ("/email", application_requests_container),
@@ -65,28 +76,21 @@ def initialize_cosmos_sql_connection():
             "users_sessions": ("/username", user_sessions_container)
         }
 
-        # Inicializar y verificar cada contenedor
+        # Inicializar contenedores
         for container_id, (expected_path, container_ref) in containers_config.items():
             container = user_database.get_container_client(container_id)
             if not verify_container_partition_key(container, expected_path):
                 logger.error(f"Verificación de partition key falló para {container_id}")
                 return False
                 
-            # Actualizar la referencia global
-            if container_id == "users":
-                user_container = container
-            elif container_id == "application_requests":
-                application_requests_container = container
-            elif container_id == "user_feedback":
-                user_feedback_container = container
-            elif container_id == "users_sessions":
-                user_sessions_container = container
-
-        logger.info("Conexión a Cosmos DB SQL API exitosa y particiones verificadas")
+            # Actualizar referencia global
+            globals()[f"{container_id}_container"] = container
+
+        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)}", exc_info=True)
+        logger.error(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
         return False
 
 def initialize_mongodb_connection():
@@ -107,39 +111,21 @@ def initialize_mongodb_connection():
             socketTimeoutMS=10000
         )
         
-        # Verificar conexión
         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)}", exc_info=True)
+        logger.error(f"Error al conectar con Cosmos DB MongoDB API: {str(e)}")
         return False
 
 def initialize_database_connections():
     """Inicializa todas las conexiones a bases de datos"""
-    sql_success = initialize_cosmos_sql_connection()
-    mongodb_success = initialize_mongodb_connection()
-    
-    if sql_success and mongodb_success:
-        logger.info("Todas las conexiones a bases de datos inicializadas correctamente")
-    else:
-        logger.error("Falló la inicialización de una o más conexiones a bases de datos")
-        
-    return sql_success and mongodb_success
-
-def get_sql_containers():
-    """Obtiene los contenedores SQL, reinicializando si es necesario"""
-    if any(container is None for container in 
-           [user_container, application_requests_container, 
-            user_feedback_container, user_sessions_container]):
-        initialize_cosmos_sql_connection()
-    return user_container, application_requests_container, user_feedback_container, user_sessions_container
+    return initialize_cosmos_sql_connection() and initialize_mongodb_connection()
 
 def get_mongodb():
-    """Obtiene la conexión MongoDB, reinicializando si es necesario"""
+    """Obtiene la conexión MongoDB"""
     if mongo_db is None:
         initialize_mongodb_connection()
     return mongo_db