Update modules/database/database_init.py

modules/database/database_init.py

@@ -36,48 +36,109 @@ def verify_container_partition_key(container, expected_path):
 ###################################################################
 def get_container(container_name):
     """Obtiene un contenedor específico"""
-    global user_container, user_sessions_container
+    logger.info(f"Solicitando contenedor: {container_name}")
     
     if not initialize_cosmos_sql_connection():
         logger.error("No se pudo inicializar la conexión")
         return None
     
+    # Verificar estado de los contenedores
+    containers_status = {
+        "users": user_container is not None,
+        "users_sessions": user_sessions_container is not None,
+        "application_requests": application_requests_container is not None
+    }
+    
+    logger.info(f"Estado actual de los contenedores: {containers_status}")
+    
+    # Mapear nombres a contenedores
     containers = {
         "users": user_container,
-        "users_sessions": user_sessions_container
+        "users_sessions": user_sessions_container,
+        "application_requests": application_requests_container
     }
     
-    return containers.get(container_name)
-
+    container = containers.get(container_name)
+    
+    if container is None:
+        logger.error(f"Contenedor '{container_name}' no encontrado o no inicializado")
+        logger.error(f"Contenedores disponibles: {[k for k, v in containers_status.items() if v]}")
+        return None
+        
+    logger.info(f"Contenedor '{container_name}' obtenido exitosamente")
+    return container
 ###################################################################
+
 def initialize_cosmos_sql_connection():
     """Inicializa la conexión a Cosmos DB SQL API"""
-    global cosmos_client, user_database, user_container, user_sessions_container
+    global cosmos_client, user_database, user_container, user_sessions_container, application_requests_container
 
     try:
-        if cosmos_client and user_database and user_container and user_sessions_container:
+        # Verificar conexión existente
+        if all([
+            cosmos_client,
+            user_database,
+            user_container,
+            user_sessions_container,
+            application_requests_container
+        ]):
+            logger.debug("Todas las conexiones ya están inicializadas")
             return True
 
+        # Obtener credenciales
         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 y COSMOS_KEY deben estar configurados")
 
+        # Inicializar cliente y base de datos
         cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
         user_database = cosmos_client.get_database_client("user_database")
         
-        # Inicializar contenedores manteniendo la estructura existente
-        user_container = user_database.get_container_client("users")
-        user_sessions_container = user_database.get_container_client("users_sessions")
+        # Inicializar cada contenedor individualmente con verificación
+        try:
+            user_container = user_database.get_container_client("users")
+            logger.info("Contenedor 'users' inicializado correctamente")
+        except Exception as e:
+            logger.error(f"Error inicializando contenedor 'users': {str(e)}")
+            user_container = None
+
+        try:
+            user_sessions_container = user_database.get_container_client("users_sessions")
+            logger.info("Contenedor 'users_sessions' inicializado correctamente")
+        except Exception as e:
+            logger.error(f"Error inicializando contenedor 'users_sessions': {str(e)}")
+            user_sessions_container = None
+
+        try:
+            application_requests_container = user_database.get_container_client("application_requests")
+            logger.info("Contenedor 'application_requests' inicializado correctamente")
+        except Exception as e:
+            logger.error(f"Error inicializando contenedor 'application_requests': {str(e)}")
+            application_requests_container = None
+
+        # Verificar el estado de los contenedores
+        containers_status = {
+            'users': user_container is not None,
+            'users_sessions': user_sessions_container is not None,
+            'application_requests': application_requests_container is not None
+        }
         
-        logger.info("Conexión a Cosmos DB SQL API exitosa")
-        return True
+        logger.info(f"Estado de los contenedores: {containers_status}")
+        
+        # Solo retornar True si todos los contenedores se inicializaron correctamente
+        if all(containers_status.values()):
+            logger.info("Todos los contenedores inicializados correctamente")
+            return True
+        else:
+            logger.error("No se pudieron inicializar todos los contenedores")
+            return False
 
     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 MongoDB"""