Feat/search bar/backend

csm_web/scheduler/urls.py

@@ -23,4 +23,7 @@
     path("matcher/<int:pk>/mentors/", views.matcher.mentors),
     path("matcher/<int:pk>/configure/", views.matcher.configure),
     path("matcher/<int:pk>/create/", views.matcher.create),
+    path(
+        "search/get-sections/", views.get_sections_of_user, name="get_sections_of_user"
+    ),
 ]

csm_web/scheduler/views/__init__.py

@@ -1,7 +1,8 @@
-from .course import CourseViewSet
 from . import matcher
+from .course import CourseViewSet
 from .profile import ProfileViewSet
 from .resource import ResourceViewSet
+from .search import get_sections_of_user
 from .section import SectionViewSet
 from .spacetime import SpacetimeViewSet
 from .student import StudentViewSet

csm_web/scheduler/views/search.py

@@ -0,0 +1,123 @@
+from django.db.models import Count, Q
+from rest_framework import status
+from rest_framework.decorators import api_view
+from rest_framework.exceptions import PermissionDenied
+from rest_framework.response import Response
+from scheduler.models import Section, Student
+from scheduler.serializers import SectionSerializer, StudentSerializer
+
+
+@api_view(["GET"])
+def get_sections_of_user(request):
+    """
+    Gets all sections that match the queried user.
+    """
+
+    is_coordinator = bool(
+        request.user.coordinator_set.filter(user=request.user).count()
+    )
+    if not is_coordinator:
+        raise PermissionDenied(
+            "You are not authorized to search through sections of this course."
+        )
+
+    query = request.query_params.get("query", "")
+    student_absences = request.query_params.get("student_absences", None)
+
+    if not query and student_absences is None:
+        return Response(
+            {"error": "Please provide a query"}, status=status.HTTP_400_BAD_REQUEST
+        )
+
+    sections = Section.objects.all()
+
+    # Fetch courses associated with the user's coordinator role
+    courses = request.user.coordinator_set.values_list("course", flat=True)
+
+    # Filter sections based on the courses associated with the user's coordinator role
+    sections = sections.filter(mentor__course__in=courses)
+
+    if query:
+        sections = sections.filter(
+            # pylint: disable-next=unsupported-binary-operation
+            Q(students__user__first_name__icontains=query)
+            | Q(students__user__last_name__icontains=query)
+            | Q(students__user__email__icontains=query)
+            | Q(mentor__user__first_name__icontains=query)
+            | Q(mentor__user__last_name__icontains=query)
+            | Q(mentor__user__email__icontains=query)
+        ).distinct()
+
+        students = Student.objects.filter(
+            # pylint: disable-next=unsupported-binary-operation
+            Q(user__first_name__icontains=query)
+            | Q(user__last_name__icontains=query)
+            | Q(user__email__icontains=query),
+            section__in=sections,
+        ).distinct()
+
+    if student_absences is not None:
+        try:
+            # Check if the query is a range or single number
+            if "-" in student_absences:
+                start, end = student_absences.split("-")
+                students = (
+                    Student.objects.annotate(
+                        num_absences=Count(
+                            "attendance", filter=Q(attendance__presence="UN")
+                        )
+                    )
+                    .filter(
+                        num_absences__gte=start,
+                        num_absences__lte=end,
+                        section__in=sections,
+                    )
+                    .distinct()
+                )
+            else:
+                num_absences = int(student_absences)
+                students = (
+                    Student.objects.annotate(
+                        num_absences=Count(
+                            "attendance", filter=Q(attendance__presence="UN")
+                        )
+                    )
+                    .filter(num_absences=num_absences, section__in=sections)
+                    .distinct()
+                )
+            sections = sections.filter(students__in=students).distinct()
+        except ValueError:
+            return Response(
+                {
+                    "error": (
+                        "Invalid value for student_absences. Please provide an integer."
+                    )
+                },
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+
+    section_serializer = SectionSerializer(
+        sections, many=True, context={"request": request}
+    )
+
+    student_data = []
+    for student in students:
+        # attendance_data = student.attendance.filter(presence="UN")
+        student_serializer = StudentSerializer(student)
+        student_data.append(
+            {
+                "id": student_serializer.data["id"],
+                "name": student_serializer.data["name"],
+                "email": student_serializer.data["email"],
+                "section": student_serializer.data["section"],
+                "mentor": student.section.mentor.user.get_full_name(),
+                "num_absences": student.num_absences,
+            }
+        )
+
+    combined_data = {
+        "sections": section_serializer.data,
+        "students": student_data,
+    }
+
+    return Response(combined_data, status=status.HTTP_200_OK)