Update modules/database/database_init.py

modules/database/database_init.py

@@ -1,31 +1,116 @@
-#modules/database/database_init.py
+# 1. modules/database/database_init.py
+
 import os
-from azure.cosmos import CosmosClient
 import logging
+from azure.cosmos import CosmosClient
+from pymongo import MongoClient
+import certifi
 
+logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
-# Conexión global
-db_client = None
+# Variables globales para Cosmos DB SQL API
+cosmos_client = None
+user_database = None
+user_container = None
+application_requests_container = None
+user_feedback_container = None
+user_sessions_container = None
 
-def initialize_database_connections():
-    """Inicializa la conexión a Cosmos DB."""
-    global db_client
+# Variables globales para Cosmos DB MongoDB API
+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:
-        endpoint = os.getenv("COSMOS_ENDPOINT")
-        key = os.getenv("COSMOS_KEY")
-        if not endpoint or not key:
-            raise ValueError("Las variables de entorno COSMOS_ENDPOINT y COSMOS_KEY no están configuradas.")
+        container_props = container.read()
+        partition_key_paths = container_props['partitionKey']['paths']
+        logger.info(f"Container: {container.id}, Partition Key Paths: {partition_key_paths}")
+        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"""
+    global user_container, user_sessions_container
+    
+    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 la conexión a Cosmos DB SQL API"""
+    global cosmos_client, user_database, user_container, user_sessions_container
+
+    try:
+        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")
         
-        db_client = CosmosClient(endpoint, key)
-        logger.info("Conexión a Cosmos DB inicializada correctamente.")
+        if not cosmos_endpoint or not cosmos_key:
+            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")
+        
+        # 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 MongoDB"""
+    global mongo_client, mongo_db
+    try:
+        connection_string = os.getenv("MONGODB_CONNECTION_STRING")
+        if not connection_string:
+            raise ValueError("MONGODB_CONNECTION_STRING debe estar configurado")
+
+        mongo_client = MongoClient(
+            connection_string,
+            tls=True,
+            tlsCAFile=certifi.where(),
+            retryWrites=False,
+            serverSelectionTimeoutMS=5000,
+            connectTimeoutMS=10000,
+            socketTimeoutMS=10000
+        )
+        
+        mongo_db = mongo_client['aideatext_db']
         return True
     except Exception as e:
-        logger.error(f"Error inicializando conexión a Cosmos DB: {e}")
+        logger.error(f"Error conectando a MongoDB: {str(e)}")
         return False
 
-def get_database_client(database_name):
-    """Devuelve el cliente de la base de datos especificada."""
-    if not db_client:
-        raise ValueError("La conexión a la base de datos no está inicializada.")
-    return db_client.get_database_client(database_name)
+###################################################################
+def initialize_database_connections():
+    """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:
+        initialize_mongodb_connection()
+    return mongo_db