Upload 2 files

modules/database/mongo_db.py

@@ -1,83 +1,62 @@
-from .database_init import get_mongodb
-import logging
-
-logger = logging.getLogger(__name__)
-
-def get_collection(collection_name):
-    try:
-        db = get_mongodb()
-        if db is None:
-            logger.error(f"No se pudo obtener la base de datos para {collection_name}")
-            return None
-            
-        collection = db[collection_name]
-        logger.info(f"Colección {collection_name} obtenida exitosamente")
-        return collection
-        
-    except Exception as e:
-        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):
-    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.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    
-    
-#######################################################
-def find_documents(collection_name, query, sort=None, limit=None):
-    collection = get_collection(collection_name)
-    try:
-        cursor = collection.find(query)
-        if sort:
-            cursor = cursor.sort(sort)
-        if limit:
-            cursor = cursor.limit(limit)
-        return list(cursor)
-    except Exception as e:
-        logger.error(f"Error al buscar documentos en {collection_name}: {str(e)}")
-        return []
-
-def update_document(collection_name, query, update):
-    collection = get_collection(collection_name)
-    try:
-        result = collection.update_one(query, update)
-        logger.info(f"Documento actualizado en {collection_name}: {result.modified_count} modificado(s)")
-        return result.modified_count
-    except Exception as e:
-        logger.error(f"Error al actualizar documento en {collection_name}: {str(e)}")
-        return 0
-
-def delete_document(collection_name, query):
-    collection = get_collection(collection_name)
-    try:
-        result = collection.delete_one(query)
-        logger.info(f"Documento eliminado de {collection_name}: {result.deleted_count} eliminado(s)")
-        return result.deleted_count
-    except Exception as e:
-        logger.error(f"Error al eliminar documento de {collection_name}: {str(e)}")
+from .database_init import get_mongodb
+import logging
+
+logger = logging.getLogger(__name__)
+
+def get_collection(collection_name):
+    try:
+        db = get_mongodb()
+        if db is None:
+            logger.error(f"No se pudo obtener la base de datos para {collection_name}")
+            return None
+            
+        collection = db[collection_name]
+        logger.info(f"Colección {collection_name} obtenida exitosamente")
+        return collection
+        
+    except Exception as e:
+        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 find_documents(collection_name, query, sort=None, limit=None):
+    collection = get_collection(collection_name)
+    try:
+        cursor = collection.find(query)
+        if sort:
+            cursor = cursor.sort(sort)
+        if limit:
+            cursor = cursor.limit(limit)
+        return list(cursor)
+    except Exception as e:
+        logger.error(f"Error al buscar documentos en {collection_name}: {str(e)}")
+        return []
+
+def update_document(collection_name, query, update):
+    collection = get_collection(collection_name)
+    try:
+        result = collection.update_one(query, update)
+        logger.info(f"Documento actualizado en {collection_name}: {result.modified_count} modificado(s)")
+        return result.modified_count
+    except Exception as e:
+        logger.error(f"Error al actualizar documento en {collection_name}: {str(e)}")
+        return 0
+
+def delete_document(collection_name, query):
+    collection = get_collection(collection_name)
+    try:
+        result = collection.delete_one(query)
+        logger.info(f"Documento eliminado de {collection_name}: {result.deleted_count} eliminado(s)")
+        return result.deleted_count
+    except Exception as e:
+        logger.error(f"Error al eliminar documento de {collection_name}: {str(e)}")
         return 0

modules/database/semantic_mongo_db.py

@@ -1,167 +1,160 @@
-#/modules/database/semantic_mongo_db.py
-
-# Importaciones estándar
-import io
-import base64
-from datetime import datetime, timezone
-import logging
-
-# Importaciones de terceros
-import matplotlib.pyplot as plt
-
-# Importaciones locales
-from .mongo_db import (
-    get_collection,
-    insert_document, 
-    find_documents, 
-    update_document, 
-    delete_document
-)
-
-# Configuración del logger
-logger = logging.getLogger(__name__)  # Cambiado de name a __name__
-COLLECTION_NAME = 'student_semantic_analysis'
-
-############################################################
-
-def store_student_semantic_result(username, text, analysis_result):
-    """
-    Guarda el resultado del análisis semántico en MongoDB.
-    """
-    try:
-        # El gráfico ya viene en bytes, solo necesitamos codificarlo a base64
-        concept_graph_data = None
-        if 'concept_graph' in analysis_result and analysis_result['concept_graph'] is not None:
-            try:
-                # Ya está en bytes, solo codificar a base64
-                concept_graph_data = base64.b64encode(analysis_result['concept_graph']).decode('utf-8')
-            except Exception as e:
-                logger.error(f"Error al codificar gráfico conceptual: {str(e)}")
-
-        # Crear documento para MongoDB
-        analysis_document = {
-            'username': username,
-            'timestamp': datetime.now(timezone.utc).isoformat(),
-            'text': text,
-            'analysis_type': 'semantic',
-            'key_concepts': analysis_result.get('key_concepts', []),
-            'concept_graph': concept_graph_data
-        }
-
-        logger.debug(f"Intentando insertar documento para usuario: {username}")
-        logger.debug(f"Documento: {analysis_document}")
-
-        # Insertar en MongoDB
-        result = insert_document(COLLECTION_NAME, analysis_document)
-        if result:
-            logger.info(f"Análisis semántico guardado con ID: {result} para el usuario: {username}")
-            return True
-        
-        logger.error("No se pudo insertar el documento en MongoDB")
-        return False
-
-    except Exception as e:
-        logger.error(f"Error al guardar el análisis semántico: {str(e)}")
-        return False
-
-
-
-####################################################################################
-def get_student_semantic_analysis(username, limit=10):
-    """
-    Recupera los análisis semánticos de un estudiante.
-    """
-    try:
-        # Obtener la colección
-        collection = get_collection(COLLECTION_NAME)
-        if collection is None:  # Cambiado de if not collection a if collection is None
-            logger.error("No se pudo obtener la colección semantic")
-            return []
-
-        # Consulta
-        query = {
-            "username": username,
-            "analysis_type": "semantic"
-        }
-        
-        # Campos a recuperar
-        projection = {
-            "timestamp": 1,
-            "concept_graph": 1,
-            "_id": 1
-        }
-        
-        # Ejecutar consulta
-        try:
-            cursor = collection.find(query, projection).sort("timestamp", -1)
-            if limit:
-                cursor = cursor.limit(limit)
-            
-            # Convertir cursor a lista
-            results = list(cursor)
-            logger.info(f"Recuperados {len(results)} análisis semánticos para {username}")
-            return results
-            
-        except Exception as db_error:
-            logger.error(f"Error en la consulta a MongoDB: {str(db_error)}")
-            return []
-        
-    except Exception as e:
-        logger.error(f"Error recuperando análisis semántico: {str(e)}")
-        return []
-####################################################################################################
-
-
-def update_student_semantic_analysis(analysis_id, update_data):
-    """
-    Actualiza un análisis semántico existente.
-    Args:
-        analysis_id: ID del análisis a actualizar
-        update_data: Datos a actualizar
-    """
-    query = {"_id": analysis_id}
-    update = {"$set": update_data}
-    return update_document(COLLECTION_NAME, query, update)
-
-def delete_student_semantic_analysis(analysis_id):
-    """
-    Elimina un análisis semántico.
-    Args:
-        analysis_id: ID del análisis a eliminar
-    """
-    query = {"_id": analysis_id}
-    return delete_document(COLLECTION_NAME, query)
-
-def get_student_semantic_data(username):
-    """
-    Obtiene todos los análisis semánticos de un estudiante.
-    Args:
-        username: Nombre del usuario
-    Returns:
-        dict: Diccionario con todos los análisis del estudiante
-    """
-    analyses = get_student_semantic_analysis(username, limit=None)
-    
-    formatted_analyses = []
-    for analysis in analyses:
-        formatted_analysis = {
-            'timestamp': analysis['timestamp'],
-            'text': analysis['text'],
-            'key_concepts': analysis['key_concepts'],
-            'entities': analysis['entities']
-            # No incluimos los gráficos en el resumen general
-        }
-        formatted_analyses.append(formatted_analysis)
-    
-    return {
-        'entries': formatted_analyses
-    }
-
-# Exportar las funciones necesarias
-__all__ = [
-    'store_student_semantic_result',
-    'get_student_semantic_analysis',
-    'update_student_semantic_analysis',
-    'delete_student_semantic_analysis',
-    'get_student_semantic_data'
+#/modules/database/semantic_mongo_db.py
+
+# Importaciones estándar
+import io
+import base64
+from datetime import datetime, timezone
+import logging
+
+# Importaciones de terceros
+import matplotlib.pyplot as plt
+
+# Importaciones locales
+from .mongo_db import (
+    get_collection,
+    insert_document, 
+    find_documents, 
+    update_document, 
+    delete_document
+)
+
+# Configuración del logger
+logger = logging.getLogger(__name__)  # Cambiado de name a __name__
+COLLECTION_NAME = 'student_semantic_analysis'
+
+def store_student_semantic_result(username, text, analysis_result):
+    """
+    Guarda el resultado del análisis semántico en MongoDB.
+    """
+    try:
+        # El gráfico ya viene en bytes, solo necesitamos codificarlo a base64
+        concept_graph_data = None
+        if 'concept_graph' in analysis_result and analysis_result['concept_graph'] is not None:
+            try:
+                # Ya está en bytes, solo codificar a base64
+                concept_graph_data = base64.b64encode(analysis_result['concept_graph']).decode('utf-8')
+            except Exception as e:
+                logger.error(f"Error al codificar gráfico conceptual: {str(e)}")
+
+        # Crear documento para MongoDB
+        analysis_document = {
+            'username': username,
+            'timestamp': datetime.now(timezone.utc).isoformat(),
+            'text': text,
+            'analysis_type': 'semantic',
+            'key_concepts': analysis_result.get('key_concepts', []),
+            'concept_graph': concept_graph_data
+        }
+
+        # Insertar en MongoDB
+        result = insert_document(COLLECTION_NAME, analysis_document)
+        if result:
+            logger.info(f"Análisis semántico guardado con ID: {result} para el usuario: {username}")
+            return True
+        
+        logger.error("No se pudo insertar el documento en MongoDB")
+        return False
+
+    except Exception as e:
+        logger.error(f"Error al guardar el análisis semántico: {str(e)}")
+        return False
+
+####################################################################################
+def get_student_semantic_analysis(username, limit=10):
+    """
+    Recupera los análisis semánticos de un estudiante.
+    """
+    try:
+        # Obtener la colección
+        collection = get_collection(COLLECTION_NAME)
+        if collection is None:  # Cambiado de if not collection a if collection is None
+            logger.error("No se pudo obtener la colección semantic")
+            return []
+
+        # Consulta
+        query = {
+            "username": username,
+            "analysis_type": "semantic"
+        }
+        
+        # Campos a recuperar
+        projection = {
+            "timestamp": 1,
+            "concept_graph": 1,
+            "_id": 1
+        }
+        
+        # Ejecutar consulta
+        try:
+            cursor = collection.find(query, projection).sort("timestamp", -1)
+            if limit:
+                cursor = cursor.limit(limit)
+            
+            # Convertir cursor a lista
+            results = list(cursor)
+            logger.info(f"Recuperados {len(results)} análisis semánticos para {username}")
+            return results
+            
+        except Exception as db_error:
+            logger.error(f"Error en la consulta a MongoDB: {str(db_error)}")
+            return []
+        
+    except Exception as e:
+        logger.error(f"Error recuperando análisis semántico: {str(e)}")
+        return []
+####################################################################################################
+
+
+def update_student_semantic_analysis(analysis_id, update_data):
+    """
+    Actualiza un análisis semántico existente.
+    Args:
+        analysis_id: ID del análisis a actualizar
+        update_data: Datos a actualizar
+    """
+    query = {"_id": analysis_id}
+    update = {"$set": update_data}
+    return update_document(COLLECTION_NAME, query, update)
+
+def delete_student_semantic_analysis(analysis_id):
+    """
+    Elimina un análisis semántico.
+    Args:
+        analysis_id: ID del análisis a eliminar
+    """
+    query = {"_id": analysis_id}
+    return delete_document(COLLECTION_NAME, query)
+
+def get_student_semantic_data(username):
+    """
+    Obtiene todos los análisis semánticos de un estudiante.
+    Args:
+        username: Nombre del usuario
+    Returns:
+        dict: Diccionario con todos los análisis del estudiante
+    """
+    analyses = get_student_semantic_analysis(username, limit=None)
+    
+    formatted_analyses = []
+    for analysis in analyses:
+        formatted_analysis = {
+            'timestamp': analysis['timestamp'],
+            'text': analysis['text'],
+            'key_concepts': analysis['key_concepts'],
+            'entities': analysis['entities']
+            # No incluimos los gráficos en el resumen general
+        }
+        formatted_analyses.append(formatted_analysis)
+    
+    return {
+        'entries': formatted_analyses
+    }
+
+# Exportar las funciones necesarias
+__all__ = [
+    'store_student_semantic_result',
+    'get_student_semantic_analysis',
+    'update_student_semantic_analysis',
+    'delete_student_semantic_analysis',
+    'get_student_semantic_data'
 ]