Update modules/database/mongo_db.py

modules/database/mongo_db.py

@@ -18,16 +18,37 @@ def get_collection(collection_name):
         logger.error(f"Error al obtener colección {collection_name}: {str(e)}")
         return None
 
+
+
+
+########################################################
+#def insert_document(collection_name, document):
+#    collection = get_collection(collection_name)
+#    try:
+#        result = collection.insert_one(document)
+#        logger.info(f"Documento insertado en {collection_name} con ID: {result.inserted_id}")
+#        return result.inserted_id
+#    except Exception as e:
+#        logger.error(f"Error al insertar documento en {collection_name}: {str(e)}")
+#        return None
+##################################################
 def insert_document(collection_name, document):
-    collection = get_collection(collection_name)
     try:
+        collection = get_collection(collection_name)
+        if collection is None:
+            logger.error(f"No se pudo obtener la colección {collection_name}")
+            return None
+            
+        # Insertar documento
         result = collection.insert_one(document)
-        logger.info(f"Documento insertado en {collection_name} con ID: {result.inserted_id}")
+        logger.debug(f"Resultado de inserción: {result.inserted_id}")
         return result.inserted_id
+
     except Exception as e:
         logger.error(f"Error al insertar documento en {collection_name}: {str(e)}")
-        return None
-
+        return None    
+    
+#######################################################
 def find_documents(collection_name, query, sort=None, limit=None):
     collection = get_collection(collection_name)
     try: