Update modules/database/morphosyntax_iterative_mongo_db.py

modules/database/morphosyntax_iterative_mongo_db.py

@@ -6,18 +6,15 @@ from .mongo_db import get_collection, insert_document, find_documents, update_do
 
 logger = logging.getLogger(__name__)
 
-# Constantes para las colecciones
-BASE_COLLECTION = 'student_morphosyntax_base'
-ITERATION_COLLECTION = 'student_morphosyntax_iterations'
+# Nombres de colecciones manteniendo coherencia
+COLLECTION_NAME = 'student_morphosyntax_analysis'
+COLLECTION_NAME_ITERATIONS = 'student_morphosyntax_iterations'
 
 def store_student_morphosyntax_base(username, text, arc_diagrams):
-    """
-    Almacena el análisis morfosintáctico base
-    Returns: ID del documento base o None si hay error
-    """
+    """Almacena el análisis morfosintáctico base"""
     try:
         base_document = {
-            'username': username,
+            'username': username,  # key principal
             'timestamp': datetime.now(timezone.utc).isoformat(),
             'text': text,
             'arc_diagrams': arc_diagrams,
@@ -25,7 +22,7 @@ def store_student_morphosyntax_base(username, text, arc_diagrams):
             'has_iterations': False
         }
         
-        collection = get_collection(BASE_COLLECTION)
+        collection = get_collection(COLLECTION_NAME)
         result = collection.insert_one(base_document)
         logger.info(f"Análisis base guardado para {username}")
         return str(result.inserted_id)
@@ -35,28 +32,25 @@ def store_student_morphosyntax_base(username, text, arc_diagrams):
         return None
 
 def store_student_morphosyntax_iteration(username, base_id, original_text, iteration_text, arc_diagrams):
-    """
-    Almacena una iteración de análisis morfosintáctico
-    Returns: ID de la iteración o None si hay error
-    """
+    """Almacena una iteración de análisis morfosintáctico"""
     try:
         iteration_document = {
-            'username': username,
+            'username': username,  # key principal
             'base_id': base_id,
             'timestamp': datetime.now(timezone.utc).isoformat(),
             'original_text': original_text,
-            'iteration_text': iteration_text, 
+            'iteration_text': iteration_text,
             'arc_diagrams': arc_diagrams,
             'analysis_type': 'morphosyntax_iteration'
         }
         
-        collection = get_collection(ITERATION_COLLECTION)
+        collection = get_collection(COLLECTION_NAME_ITERATIONS)
         result = collection.insert_one(iteration_document)
         
-        # Actualizar documento base para indicar que tiene iteraciones
-        base_collection = get_collection(BASE_COLLECTION)
+        # Actualizar documento base
+        base_collection = get_collection(COLLECTION_NAME)
         base_collection.update_one(
-            {'_id': base_id},
+            {'_id': base_id, 'username': username},  # incluir username en queries
             {'$set': {'has_iterations': True}}
         )
         
@@ -65,90 +59,4 @@ def store_student_morphosyntax_iteration(username, base_id, original_text, itera
         
     except Exception as e:
         logger.error(f"Error almacenando iteración: {str(e)}")
-        return None
-
-def get_student_morphosyntax_analysis(username, limit=10):
-    """
-    Obtiene los análisis base y sus iteraciones
-    Returns: Lista de análisis con sus iteraciones
-    """
-    try:
-        # Obtener análisis base
-        base_collection = get_collection(BASE_COLLECTION)
-        base_query = {
-            "username": username,
-            "analysis_type": "morphosyntax_base"
-        }
-        base_analyses = list(base_collection.find(base_query).sort("timestamp", -1).limit(limit))
-        
-        # Para cada análisis base, obtener sus iteraciones
-        iteration_collection = get_collection(ITERATION_COLLECTION)
-        for analysis in base_analyses:
-            base_id = analysis['_id']
-            iterations = list(iteration_collection.find({"base_id": base_id}).sort("timestamp", -1))
-            analysis['iterations'] = iterations
-            
-        return base_analyses
-        
-    except Exception as e:
-        logger.error(f"Error obteniendo análisis: {str(e)}")
-        return []
-
-def update_student_morphosyntax_analysis(analysis_id, is_base, update_data):
-    """
-    Actualiza un análisis base o iteración
-    """
-    try:
-        collection_name = BASE_COLLECTION if is_base else ITERATION_COLLECTION
-        collection = get_collection(collection_name)
-        
-        query = {"_id": analysis_id}
-        update = {"$set": update_data}
-        
-        result = update_document(collection_name, query, update)
-        return result
-        
-    except Exception as e:
-        logger.error(f"Error actualizando análisis: {str(e)}")
-        return False
-
-def delete_student_morphosyntax_analysis(analysis_id, is_base):
-    """
-    Elimina un análisis base o iteración
-    Si es base, también elimina todas sus iteraciones
-    """
-    try:
-        if is_base:
-            # Primero eliminar todas las iteraciones
-            iteration_collection = get_collection(ITERATION_COLLECTION)
-            iteration_collection.delete_many({"base_id": analysis_id})
-            
-            # Luego eliminar el análisis base
-            collection = get_collection(BASE_COLLECTION)
-        else:
-            collection = get_collection(ITERATION_COLLECTION)
-            
-        query = {"_id": analysis_id}
-        result = delete_document(collection.name, query)
-        return result
-        
-    except Exception as e:
-        logger.error(f"Error eliminando análisis: {str(e)}")
-        return False
-
-def get_student_morphosyntax_data(username):
-    """
-    Obtiene todos los datos de análisis morfosintáctico de un estudiante
-    Returns: Diccionario con todos los análisis y sus iteraciones
-    """
-    try:
-        analyses = get_student_morphosyntax_analysis(username, limit=None)
-        return {
-            'entries': analyses,
-            'total_analyses': len(analyses),
-            'has_iterations': any(a.get('has_iterations', False) for a in analyses)
-        }
-        
-    except Exception as e:
-        logger.error(f"Error obteniendo datos del estudiante: {str(e)}")
-        return {'entries': [], 'total_analyses': 0, 'has_iterations': False}
+        return None