AIdeaText commited on
Commit
5780733
1 Parent(s): 8f2cd6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -35
app.py CHANGED
@@ -49,42 +49,35 @@ except Exception as e:
49
  #mongo_connection_string = os.environ.get("MONGODB_CONNECTION_STRING")
50
  cosmos_mongodb_connection_string = os.getenv("MONGODB_CONNECTION_STRING")
51
 
52
- if not cosmos_mongodb_connection_string:
53
- raise ValueError("La variable de entorno MONGODB_CONNECTION_STRING debe estar configurada")
 
 
54
 
55
- try:
56
- #Crear el cliente de MongoDB
57
- client = MongoClient(cosmos_mongodb_connection_string,
58
- tls=True,
59
- tlsCAFile=certifi.where(),
60
- retryWrites=False,
61
- serverSelectionTimeoutMS=5000,
62
- connectTimeoutMS=10000,
63
- socketTimeoutMS=10000)
64
-
65
- #Forzar una conexión para verificar
66
- mongo_client.admin.command('ping')
67
-
68
- #Seleccionar la base de datos
69
- client.admin.command('ismaster')
70
-
71
- #Seleccionar la base de datos
72
- db = client['aideatext_db']
73
- analysis_collection = db['text_analysis']
74
-
75
- # Prueba de conexión
76
- mongo_client.server_info()
77
- print("Conexión a MongoDB API exitosa")
78
-
79
- # Verificar que la colección existe
80
- if 'text_analysis' in mongo_db.list_collection_names():
81
- print("Colección 'text_analysis' encontrada")
82
- else:
83
- print("Advertencia: La colección 'text_analysis' no existe. Se creará automáticamente al insertar el primer documento.")
84
-
85
- except Exception as e:
86
- logger.error(f"Error al conectar con Cosmos DB MongoDB API: {str(e)}", exc_info=True)
87
- raise
88
 
89
  # Función para insertar un documento
90
  def insert_document(document):
 
49
  #mongo_connection_string = os.environ.get("MONGODB_CONNECTION_STRING")
50
  cosmos_mongodb_connection_string = os.getenv("MONGODB_CONNECTION_STRING")
51
 
52
+ # Variable global para el cliente de MongoDB
53
+ mongo_client = None
54
+ db = None
55
+ analysis_collection = None
56
 
57
+ def initialize_mongodb_connection():
58
+ global mongo_client, db, analysis_collection
59
+ try:
60
+ # Crear el cliente de MongoDB con configuración TLS
61
+ mongo_client = MongoClient(cosmos_mongodb_connection_string,
62
+ tls=True,
63
+ tlsCAFile=certifi.where(),
64
+ retryWrites=False,
65
+ serverSelectionTimeoutMS=5000,
66
+ connectTimeoutMS=10000,
67
+ socketTimeoutMS=10000)
68
+
69
+ # Forzar una conexión para verificar
70
+ mongo_client.admin.command('ping')
71
+
72
+ # Seleccionar la base de datos y la colección
73
+ db = mongo_client['aideatext_db']
74
+ analysis_collection = db['text_analysis']
75
+
76
+ logger.info("Conexión a Cosmos DB MongoDB API exitosa")
77
+ return True
78
+ except Exception as e:
79
+ logger.error(f"Error al conectar con Cosmos DB MongoDB API: {str(e)}", exc_info=True)
80
+ return False
 
 
 
 
 
 
 
 
 
81
 
82
  # Función para insertar un documento
83
  def insert_document(document):