Update modules/database/database_init.py

modules/database/database_init.py

@@ -34,31 +34,39 @@ def verify_container_partition_key(container, expected_path):
 
 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")
+    global containers
+    
+    container_config = {
+        "users": "/id",
+        "application_requests": "/email",
+        "user_feedback": "/username",
+        "users_sessions": "/username"
     }
     
-    if container_name not in containers:
+    if container_name not in container_config:
         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]
+    # Si el contenedor ya está en caché, devolverlo
+    if container_name in containers and containers[container_name] is not None:
+        return containers[container_name]
     
-    return container
+    # 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
+        
+    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, user_container, \
-           application_requests_container, user_feedback_container, \
-           user_sessions_container
+    global cosmos_client, user_database, containers
 
     try:
+        # Si ya está inicializado, retornar True
+        if cosmos_client and user_database and containers:
+            return True
+
         cosmos_endpoint = os.environ.get("COSMOS_ENDPOINT")
         cosmos_key = os.environ.get("COSMOS_KEY")
         
@@ -69,22 +77,25 @@ def initialize_cosmos_sql_connection():
         user_database = cosmos_client.get_database_client("user_database")
         
         # Configuración de contenedores
-        containers_config = {
-            "users": ("/id", user_container),
-            "application_requests": ("/email", application_requests_container),
-            "user_feedback": ("/username", user_feedback_container),
-            "users_sessions": ("/username", user_sessions_container)
+        container_config = {
+            "users": "/id",
+            "application_requests": "/email",
+            "user_feedback": "/username",
+            "users_sessions": "/username"
         }
 
         # 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}")
+        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
-                
-            # Actualizar referencia global
-            globals()[f"{container_id}_container"] = container
 
         logger.info("Conexión a Cosmos DB SQL API exitosa")
         return True