add/remove reading exercises

Author: rockstarr-programmerr

classroom/business/__init__.py

@@ -3,6 +3,7 @@
 from django.contrib.auth import get_user_model
 
 from classroom.tasks import send_temp_password_for_new_students_task
+from classroom.models import ReadingExercise
 
 User = get_user_model()
 
@@ -42,3 +43,12 @@ def add_students_to_classroom(classroom, students_data):
 def remove_students_from_classroom(classroom, student_emails):
     students_to_remove = classroom.students.filter(email__in=student_emails)
     classroom.students.remove(*students_to_remove)
+
+
+def add_reading_exercises_to_classroom(classroom, exercise_pks, user):
+    exercises = user.reading_exercises_created.filter(pk__in=exercise_pks)
+    classroom.reading_exercises.add(*exercises)
+
+
+def remove_reading_exercises_to_classroom(classroom, exercise_pks):
+    classroom.reading_exercises.remove(*exercise_pks)

classroom/business/classroom_teacher.py renamed to classroom/business/teacher/classroom.py

@@ -1,3 +1,5 @@
-from .classroom import (AddStudentSerializer, ClassroomTeacherSerializer,
+from .classroom import (AddReadingExerciseSerializer, AddStudentSerializer,
+                        ClassroomTeacherSerializer,
+                        RemoveReadingExerciseSerializer,
                         RemoveStudentSerializer)
 from .exercise import ReadingExerciseSerializer

classroom/serializers/teacher/__init__.py

@@ -14,10 +14,11 @@ class Meta:
         model = Classroom
         fields = [
             'pk', 'url', 'name', 'description', 'create_datetime',
-            'teacher', 'students',
+            'teacher', 'students', 'reading_exercises',
         ]
         extra_kwargs = {
             'url': {'view_name': 'classroom-teacher-detail'},
+            'reading_exercises': {'view_name': 'reading-exercise-teacher-detail'},
             'create_datetime': {'read_only': True},
         }
 
@@ -41,3 +42,11 @@ class AddStudentSerializer(serializers.Serializer):
 
 class RemoveStudentSerializer(serializers.Serializer):
     email = serializers.EmailField()
+
+
+class AddReadingExerciseSerializer(serializers.Serializer):
+    pk = serializers.IntegerField()
+
+
+class RemoveReadingExerciseSerializer(serializers.Serializer):
+    pk = serializers.IntegerField()

classroom/serializers/teacher/classroom.py

@@ -3,12 +3,14 @@
 from rest_framework.response import Response
 from rest_framework.viewsets import ModelViewSet
 
-from classroom import business
+from classroom.business.teacher import classroom as business
 from classroom.filters.teacher import ClassroomTeacherFilter
 from classroom.models import Classroom
 from classroom.permissions import IsClassroomTeacher
-from classroom.serializers.teacher import (AddStudentSerializer,
+from classroom.serializers.teacher import (AddReadingExerciseSerializer,
+                                           AddStudentSerializer,
                                            ClassroomTeacherSerializer,
+                                           RemoveReadingExerciseSerializer,
                                            RemoveStudentSerializer)
 
 
@@ -60,3 +62,27 @@ def remove_students(self, request, pk):
         classroom = self.get_object()
         business.remove_students_from_classroom(classroom, emails)
         return Response()
+
+    @action(
+        methods=['POST'], detail=True, url_path='add-reading-exercises',
+        serializer_class=AddReadingExerciseSerializer
+    )
+    def add_reading_exercises(self, request, pk):
+        serializer = self.get_serializer(data=request.data, many=True)
+        serializer.is_valid(raise_exception=True)
+        exercise_pks = [data['pk'] for data in serializer.validated_data]
+        classroom = self.get_object()
+        business.add_reading_exercises_to_classroom(classroom, exercise_pks, request.user)
+        return Response()
+
+    @action(
+        methods=['POST'], detail=True, url_path='remove-reading-exercises',
+        serializer_class=RemoveReadingExerciseSerializer,
+    )
+    def remove_reading_exercises(self, request, pk):
+        serializer = self.get_serializer(data=request.data, many=True)
+        serializer.is_valid(raise_exception=True)
+        exercise_pks = [data['pk'] for data in serializer.validated_data]
+        classroom = self.get_object()
+        business.remove_reading_exercises_to_classroom(classroom, exercise_pks)
+        return Response()