Create modules/database/database_init.py

modules/database/database_init.py

@@ -0,0 +1,31 @@
+#modules/database/database_init.py
+import os
+from azure.cosmos import CosmosClient
+import logging
+
+logger = logging.getLogger(__name__)
+
+# Conexión global
+db_client = None
+
+def initialize_database_connections():
+    """Inicializa la conexión a Cosmos DB."""
+    global db_client
+    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.")
+        
+        db_client = CosmosClient(endpoint, key)
+        logger.info("Conexión a Cosmos DB inicializada correctamente.")
+        return True
+    except Exception as e:
+        logger.error(f"Error inicializando conexión a Cosmos DB: {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)