Update modules/database/sql_db.py

modules/database/sql_db.py

@@ -201,3 +201,71 @@ def get_user_total_time(username):
     except Exception as e:
         logger.error(f"Error obteniendo tiempo total: {str(e)}")
         return 0
+
+
+# Agregar estas funciones en sql_db.py
+
+def update_student_user(username, new_info):
+    """Actualiza la información de un estudiante"""
+    container = get_container("users")
+    try:
+        # Primero obtener el usuario existente
+        user = get_student_user(username)
+        if user:
+            # Actualizar la información
+            if 'additional_info' in user:
+                user['additional_info'].update(new_info)
+            else:
+                user['additional_info'] = new_info
+                
+            # Actualizar el documento usando el partition key correcto
+            container.upsert_item(
+                body=user,
+                partition_key=username
+            )
+            logger.info(f"Información del estudiante actualizada: {username}")
+            return True
+        else:
+            logger.warning(f"Intento de actualizar estudiante no existente: {username}")
+            return False
+    except Exception as e:
+        logger.error(f"Error al actualizar información del estudiante {username}: {str(e)}")
+        return False
+
+def delete_student_user(username):
+    """Elimina un estudiante"""
+    container = get_container("users")
+    try:
+        # Verificar que el usuario existe y es un estudiante
+        user = get_student_user(username)
+        if user:
+            # Eliminar usando el partition key correcto
+            container.delete_item(
+                item=user['id'],
+                partition_key=username
+            )
+            logger.info(f"Estudiante eliminado: {username}")
+            return True
+        else:
+            logger.warning(f"Intento de eliminar estudiante no existente: {username}")
+            return False
+    except Exception as e:
+        logger.error(f"Error al eliminar estudiante {username}: {str(e)}")
+        return False
+
+__all__ = [
+    'get_user',
+    'get_admin_user',
+    'get_student_user',
+    'get_teacher_user',
+    'create_user',
+    'create_student_user',
+    'create_teacher_user',
+    'create_admin_user',
+    'update_student_user',  # Agregada
+    'delete_student_user',  # Agregada
+    'record_login',
+    'record_logout',
+    'get_recent_sessions',
+    'get_user_total_time'
+]