Update modules/database/database_init.py

modules/database/database_init.py

@@ -35,6 +35,43 @@ def verify_container_partition_key(container, expected_path):
 
 ###################################################################
 
+def get_container(container_name):
+    """Obtiene un contenedor específico"""
+    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,
+        "application_requests": application_requests_container
+    }
+    
+    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 get_container(container_name):
     """Obtiene un contenedor específico"""
     if not initialize_cosmos_sql_connection():
@@ -64,8 +101,85 @@ def get_container(container_name):
         return None
         
     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, application_requests_container
+
+    try:
+        # 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 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(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_cosmos_sql_connection():
     """Inicializa la conexión a Cosmos DB SQL API"""
     global cosmos_client, user_database, user_container, user_sessions_container
@@ -95,6 +209,7 @@ def initialize_cosmos_sql_connection():
     except Exception as e:
         logger.error(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
         return False
+'''        
         
 ###################################################################
 def initialize_mongodb_connection():