|
|
|
|
| import os
|
| import logging
|
| from azure.cosmos import CosmosClient
|
| from pymongo import MongoClient
|
| import certifi
|
|
|
| logging.basicConfig(level=logging.DEBUG)
|
| logger = logging.getLogger(__name__)
|
|
|
|
|
| cosmos_client = None
|
| user_database = None
|
| user_container = None
|
| application_requests_container = None
|
| user_feedback_container = None
|
| user_sessions_container = None
|
|
|
|
|
| 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:
|
| 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")
|
|
|
| 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")
|
|
|
|
|
| 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 conectando a MongoDB: {str(e)}")
|
| return False
|
|
|
|
|
| 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 |
