refactor: [AXM-288] Refactor get_course_assignments

Co-authored-by: monteri <[email protected]>
raccoongang · Apr 24, 2024 · f53aff4 · f53aff4
1 parent b6f12cb
commit f53aff4
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/lms/djangoapps/mobile_api/users/serializers.py b/lms/djangoapps/mobile_api/users/serializers.py
@@ -279,11 +279,13 @@ def get_course_assignments(self, model: CourseEnrollment) -> Dict[str, Optional[
                 if not assignment.complete:
                     next_assignment_all.append(assignment)
 
-        if next_assignment_all:
-            future_assignment_date = next_assignment_all[0].date.date()
-            next_assignment = [
-                assignment for assignment in next_assignment_all if assignment.date.date() == future_assignment_date
-            ]
+if future_assignments:
+        future_assignment_date = future_assignments[0].date.date()
+        next_assignments = [
+            assignment for assignment in future_assignments if assignment.date.date() == future_assignment_date
+        ]
+    else:
+        next_assignments = []
 
         return {
             'future_assignments': DateSummarySerializer(next_assignment, many=True).data,