lint fix

Author: brandonistfan

tcf_website/api/urls.py

@@ -12,12 +12,8 @@
 router.register(r"courses", views.CourseViewSet)
 router.register(r"instructors", views.InstructorViewSet)
 router.register(r"semesters", views.SemesterViewSet)
+router.register(r"enrollment", views.SectionEnrollmentViewSet, basename='enrollment')
 
 urlpatterns = [
     path("", include(router.urls)),
-    path(
-        'enrollment/<int:course_id>/',
-        views.get_section_enrollment,
-        name='get_section_enrollment'
-    ),
 ]

tcf_website/api/views.py

@@ -1,9 +1,10 @@
 # pylint: disable=too-many-ancestors,fixme
 """DRF Viewsets"""
 from django.db.models import Avg, Sum
-from rest_framework import viewsets
 from django.http import JsonResponse
 
+from rest_framework import viewsets
+
 from ..models import (
     Course,
     Department,
@@ -149,19 +150,22 @@ def get_queryset(self):
         return super().get_queryset().filter(**params).distinct().order_by("-number")
 
 
-def get_section_enrollment(request, course_id):
-    """Retrieves enrollment data for all sections of a given course."""
-    sections = Section.objects.filter(course_id=course_id)
-    enrollment_data = {}
+class SectionEnrollmentViewSet(viewsets.ViewSet):
+    """ViewSet for retrieving section enrollment data."""
+
+    def retrieve(self, request, pk=None):
+        """Retrieves enrollment data for all sections of a given course."""
+        sections = Section.objects.filter(course_id=pk)
+        enrollment_data = {}
 
-    for section in sections:
-        section_enrollment = SectionEnrollment.objects.filter(section=section).first()
-        if section_enrollment:
-            enrollment_data[section.sis_section_number] = {
-                'enrollment_taken': section_enrollment.enrollment_taken,
-                'enrollment_limit': section_enrollment.enrollment_limit,
-                'waitlist_taken': section_enrollment.waitlist_taken,
-                'waitlist_limit': section_enrollment.waitlist_limit
-            }
+        for section in sections:
+            section_enrollment = SectionEnrollment.objects.filter(section=section).first()
+            if section_enrollment:
+                enrollment_data[section.sis_section_number] = {
+                    'enrollment_taken': section_enrollment.enrollment_taken,
+                    'enrollment_limit': section_enrollment.enrollment_limit,
+                    'waitlist_taken': section_enrollment.waitlist_taken,
+                    'waitlist_limit': section_enrollment.waitlist_limit
+                }
 
-    return JsonResponse({'enrollment_data': enrollment_data})
+        return JsonResponse({'enrollment_data': enrollment_data})