AIdeaText commited on
Commit
fb5f4d5
1 Parent(s): 60795c9

Update modules/database.py

Browse files
Files changed (1) hide show
  1. modules/database.py +5 -0
modules/database.py CHANGED
@@ -96,8 +96,10 @@ def get_student_data(username):
96
  return None
97
 
98
  try:
 
99
  cursor = analysis_collection.find({"username": username}).sort("timestamp", -1)
100
 
 
101
  formatted_data = {
102
  "username": username,
103
  "entries": [],
@@ -114,15 +116,18 @@ def get_student_data(username):
114
  })
115
  formatted_data["entries_count"] += 1
116
 
 
117
  for category, count in entry.get("word_count", {}).items():
118
  if category in formatted_data["word_count"]:
119
  formatted_data["word_count"][category] += count
120
  else:
121
  formatted_data["word_count"][category] = count
122
 
 
123
  formatted_data["arc_diagrams"].extend(entry.get("arc_diagrams", []))
124
  formatted_data["network_diagrams"].append(entry.get("network_diagram", ""))
125
 
 
126
  return formatted_data if formatted_data["entries_count"] > 0 else None
127
  except Exception as e:
128
  logger.error(f"Error al obtener datos del estudiante {username}: {str(e)}")
 
96
  return None
97
 
98
  try:
99
+ # Buscar los datos del estudiante
100
  cursor = analysis_collection.find({"username": username}).sort("timestamp", -1)
101
 
102
+ # Formatear los datos
103
  formatted_data = {
104
  "username": username,
105
  "entries": [],
 
116
  })
117
  formatted_data["entries_count"] += 1
118
 
119
+ # Agregar conteo de palabras
120
  for category, count in entry.get("word_count", {}).items():
121
  if category in formatted_data["word_count"]:
122
  formatted_data["word_count"][category] += count
123
  else:
124
  formatted_data["word_count"][category] = count
125
 
126
+ # Agregar diagramas
127
  formatted_data["arc_diagrams"].extend(entry.get("arc_diagrams", []))
128
  formatted_data["network_diagrams"].append(entry.get("network_diagram", ""))
129
 
130
+ logger.info(f"Datos obtenidos para {username}: {formatted_data}")
131
  return formatted_data if formatted_data["entries_count"] > 0 else None
132
  except Exception as e:
133
  logger.error(f"Error al obtener datos del estudiante {username}: {str(e)}")