Spaces:
Sleeping
Sleeping
Update modules/database/database_init.py
Browse files
modules/database/database_init.py
CHANGED
|
@@ -73,16 +73,16 @@ def get_container(container_name):
|
|
| 73 |
|
| 74 |
def initialize_cosmos_sql_connection():
|
| 75 |
"""Inicializa la conexi贸n a Cosmos DB SQL API"""
|
| 76 |
-
global cosmos_client, user_database, user_container, user_sessions_container, application_requests_container
|
| 77 |
|
| 78 |
try:
|
| 79 |
# Verificar conexi贸n existente
|
| 80 |
if all([
|
| 81 |
cosmos_client,
|
| 82 |
-
user_database,
|
| 83 |
user_container,
|
| 84 |
user_sessions_container,
|
| 85 |
-
application_requests_container,
|
| 86 |
user_feedback_container
|
| 87 |
]):
|
| 88 |
logger.debug("Todas las conexiones ya est谩n inicializadas")
|
|
@@ -99,7 +99,7 @@ def initialize_cosmos_sql_connection():
|
|
| 99 |
cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
|
| 100 |
user_database = cosmos_client.get_database_client("user_database")
|
| 101 |
|
| 102 |
-
# Inicializar
|
| 103 |
try:
|
| 104 |
user_container = user_database.get_container_client("users")
|
| 105 |
logger.info("Contenedor 'users' inicializado correctamente")
|
|
@@ -121,25 +121,23 @@ def initialize_cosmos_sql_connection():
|
|
| 121 |
logger.error(f"Error inicializando contenedor 'application_requests': {str(e)}")
|
| 122 |
application_requests_container = None
|
| 123 |
|
| 124 |
-
# A帽adir inicializaci贸n del contenedor user_feedback
|
| 125 |
try:
|
| 126 |
user_feedback_container = user_database.get_container_client("user_feedback")
|
| 127 |
logger.info("Contenedor 'user_feedback' inicializado correctamente")
|
| 128 |
except Exception as e:
|
| 129 |
logger.error(f"Error inicializando contenedor 'user_feedback': {str(e)}")
|
| 130 |
user_feedback_container = None
|
| 131 |
-
|
| 132 |
# Verificar el estado de los contenedores
|
| 133 |
containers_status = {
|
| 134 |
'users': user_container is not None,
|
| 135 |
'users_sessions': user_sessions_container is not None,
|
| 136 |
'application_requests': application_requests_container is not None,
|
| 137 |
-
'user_feedback': user_feedback_container is not None
|
| 138 |
}
|
| 139 |
|
| 140 |
logger.info(f"Estado de los contenedores: {containers_status}")
|
| 141 |
|
| 142 |
-
# Solo retornar True si todos los contenedores se inicializaron correctamente
|
| 143 |
if all(containers_status.values()):
|
| 144 |
logger.info("Todos los contenedores inicializados correctamente")
|
| 145 |
return True
|
|
@@ -150,7 +148,8 @@ def initialize_cosmos_sql_connection():
|
|
| 150 |
except Exception as e:
|
| 151 |
logger.error(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
|
| 152 |
return False
|
| 153 |
-
|
|
|
|
| 154 |
###################################################################
|
| 155 |
def initialize_mongodb_connection():
|
| 156 |
"""Inicializa la conexi贸n a MongoDB"""
|
|
|
|
| 73 |
|
| 74 |
def initialize_cosmos_sql_connection():
|
| 75 |
"""Inicializa la conexi贸n a Cosmos DB SQL API"""
|
| 76 |
+
global cosmos_client, user_database, user_container, user_sessions_container, application_requests_container, user_feedback_container # A帽adida aqu铆 user_feedback_container
|
| 77 |
|
| 78 |
try:
|
| 79 |
# Verificar conexi贸n existente
|
| 80 |
if all([
|
| 81 |
cosmos_client,
|
| 82 |
+
user_database,
|
| 83 |
user_container,
|
| 84 |
user_sessions_container,
|
| 85 |
+
application_requests_container,
|
| 86 |
user_feedback_container
|
| 87 |
]):
|
| 88 |
logger.debug("Todas las conexiones ya est谩n inicializadas")
|
|
|
|
| 99 |
cosmos_client = CosmosClient(cosmos_endpoint, cosmos_key)
|
| 100 |
user_database = cosmos_client.get_database_client("user_database")
|
| 101 |
|
| 102 |
+
# Inicializar contenedores
|
| 103 |
try:
|
| 104 |
user_container = user_database.get_container_client("users")
|
| 105 |
logger.info("Contenedor 'users' inicializado correctamente")
|
|
|
|
| 121 |
logger.error(f"Error inicializando contenedor 'application_requests': {str(e)}")
|
| 122 |
application_requests_container = None
|
| 123 |
|
|
|
|
| 124 |
try:
|
| 125 |
user_feedback_container = user_database.get_container_client("user_feedback")
|
| 126 |
logger.info("Contenedor 'user_feedback' inicializado correctamente")
|
| 127 |
except Exception as e:
|
| 128 |
logger.error(f"Error inicializando contenedor 'user_feedback': {str(e)}")
|
| 129 |
user_feedback_container = None
|
| 130 |
+
|
| 131 |
# Verificar el estado de los contenedores
|
| 132 |
containers_status = {
|
| 133 |
'users': user_container is not None,
|
| 134 |
'users_sessions': user_sessions_container is not None,
|
| 135 |
'application_requests': application_requests_container is not None,
|
| 136 |
+
'user_feedback': user_feedback_container is not None
|
| 137 |
}
|
| 138 |
|
| 139 |
logger.info(f"Estado de los contenedores: {containers_status}")
|
| 140 |
|
|
|
|
| 141 |
if all(containers_status.values()):
|
| 142 |
logger.info("Todos los contenedores inicializados correctamente")
|
| 143 |
return True
|
|
|
|
| 148 |
except Exception as e:
|
| 149 |
logger.error(f"Error al conectar con Cosmos DB SQL API: {str(e)}")
|
| 150 |
return False
|
| 151 |
+
|
| 152 |
+
|
| 153 |
###################################################################
|
| 154 |
def initialize_mongodb_connection():
|
| 155 |
"""Inicializa la conexi贸n a MongoDB"""
|