Update modules/database/current_situation_mongo_db.py

modules/database/current_situation_mongo_db.py

@@ -1,18 +1,12 @@
 # modules/database/current_situation_mongo_db.py
-
-from datetime import datetime, timezone
+from datetime import datetime, timezone, timedelta
 import logging
-from .mongo_db import get_collection, insert_document, find_documents
+from .mongo_db import get_collection
 
 logger = logging.getLogger(__name__)
 COLLECTION_NAME = 'student_current_situation'
 
-# En modules/database/current_situation_mongo_db.py
 def store_current_situation_result(username, text, metrics, feedback):
-    success = store_current_situation_result(username, text, metrics, feedback)
-    if success:
-        verify_storage(username) #Verificar que se guarda correctamente
-    return success
     """
     Guarda los resultados del análisis de situación actual.
     """
@@ -21,8 +15,8 @@ def store_current_situation_result(username, text, metrics, feedback):
         if not all([username, text, metrics]):
             logger.error("Faltan parámetros requeridos")
             return False
-        
-        collection = get_collection('student_current_situation')
+
+        collection = get_collection(COLLECTION_NAME)
         if collection is None:
             logger.error("No se pudo obtener la colección")
             return False
@@ -46,20 +40,19 @@ def store_current_situation_result(username, text, metrics, feedback):
                 'details': metrics['clarity']['details']
             }
         }
-    
+
         # Crear documento
         document = {
             'username': username,
             'timestamp': datetime.now(timezone.utc).isoformat(),
             'text': text,
-            'metrics': metrics,
+            'metrics': formatted_metrics,  # Usar las métricas formateadas
             'feedback': feedback,
             'analysis_type': 'current_situation'
         }
-        
+
         # Insertar documento
         result = collection.insert_one(document)
-        
         if result.inserted_id:
             logger.info(f"""
                 Análisis guardado exitosamente:
@@ -68,11 +61,18 @@ def store_current_situation_result(username, text, metrics, feedback):
                 - Longitud del texto: {len(text)}
                 - Métricas: {formatted_metrics}
             """)
-            return True
             
+            # Verificar almacenamiento
+            if verify_storage(username):
+                logger.info("Verificación de almacenamiento exitosa")
+                return True
+            else:
+                logger.warning("Verificación de almacenamiento falló")
+                return False
+
         logger.error("No se pudo insertar el documento")
         return False
-        
+
     except Exception as e:
         logger.error(f"Error guardando análisis de situación actual: {str(e)}")
         return False
@@ -84,14 +84,15 @@ def verify_storage(username):
     try:
         collection = get_collection(COLLECTION_NAME)
         if collection is None:
+            logger.error("No se pudo obtener la colección para verificación")
             return False
             
         # Buscar documentos recientes del usuario
+        timestamp_threshold = (datetime.now(timezone.utc) - timedelta(minutes=5)).isoformat()
+        
         recent_docs = collection.find({
             'username': username,
-            'timestamp': {
-                '$gte': (datetime.now(timezone.utc) - timedelta(minutes=5)).isoformat()
-            }
+            'timestamp': {'$gte': timestamp_threshold}
         }).sort('timestamp', -1).limit(1)
 
         docs = list(recent_docs)
@@ -109,4 +110,23 @@ def verify_storage(username):
 
     except Exception as e:
         logger.error(f"Error verificando almacenamiento: {str(e)}")
-        return False
+        return False
+
+def get_recent_situation_analysis(username, limit=5):
+    """
+    Obtiene los análisis más recientes de un usuario.
+    """
+    try:
+        collection = get_collection(COLLECTION_NAME)
+        if collection is None:
+            return []
+            
+        results = collection.find(
+            {'username': username}
+        ).sort('timestamp', -1).limit(limit)
+        
+        return list(results)
+        
+    except Exception as e:
+        logger.error(f"Error obteniendo análisis recientes: {str(e)}")
+        return []