Merge pull request #25 from DevOps-Cloud-Team5/SCRUM-90-Course-info

allow disenrollment from course

Author: Imafe

api/models.py

@@ -82,6 +82,11 @@ def get_lectures_week(self, year : int, week : int):
     def is_user_enrolled(self, user : User):
         return bool(UserCourse.objects.filter(user=user, course=self))
 
+    def remove_user_from_course(self, user : User):
+        queryset = UserCourse.objects.filter(user=user, course=self)
+        if not queryset: return
+        queryset[0].delete()
+
     def add_user_to_course(self, user: User):
         UserCourse.objects.create(user=user, course=self).save()
 

api/urls.py

@@ -19,6 +19,7 @@
 from django_rest_passwordreset.views import ResetPasswordConfirm, ResetPasswordValidateToken
 from .views import (
                     AddLectureView,
+                    DisenrollCourseView,
                     GetCourseLecturesView,
                     GetFullCoursePage,
                     GetLectureView,
@@ -75,6 +76,7 @@
     path('course/update/<pk>', UpdateCourseView.as_view(), name='course_update'),
     path('course/delete/<pk>', DestroyCourseView.as_view(), name='course_delete'),
     path('course/enroll/<pk>', EnrollCourseView.as_view(), name='course_enroll'),
+    path('course/disenroll/<pk>', DisenrollCourseView.as_view(), name='course_enroll'),
     path('course/mass_enroll/<pk>', MassEnrollCourseView.as_view(), name='course_mass_enroll'),
     path('course/get/<pk>', GetFullCoursePage.as_view(), name='course_get'),
     path('course/getall/', GetCoursesAll.as_view(), name='course_getall'),

api/views.py

@@ -17,7 +17,7 @@
 from .permissions import IsTeacher, IsAdmin, IsStudent
 from .models import Course, AccountRoles, CourseLecture
 
-from .serializers import AddLectureSerializer, CourseUserSerializer, CustomTokenSerializer, CreateUserSerializer, LectureSerializer, MassEnrollSerializer, SetAttendenceTeacherSerializer, UserSerializer, CourseCreateSerializer, CourseSerializer
+from .serializers import AddLectureSerializer, CourseUserSerializer, CustomTokenSerializer, CreateUserSerializer, LectureSerializer, MailTestSerializer, MassEnrollSerializer, SetAttendenceTeacherSerializer, UserSerializer, CourseCreateSerializer, CourseSerializer
 
 
 
@@ -242,6 +242,7 @@ def get(self, request, pk):
         if not queryset:
             return Response({"error": f"course id '{pk}' not found"}, status=status.HTTP_404_NOT_FOUND)
 
+        user = User.objects.all().filter(username=request.user.username)[0]
         course : Course = queryset[0]
         teachers = course.get_teachers()
         students = course.get_enrolled_students()
@@ -251,11 +252,11 @@ def get(self, request, pk):
         response_data["course_name"] = course.course_name
         response_data["num_teachers"] = len(teachers)
         response_data["num_students"] = len(students)
+        response_data["enrolled"] = course.is_user_enrolled(user=user)
         response_data["attended"] = -1
         response_data["missed"] = -1
         response_data["users"] = CourseUserSerializer((teachers + students), many=True).data
 
-        user = User.objects.all().filter(username=request.user.username)[0]
         if user.role == AccountRoles.STUDENT: 
             response_data |= self.get_attendence_stats(course, user)
 
@@ -305,6 +306,29 @@ def post(self, request, *args, **kwargs):
 
         return Response({"ok": f"succesfully enrolled {username} in {obj.course_name}"}, status=status.HTTP_200_OK)
 
+class DisenrollCourseView(generics.GenericAPIView):
+    authentication_classes = [JWTAuthentication]
+    permission_classes = [IsStudent]
+    lookup_field = 'pk'
+    
+    queryset = Course.objects.all()
+    serializer_class = CourseSerializer
+    
+    def post(self, request, *args, **kwargs):
+        obj : Course = self.get_object()
+        user = request.user
+        username = user.username
+        
+        if not obj.is_user_enrolled(request.user):
+            return Response({"error": f"{username} is not enrolled in {obj.course_name}"}, status=status.HTTP_400_BAD_REQUEST)
+            
+        obj.remove_user_from_course(user)
+
+        if getattr(obj, '_prefetched_objects_cache', None):
+            obj._prefetched_objects_cache = {}
+            
+        return Response({"ok": f"succesfully disenrolled {username} from {obj.course_name}"}, status=status.HTTP_200_OK)
+
 class MassEnrollCourseView(generics.GenericAPIView):
     authentication_classes = [JWTAuthentication]
     permission_classes = [IsAdmin]